revert github workflow file, py code, ut changes

Author: frainfreeze

.github/workflows/release.yml

@@ -101,3 +101,19 @@ jobs:
           files: dist/*
           fail_on_unmatched_files: true
           generate_release_notes: true
+
+      # TODO: use python exec instead
+      - uses: slackapi/[email protected]
+        if: always()
+        with:
+          method: chat.postMessage
+          token: ${{ secrets.SLACK_BOT_TOKEN }}
+          payload: |
+            username: ${{ job.status == 'success' && format('Released codegen@{0}', github.ref_name) || format('Failed to release codegen@{0}', github.ref_name) }}
+            channel: "#release"
+            icon_emoji: "${{ job.status == 'success' && ':white_check_mark:' || ':x:' }}"
+            text: |
+              Actor: `${{ github.triggering_actor }}`
+              Author: `${{ github.event.head_commit.author.username }}`
+              ${{ format('Commit: <{0}/{1}/commit/{2}|{1}@{2}>', github.server_url, github.repository, github.sha) || ''}}
+              View <${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}|GHA logs>

src/codegen/sdk/python/file.py

@@ -200,6 +200,13 @@ def remove_unused_imports(self) -> None:
 
         self.G.commit_transactions()
 
+    def remove_unused_exports(self) -> None:
+        """Removes unused exports from the file.
+        In Python this is equivalent to removing unused imports since Python doesn't have
+        explicit export statements. Calls remove_unused_imports() internally.
+        """
+        self.remove_unused_imports()
+
     @cached_property
     @noapidoc
     @reader(cache=True)

src/codegen/sdk/typescript/file.py

@@ -6,7 +6,7 @@
 from codegen.sdk.core.autocommit import commiter, mover, reader, writer
 from codegen.sdk.core.file import SourceFile
 from codegen.sdk.core.interfaces.exportable import Exportable
-from codegen.sdk.enums import ImportType, NodeType, ProgrammingLanguage, SymbolType
+from codegen.sdk.enums import ImportType, ProgrammingLanguage, SymbolType
 from codegen.sdk.extensions.sort import sort_editables
 from codegen.sdk.extensions.utils import cached_property
 from codegen.sdk.typescript.assignment import TSAssignment
@@ -449,56 +449,31 @@ def remove_unused_exports(self) -> None:
         - Exports used by other files through imports
         - Exports used within the same file
         """
+        exports_to_remove = []
+
         for export in self.exports:
-            # Skip type exports and default exports
-            if export.is_type_export() or export.is_default_export():
+            # Skip type exports
+            if export.is_type_export():
+                continue
+
+            # Skip default exports
+            if export.is_default_export():
                 continue
 
-            symbol_export_unused = True
-            symbols_to_remove = []
-
-            exported_symbol = export.resolved_symbol
-            for export_usage in export.symbol_usages:
-                if export_usage.node_type == NodeType.IMPORT or (export_usage.node_type == NodeType.EXPORT and export_usage.resolved_symbol != exported_symbol):
-                    # If the import has no usages then we can add the import to the list of symbols to remove
-                    reexport_usages = export_usage.symbol_usages
-                    if len(reexport_usages) == 0:
-                        symbols_to_remove.append(export_usage)
-                        break
-
-                    # If any of the import's usages are valid symbol usages, export is used.
-                    if any(usage.node_type == NodeType.SYMBOL for usage in reexport_usages):
-                        symbol_export_unused = False
-                        break
-
-                    symbols_to_remove.append(export_usage)
-
-                elif export_usage.node_type == NodeType.SYMBOL:
-                    symbol_export_unused = False
-                    break
-
-            # export is not used, remove it
-            if symbol_export_unused:
-                # remove the unused imports
-                for imp in symbols_to_remove:
-                    imp.remove()
-
-                # Handle different export types
-                if hasattr(export, "source") and export.source:
-                    # Re-export case (export { x } from 'y')
-                    export.remove()
-                elif exported_symbol and hasattr(exported_symbol, "export") and exported_symbol.export:
-                    if exported_symbol.export.declared_symbol == exported_symbol:
-                        # Direct export case (export function x)
-                        if exported_symbol.source.startswith("export default "):
-                            exported_symbol.replace("export default ", "")
-                        else:
-                            exported_symbol.replace("export ", "")
-                    else:
-                        # Export statement case (export { x })
-                        exported_symbol.export.remove()
-                else:
-                    # Fallback - just remove the export
-                    export.remove()
+            # Check if export is used
+            has_usages = bool(export.symbol_usages)
+
+            # For re-exports, check if the re-exported symbol is used
+            if export.is_reexport():
+                if export.resolved_symbol and export.resolved_symbol.symbol_usages:
+                    continue
+
+            # Remove if no usages found
+            if not has_usages:
+                exports_to_remove.append(export)
+
+        # Remove unused exports
+        for export in exports_to_remove:
+            export.remove()
 
         self.G.commit_transactions()

tests/unit/codegen/sdk/typescript/file/test_file_remove.py

@@ -1,5 +1,7 @@
 import os
 
+import pytest
+
 from codegen.sdk.codebase.factory.get_session import get_codebase_session
 from codegen.sdk.enums import ProgrammingLanguage
 
@@ -103,19 +105,20 @@ def test_remove_unused_imports_with_moved_symbols(tmpdir):
         assert main_file.content.strip() == ""
 
 
+@pytest.mark.skip(reason="This test is not implemented properly yet")
 def test_remove_unused_exports_with_side_effects(tmpdir):
     content = """
-    import './styles.css';
-    export const unused = 5;
-    export function usedFunction() { return true; }
+import './styles.css';
+export const unused = 5;
+export function usedFunction() { return true; }
 
-    const x = usedFunction();
+const x = usedFunction();
     """
     expected = """
-    import './styles.css';
-    export function usedFunction() { return true; }
+import './styles.css';
+export function usedFunction() { return true; }
 
-    const x = usedFunction();
+const x = usedFunction();
     """
 
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.TYPESCRIPT, files={"test.ts": content}) as codebase:
@@ -124,22 +127,23 @@ def test_remove_unused_exports_with_side_effects(tmpdir):
         assert file.content.strip() == expected.strip()
 
 
+@pytest.mark.skip(reason="This test is not implemented properly yet")
 def test_remove_unused_exports_with_multiple_types(tmpdir):
     content = """
-    export const UNUSED_CONSTANT = 42;
-    export type UnusedType = string;
-    export interface UnusedInterface {}
-    export default function main() { return true; }
-    export function usedFunction() { return true; }
-    const x = usedFunction();
+export const UNUSED_CONSTANT = 42;
+export type UnusedType = string;
+export interface UnusedInterface {}
+export default function main() { return true; }
+export function usedFunction() { return true; }
+const x = usedFunction();
     """
     # Only value exports that are unused should be removed
     expected = """
-    export type UnusedType = string;
-    export interface UnusedInterface {}
-    export default function main() { return true; }
-    export function usedFunction() { return true; }
-    const x = usedFunction();
+export type UnusedType = string;
+export interface UnusedInterface {}
+export default function main() { return true; }
+export function usedFunction() { return true; }
+const x = usedFunction();
     """
 
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.TYPESCRIPT, files={"test.ts": content}) as codebase:
@@ -148,19 +152,20 @@ def test_remove_unused_exports_with_multiple_types(tmpdir):
         assert file.content.strip() == expected.strip()
 
 
+@pytest.mark.skip(reason="This test is not implemented properly yet")
 def test_remove_unused_exports_with_reexports(tmpdir):
     content1 = """
-    export { helper } from './utils';
-    export { unused } from './other';
-    export function localFunction() { return true; }
+export { helper } from './utils';
+export { unused } from './other';
+export function localFunction() { return true; }
     """
     content2 = """
-    import { helper } from './main';
-    const x = helper();
+import { helper } from './main';
+const x = helper();
     """
     expected1 = """
-    export { helper } from './utils';
-    export function localFunction() { return true; }
+export { helper } from './utils';
+export function localFunction() { return true; }
     """
 
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.TYPESCRIPT, files={"main.ts": content1, "other.ts": content2}) as codebase:

tests/unit/codegen/sdk/typescript/move_symbol_to_file/test_move_tsx_to_file.py

@@ -74,9 +74,9 @@ def test_move_component_with_dependencies(tmpdir) -> None:
 
 def test_remove_unused_exports(tmpdir) -> None:
     """Tests removing unused exports when moving components between files"""
-    src_filename = "Component.tsx"
+    # ========== [ BEFORE ] ==========
     # language=typescript jsx
-    src_content = """
+    SRC_CONTENT = """
 export default function MainComponent() {
   const [state, setState] = useState<StateType | null>()
   return (<div>
@@ -116,9 +116,8 @@ def test_remove_unused_exports(tmpdir) -> None:
     )
 }
 """
-    adj_filename = "adjacent.tsx"
     # language=typescript jsx
-    adj_content = """
+    ADJ_CONTENT = """
 import MainComponent from 'Component'
 import { SharedComponent } from 'Component'
 import { StateComponent } from 'utils'
@@ -127,26 +126,82 @@ def test_remove_unused_exports(tmpdir) -> None:
     return (<Wrapper components={[MainComponent, SharedComponent]}/>)
 }
 """
-    misc_filename = "misc.tsx"
     # language=typescript jsx
-    misc_content = """
+    MISC_CONTENT = """
 export { UnusedComponent } from 'Component'
 function Helper({ props }: HelperProps) {}
 
 export { Helper }
 """
-    import_filename = "import.tsx"
     # language=typescript jsx
-    import_content = """
+    IMPORT_CONTENT = """
 import { UnusedComponent } from 'misc'
 """
 
-    files = {src_filename: src_content, adj_filename: adj_content, misc_filename: misc_content, import_filename: import_content}
+    # ========== [ AFTER ] ==========
+    # language=typescript jsx
+    EXPECTED_SRC_CONTENT = """
+export default function MainComponent() {
+  const [state, setState] = useState<StateType | null>()
+  return (<div>
+            <div>
+                <SubComponent/>
+            </div>
+          </div>)
+}
+
+function UnusedComponent({ props }: UnusedProps) {
+    return (
+        <div> Unused </div>
+    )
+}
+"""
+    # language=typescript jsx
+    EXPECTED_NEW_CONTENT = """
+export function SubComponent({ props }: SubComponentProps) {
+  return (
+    <HelperComponent size='s'/>
+  )
+}
+
+function HelperComponent({ props }: HelperComponentProps) {
+  return (
+    <SharedComponent size='l'/>
+  )
+}
+
+export function SharedComponent({ props }: SharedComponentProps) {
+  return (
+    <div> <StateComponent/> </div>
+  )
+}
+"""
+    # language=typescript jsx
+    EXPECTED_ADJ_CONTENT = """
+import MainComponent from 'Component'
+import { SharedComponent } from 'new'
+import { StateComponent } from 'utils'
+
+function Container(props: ContainerProps) {
+    return (<Wrapper components={[MainComponent, SharedComponent]}/>)
+}
+"""
+    # language=typescript jsx
+    EXPECTED_MISC_CONTENT = """
+function Helper({ props }: HelperProps) {}
+"""
+
+    files = {
+        "Component.tsx": SRC_CONTENT,
+        "adjacent.tsx": ADJ_CONTENT,
+        "misc.tsx": MISC_CONTENT,
+        "import.tsx": IMPORT_CONTENT
+    }
 
     with get_codebase_session(tmpdir=tmpdir, programming_language=ProgrammingLanguage.TYPESCRIPT, files=files) as codebase:
-        src_file = codebase.get_file(src_filename)
-        adj_file = codebase.get_file(adj_filename)
-        misc_file = codebase.get_file(misc_filename)
+        src_file = codebase.get_file("Component.tsx")
+        adj_file = codebase.get_file("adjacent.tsx")
+        misc_file = codebase.get_file("misc.tsx")
         new_file = codebase.create_file("new.tsx")
 
         sub_component = src_file.get_symbol("SubComponent")
@@ -159,20 +214,7 @@ def test_remove_unused_exports(tmpdir) -> None:
         src_file.remove_unused_exports()
         misc_file.remove_unused_exports()
 
-    # Verify exports in new file
-    assert "export function SubComponent" in new_file.content
-    assert "function HelperComponent" in new_file.content
-    assert "export function HelperComponent" not in new_file.content
-    assert "export function SharedComponent" in new_file.content
-
-    # Verify imports updated
-    assert "import { SharedComponent } from 'new'" in adj_file.content
-
-    # Verify original file exports
-    assert "export default function MainComponent()" in src_file.content
-    assert "function UnusedComponent" in src_file.content
-    assert "export function UnusedComponent" not in src_file.content
-
-    # Verify misc file exports cleaned up
-    assert "export { Helper }" not in misc_file.content
-    assert "export { UnusedComponent } from 'Component'" not in misc_file.content
+    assert src_file.content.strip() == EXPECTED_SRC_CONTENT.strip()
+    assert new_file.content.strip() == EXPECTED_NEW_CONTENT.strip()
+    assert adj_file.content.strip() == EXPECTED_ADJ_CONTENT.strip()
+    assert misc_file.content.strip() == EXPECTED_MISC_CONTENT.strip()