Renderer to website

.github/workflows/ci.yml

@@ -0,0 +1,26 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+    paths:
+      - 'media/**'
+      - 'extension/**'
+  pull_request:
+    branches: [main]
+    paths:
+      - 'media/**'
+      - 'extension/**'
+
+jobs:
+  extension:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-node@v3
+        with:
+          node-version: 18
+      - name: Install extension deps
+        run: cd extension && npm install
+      - name: Compile extension
+        run: cd extension && npm run compile

.vscode/launch.json

@@ -10,10 +10,10 @@
 			"type": "extensionHost",
 			"request": "launch",
 			"args": [
-				"--extensionDevelopmentPath=${workspaceFolder}"
+				"--extensionDevelopmentPath=${workspaceFolder}/extension"
 			],
 			"outFiles": [
-				"${workspaceFolder}/out/**/*.js"
+				"${workspaceFolder}/extension/out/**/*.js"
 			],
 			"preLaunchTask": "${defaultBuildTask}"
 		}

README.md

@@ -1,6 +1,6 @@
 # Codescape
 
-Codescape is a VS Code extension that parses Java code and renders it as an isometric city.
+Codescape is a VS Code extension that parses Java and Python code and renders it as an isometric city.
 
 - Buildings represent classes/interfaces.
 - Height is based on methods + fields.
@@ -15,6 +15,7 @@ Active prototype with working parser, watcher, relationship graph, and canvas re
 | Area | Status | Notes |
 |---|---|---|
 | Java parsing (Tree-sitter) | Implemented | Classes/interfaces, methods, fields, constructors |
+| Python parsing (Tree-sitter) | Implemented | Classes, module nodes, methods, fields, inheritance |
 | Relationship graph | Implemented | Extends/implements/field/ctor dependencies |
 | Incremental updates | Implemented | Watcher emits `PARTIAL_STATE` |
 | Explorer webview | Implemented | `codescape.Cityview` |
@@ -43,6 +44,10 @@ npm run compile
    - Open the explorer view `Codescape City`, or
    - Run command `Create Panel` (`codescape.createPanel`).
 
+Known-good example workspaces:
+- `examples/java-city`
+- `examples/python-city`
+
 ## Development
 
 - Watch compile:

docs/USAGE.md

@@ -5,7 +5,7 @@
 ### Prerequisites
 - VS Code `^1.74.0`
 - Node.js + npm
-- Java project/workspace with `.java` files
+- Java and/or Python project/workspace with `.java` / `.py` files
 
 ### Build and run from source
 1. Clone and enter repo.
@@ -37,7 +37,7 @@ Placement options in VS Code:
 
 ## 3) What the Visualization Means
 
-- Building = Java class/interface (`ClassInfo`).
+- Building = parsed Java or Python entity (`ClassInfo`).
 - Building height = complexity proxy (`methods + fields`, min 1).
 - Building color = relationship signal (intended UX) and currently implemented as stable per-class palette assignment.
 - Related classes = computed in backend (`relations.ts`) for incremental updates.
@@ -56,7 +56,7 @@ Important current behavior:
 
 ## 5) Incremental Updates
 
-Codescape watches `**/*.java` and sends partial updates:
+Codescape watches `**/*.java` and `**/*.py` and sends partial updates:
 
 - File changed:
   - Re-parse file.
@@ -92,20 +92,29 @@ Runtime-registered internal commands (not contributed in `package.json`):
 ## 8) Practical Workflow
 
 1. Start extension host (`F5`).
-2. Open Java workspace in extension host window.
+2. Open a Java or Python workspace in extension host window.
 3. Open Codescape via sidebar or `Create Panel` command.
-4. Edit/save Java files and observe updates.
+4. Edit/save Java or Python files and observe updates.
 5. Use zoom wheel to inspect layout.
 
 ## 9) Troubleshooting
 
 - City view not updating:
-  - Confirm `.java` file is not excluded by `.exclude`.
+  - Confirm the `.java` or `.py` file is not excluded by `.exclude`.
   - Check extension host logs for parser errors.
 
 - Empty view:
-  - Verify workspace has Java files.
+  - Verify workspace has Java or Python files.
   - Run `codescape.scan` (re-parses backend store; current implementation may still require reopening/reloading view to reflect a full-state refresh).
 
 - Commands missing:
   - Recompile (`npm run compile`) and relaunch extension host.
+
+## 10) Example Workspaces
+
+For a known-good visual smoke test, open one of these folders in the extension host:
+
+- `examples/java-city`
+- `examples/python-city`
+
+Both are intentionally small so the generated city stays compact and in view with the current layout algorithm.

examples/java-city/RouteMap.java

@@ -0,0 +1,11 @@
+public class RouteMap {
+  private int districtCode;
+
+  public RouteMap(int districtCode) {
+    this.districtCode = districtCode;
+  }
+
+  public int getDistrictCode() {
+    return districtCode;
+  }
+}

examples/java-city/ShuttleService.java

@@ -0,0 +1,11 @@
+public class ShuttleService {
+  private TransitHub hub;
+
+  public ShuttleService(TransitHub hub) {
+    this.hub = hub;
+  }
+
+  public TransitHub getHub() {
+    return hub;
+  }
+}

examples/java-city/TransitHub.java

@@ -0,0 +1,29 @@
+public class TransitHub {
+  private RouteMap routeMap;
+  private Gate primaryGate;
+
+  public TransitHub(RouteMap routeMap, Gate primaryGate) {
+    this.routeMap = routeMap;
+    this.primaryGate = primaryGate;
+  }
+
+  public RouteMap getRouteMap() {
+    return routeMap;
+  }
+
+  public Gate getPrimaryGate() {
+    return primaryGate;
+  }
+
+  static class Gate {
+    private int gateNumber;
+
+    Gate(int gateNumber) {
+      this.gateNumber = gateNumber;
+    }
+
+    public int getGateNumber() {
+      return gateNumber;
+    }
+  }
+}

examples/python-city/city_tools.py

@@ -0,0 +1,5 @@
+DEFAULT_BLOCKS = 12
+
+
+def estimate_blocks(distance):
+    return (distance + DEFAULT_BLOCKS - 1) // DEFAULT_BLOCKS

examples/python-city/network.py

@@ -0,0 +1,15 @@
+class BaseStation:
+    def __init__(self, district):
+        self.district = district
+
+    def label(self):
+        return f"Station<{self.district}>"
+
+
+class DispatchCenter(BaseStation):
+    def __init__(self, district):
+        super().__init__(district)
+        self.active_routes = 3
+
+    def assign_route(self, route_name):
+        return f"{route_name}:{self.active_routes}"

examples/python-city/route_map.py

@@ -0,0 +1,6 @@
+class RouteMap:
+    def __init__(self):
+        self.routes = ["A", "B"]
+
+    def count(self):
+        return len(self.routes)

src/JavaFileWatcher.ts → extension/src/JavaFileWatcher.ts

@@ -5,30 +5,26 @@ import { parseAndStore } from './parser';
 import { ClassInfo } from './parser/javaExtractor';
 import { buildGraph, getRelated } from './relations';
 import { WebviewManager } from './WebviewManager';
-
-type IncrementalChangePayload = {
-    changed?: ClassInfo[];
-    related?: ClassInfo[];
-    removed?: string[];
-};
+import { computeCityLayout } from './cityLayout';
+import type { PartialStatePayload } from './types/messages';
 export class JavaFileWatcher {
-    private _watcher: vscode.FileSystemWatcher;
-    private _pythonWatcher: vscode.FileSystemWatcher;
+    private _javaWatcher: vscode.FileSystemWatcher;
+    private _pythonWatcher : vscode.FileSystemWatcher;
 
     constructor(store: FileParseStore, private webviewManager: WebviewManager) {
-        this._watcher = vscode.workspace.createFileSystemWatcher('**/*.java');
+        this._javaWatcher = vscode.workspace.createFileSystemWatcher('**/*.java');
 
-        this._watcher.onDidChange(async (uri: vscode.Uri) => {
+        this._javaWatcher.onDidChange(async (uri: vscode.Uri) => {
             console.log('Java file changed:', uri.fsPath);
             this.handleIncrementalChange(uri, store);
         });
 
-        this._watcher.onDidDelete((uri: vscode.Uri) => {
+        this._javaWatcher.onDidDelete((uri: vscode.Uri) => {
             console.log('Java file deleted:', uri.fsPath);
             const before = store.get(uri);
             const removedNames = (before?.data ?? []).map((c: ClassInfo) => c.Classname);
             store.remove(uri);
-            this.postIncrementalChange({ removed: removedNames });
+            this.postIncrementalChange(this.buildPartialStatePayload([], removedNames, store));
         });
 
         // Python file watcher — same incremental pipeline as Java
@@ -49,37 +45,42 @@ export class JavaFileWatcher {
             const before = store.get(uri);
             const removedNames = (before?.data ?? []).map((c: ClassInfo) => c.Classname);
             store.remove(uri);
-            this.postIncrementalChange({ removed: removedNames });
+            this.postIncrementalChange(this.buildPartialStatePayload([], removedNames, store));
         });
     }
     private buildPartialStatePayload(
         changedClasses: ClassInfo[],
         removedNames: string[],
         store: FileParseStore
-    ): { changed: ClassInfo[]; related: ClassInfo[]; removed: string[] } {
+    ): PartialStatePayload {
         const allClasses = store.snapshot().flatMap(e => e.entry.data ?? []);
+        const layout = computeCityLayout(allClasses);
         const graph = buildGraph(allClasses);
         const changedNames = changedClasses.map(c => c.Classname);
         const relatedNames = getRelated([...changedNames, ...removedNames], graph);
         const relatedClasses = allClasses.filter(c => relatedNames.includes(c.Classname));
-        return { changed: changedClasses, related: relatedClasses, removed: removedNames };
+        return {
+            changed: changedClasses,
+            related: relatedClasses,
+            removed: removedNames,
+            fullClasses: allClasses,
+            layout,
+        };
     }
 
     private async handleIncrementalChange(uri: vscode.Uri, store: FileParseStore) {
         if (!await isExcluded(uri)) {
             const { changed, removed } = await parseAndStore(uri, store);
-            //create payload from parsed data
-            const payload: IncrementalChangePayload = this.buildPartialStatePayload(changed, removed, store);
-            //send message to frontend
+            const payload = this.buildPartialStatePayload(changed, removed, store);
             this.postIncrementalChange(payload);
         }
     }
-    private async postIncrementalChange(payload: IncrementalChangePayload) {
+    private postIncrementalChange(payload: PartialStatePayload): void {
         this.webviewManager.broadcastPartialState(payload);
     }
 
     dispose() {
-        this._watcher.dispose();
+        this._javaWatcher.dispose();
         this._pythonWatcher.dispose();
     }
-}
+}