feat: add routes on configure

Author: Rei-x

packages/ally-solvro-auth/configure.ts

@@ -4,10 +4,78 @@
  * @license MIT
  * @copyright Romain Lanz <[email protected]>
  */
+import assert from "node:assert";
+
+import { CodeTransformer } from "@adonisjs/assembler/code_transformer";
 import type Configure from "@adonisjs/core/commands/configure";
 
 import { stubsRoot } from "./stubs/main.js";
 
+type CodeTransformerProject = CodeTransformer["project"];
+
+type SourceFile = ReturnType<CodeTransformerProject["createSourceFile"]>;
+
+function addImportIfNotExists(
+  sourceFile: SourceFile,
+  importPath: string,
+  importName: string,
+) {
+  const existingImports = sourceFile.getImportDeclarations();
+  const hasImport = existingImports.some(
+    (imp) =>
+      imp.getModuleSpecifierValue() === importPath &&
+      imp.getNamedImports().some((named) => named.getName() === importName),
+  );
+
+  if (!hasImport) {
+    sourceFile.addImportDeclaration({
+      moduleSpecifier: importPath,
+      namedImports: [{ name: importName }],
+    });
+  }
+}
+
+function addControllerImportIfNotExists(sourceFile: SourceFile) {
+  const existingVariables = sourceFile.getVariableDeclarations();
+  const hasControllerImport = existingVariables.some((imp) =>
+    imp.getText().includes("AuthController"),
+  );
+
+  if (!hasControllerImport) {
+    sourceFile.addVariableStatement({
+      declarations: [
+        {
+          name: "AuthController",
+          initializer: '() => import("#controllers/auth_controller")',
+        },
+      ],
+    });
+  }
+}
+
+function addNewRoutes(sourceFile: SourceFile) {
+  // Find the last router statement
+  const routerStatements = sourceFile
+    // @ts-expect-error ??
+    .getDescendantsOfKind(SyntaxKind.CallExpression)
+    // @ts-expect-error ??
+    .filter((call) => call.getExpression().getText().startsWith("router."));
+
+  const lastRouterStatement = routerStatements[routerStatements.length - 1];
+
+  // Add new routes after the last existing route
+  const newRoutes = `
+router.get("/auth/login", [AuthController, "login"]).use(middleware.guest());
+router.get("/auth/callback", [AuthController, "callback"]).use(middleware.guest());
+router.post("/auth/logout", [AuthController, "logout"]).use(middleware.auth());
+`;
+  if (lastRouterStatement) {
+    sourceFile.insertText(lastRouterStatement.getEnd() + 1, newRoutes);
+  } else {
+    sourceFile.addStatements(newRoutes);
+  }
+}
+
 export async function configure(command: Configure) {
   const codemods = await command.createCodemods();
 
@@ -46,4 +114,15 @@ export async function configure(command: Configure) {
     },
     leadingComment: "Variables for @solvro/auth",
   });
+
+  const project = await codemods.getTsMorphProject();
+  assert(project);
+
+  const file = project?.getSourceFileOrThrow("start/routes.ts");
+
+  addImportIfNotExists(file, "./kernel.js", "middleware");
+  addControllerImportIfNotExists(file);
+  addNewRoutes(file);
+
+  await file.save();
 }

packages/ally-solvro-auth/package.json

@@ -63,6 +63,7 @@
     "eslint": "^9.20.1",
     "prettier": "^3.5.1",
     "semantic-release": "^24.2.2",
+    "ts-morph": "^25.0.1",
     "ts-node": "^10.9.2",
     "typescript": "^5.7.3"
   },