fix(ui): fix field sorting

closes #4934

invokeai/frontend/web/src/features/nodes/hooks/useAnyOrDirectInputFieldNames.ts

@@ -9,6 +9,7 @@ import {
   POLYMORPHIC_TYPES,
   TYPES_WITH_INPUT_COMPONENTS,
 } from '../types/constants';
+import { getSortedFilteredFieldNames } from '../util/getSortedFilteredFieldNames';
 
 export const useAnyOrDirectInputFieldNames = (nodeId: string) => {
   const selector = useMemo(
@@ -24,17 +25,13 @@ export const useAnyOrDirectInputFieldNames = (nodeId: string) => {
           if (!nodeTemplate) {
             return [];
           }
-          return map(nodeTemplate.inputs)
+          const fields = map(nodeTemplate.inputs).filter(
-            .filter(
             (field) =>
               (['any', 'direct'].includes(field.input) ||
                 POLYMORPHIC_TYPES.includes(field.type)) &&
               TYPES_WITH_INPUT_COMPONENTS.includes(field.type)
-            )
+          );
-            .filter((field) => !field.ui_hidden)
+          return getSortedFilteredFieldNames(fields);
-            .sort((a, b) => (a.ui_order ?? 0) - (b.ui_order ?? 0))
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
         },
         defaultSelectorOptions
       ),

invokeai/frontend/web/src/features/nodes/hooks/useConnectionInputFieldNames.ts

@@ -9,6 +9,7 @@ import {
   TYPES_WITH_INPUT_COMPONENTS,
 } from '../types/constants';
 import { isInvocationNode } from '../types/types';
+import { getSortedFilteredFieldNames } from '../util/getSortedFilteredFieldNames';
 
 export const useConnectionInputFieldNames = (nodeId: string) => {
   const selector = useMemo(
@@ -24,17 +25,16 @@ export const useConnectionInputFieldNames = (nodeId: string) => {
           if (!nodeTemplate) {
             return [];
           }
-          return map(nodeTemplate.inputs)
+
-            .filter(
+          // get the visible fields
+          const fields = map(nodeTemplate.inputs).filter(
             (field) =>
               (field.input === 'connection' &&
                 !POLYMORPHIC_TYPES.includes(field.type)) ||
               !TYPES_WITH_INPUT_COMPONENTS.includes(field.type)
-            )
+          );
-            .filter((field) => !field.ui_hidden)
+
-            .sort((a, b) => (a.ui_order ?? 0) - (b.ui_order ?? 0))
+          return getSortedFilteredFieldNames(fields);
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
         },
         defaultSelectorOptions
       ),

invokeai/frontend/web/src/features/nodes/hooks/useOutputFieldNames.ts

@@ -5,6 +5,7 @@ import { defaultSelectorOptions } from 'app/store/util/defaultMemoizeOptions';
 import { map } from 'lodash-es';
 import { useMemo } from 'react';
 import { isInvocationNode } from '../types/types';
+import { getSortedFilteredFieldNames } from '../util/getSortedFilteredFieldNames';
 
 export const useOutputFieldNames = (nodeId: string) => {
   const selector = useMemo(
@@ -20,11 +21,8 @@ export const useOutputFieldNames = (nodeId: string) => {
           if (!nodeTemplate) {
             return [];
           }
-          return map(nodeTemplate.outputs)
+
-            .filter((field) => !field.ui_hidden)
+          return getSortedFilteredFieldNames(map(nodeTemplate.outputs));
-            .sort((a, b) => (a.ui_order ?? 0) - (b.ui_order ?? 0))
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
         },
         defaultSelectorOptions
       ),

invokeai/frontend/web/src/features/nodes/util/getSortedFilteredFieldNames.ts

@@ -0,0 +1,20 @@
+import { isNil } from 'lodash-es';
+import { InputFieldTemplate, OutputFieldTemplate } from '../types/types';
+
+export const getSortedFilteredFieldNames = (
+  fields: InputFieldTemplate[] | OutputFieldTemplate[]
+): string[] => {
+  const visibleFields = fields.filter((field) => !field.ui_hidden);
+
+  // we want explicitly ordered fields to be before unordered fields; split the list
+  const orderedFields = visibleFields
+    .filter((f) => !isNil(f.ui_order))
+    .sort((a, b) => (a.ui_order ?? 0) - (b.ui_order ?? 0));
+  const unorderedFields = visibleFields.filter((f) => isNil(f.ui_order));
+
+  // concat the lists, and return the field names, skipping `is_intermediate`
+  return orderedFields
+    .concat(unorderedFields)
+    .map((f) => f.name)
+    .filter((fieldName) => fieldName !== 'is_intermediate');
+};