refactor: function return type container

Author: canerakdas

packages/ui-components/src/Common/Signature/SignatureRoot/index.module.css

@@ -12,7 +12,8 @@
 }
 
 .root {
-  @apply flex
+  @apply mb-1
+    flex
     flex-col
     gap-4
     rounded-sm

packages/ui-components/src/Containers/FunctionSignature/index.stories.tsx

@@ -118,6 +118,17 @@ export const Nested: Story = {
           },
         ],
       },
+      {
+        name: 'Returns',
+        kind: 'return',
+        description: 'description',
+        type: (
+          <>
+            <a href="#">&lt;Promise&gt;</a>|
+            <a href="#">&lt;AsyncIterable&gt;</a>
+          </>
+        ),
+      },
     ],
   },
 };

packages/ui-components/src/Containers/FunctionSignature/index.tsx

@@ -29,10 +29,24 @@ const renderSignature = (param: SignatureDefinition, index: number) => (
 
 const FunctionSignature: FC<FunctionSignatureProps> = ({ title, items }) => {
   if (title) {
+    const attributes: Array<SignatureDefinition> = [];
+    const returnTypes: Array<SignatureDefinition> = [];
+
+    for (const item of items) {
+      (item.kind === 'return' ? returnTypes : attributes).push(item);
+    }
+
     return (
-      <Signature title={title}>
-        {items.map((param, i) => renderSignature(param, i))}
-      </Signature>
+      <>
+        <Signature title={title}>
+          {attributes.map((param, i) => renderSignature(param, i))}
+        </Signature>
+        {returnTypes.length > 0 && (
+          <Signature>
+            {returnTypes.map((param, i) => renderSignature(param, i))}
+          </Signature>
+        )}
+      </>
     );
   }