fix(ui): fix field sorting

closes invoke-ai#4934
mickr777 · Oct 18, 2023 · 024aa5e · 024aa5e
1 parent 67a343b
commit 024aa5e
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 27 deletions.
diff --git a/invokeai/frontend/web/src/features/nodes/hooks/useAnyOrDirectInputFieldNames.ts b/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)
-            .filter(
-              (field) =>
-                (['any', 'direct'].includes(field.input) ||
-                  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))
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
+          const fields = map(nodeTemplate.inputs).filter(
+            (field) =>
+              (['any', 'direct'].includes(field.input) ||
+                POLYMORPHIC_TYPES.includes(field.type)) &&
+              TYPES_WITH_INPUT_COMPONENTS.includes(field.type)
+          );
+          return getSortedFilteredFieldNames(fields);
         },
         defaultSelectorOptions
       ),

diff --git a/invokeai/frontend/web/src/features/nodes/hooks/useConnectionInputFieldNames.ts b/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(
-              (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))
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
+
+          // 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)
+          );
+
+          return getSortedFilteredFieldNames(fields);
         },
         defaultSelectorOptions
       ),

diff --git a/invokeai/frontend/web/src/features/nodes/hooks/useOutputFieldNames.ts b/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)
-            .sort((a, b) => (a.ui_order ?? 0) - (b.ui_order ?? 0))
-            .map((field) => field.name)
-            .filter((fieldName) => fieldName !== 'is_intermediate');
+
+          return getSortedFilteredFieldNames(map(nodeTemplate.outputs));
         },
         defaultSelectorOptions
       ),

diff --git a/invokeai/frontend/web/src/features/nodes/util/getSortedFilteredFieldNames.ts b/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');
+};