✨ Add support for route generator

Author: AnandChowdhary

.gitignore

@@ -134,3 +134,4 @@ typings/
 # Built files
 dist/
 docs/
+src/app.ts

package.json

@@ -7,15 +7,15 @@
   "license": "MIT",
   "scripts": {
     "build-babel": "touch .env && babel src -d dist --extensions '.ts' && cp .env dist/.env",
-    "build": "touch .env && mkdir -p dist && cp .env dist/.env && tsc",
-    "check-types": "tsc",
-    "start": "npm run build-babel && node dist/index.js",
+    "build": "touch .env && mkdir -p dist && cp .env dist/.env && npm run generate-routes && tsc",
+    "start": "npm run build && node dist/index.js",
     "dev": "concurrently 'yarn watch-build' 'yarn watch-server'",
-    "prettier-all": "prettier '{src,examples}/**/*.{ts,json}' --write",
+    "prettier-all": "prettier '{src,examples,setup}/**/*.{ts,js,json}' --write",
     "watch-build": "onchange '{src,examples}/**/*.ts' -- yarn build-babel",
     "watch-prettier": "onchange '{src,examples}/**/*.{ts,json}' -- prettier --write {{changed}}",
     "watch-server": "cd dist && nodemon index.js",
-    "generate-docs": "typedoc --out docs ./src"
+    "generate-docs": "typedoc --out docs ./src",
+    "generate-routes": "node setup/controllers.js"
   },
   "husky": {
     "hooks": {

setup/controllers.js

@@ -0,0 +1,32 @@
+const fs = require("fs-extra");
+const path = require("path");
+
+const SRC = path.join(__dirname, "..", "src");
+let server = fs.readFileSync(path.join(SRC, "server.ts")).toString();
+
+// Find controllers
+const controllers = fs.readdirSync(path.join(SRC, "controllers"));
+const exportName = [];
+controllers.forEach(controller => {
+  const controllerFile = fs
+    .readFileSync(path.join(SRC, "controllers", controller))
+    .toString();
+  exportName.push(controllerFile.split("export class ")[1].split(" ")[0]);
+});
+
+const importCode = `${exportName
+  .map(
+    (e, i) =>
+      `import { ${e} } from "./controllers/${controllers[i].split(".ts")[0]}";`
+  )
+  .join("\n")}`;
+
+const insertCode = `
+  super.addControllers([${exportName.map(e => `new ${e}()`).join(", ")}]);
+`;
+
+server = importCode + server.replace("// staart:setup/controllers", insertCode);
+
+fs.writeFileSync(path.join(SRC, "app.ts"), server);
+console.log("Paths generated successfully!");
+process.exit(0);

src/index.ts

@@ -1,4 +1,4 @@
-import { Staart } from "./server";
+import { Staart } from "./app";
 import { PORT, SENTRY_DSN } from "./config";
 import { init } from "@sentry/node";
 

src/server.ts

@@ -6,17 +6,12 @@ import rfs from "rotating-file-stream";
 import responseTime from "response-time";
 import { json, urlencoded } from "body-parser";
 import { Server } from "@overnightjs/core";
-import { UserController } from "./controllers/user";
 import {
   errorHandler,
   trackingHandler,
   rateLimitHandler,
   speedLimitHandler
 } from "./helpers/middleware";
-import { OrganizationController } from "./controllers/organization";
-import { AdminController } from "./controllers/admin";
-import { AuthController } from "./controllers/auth";
-import { MembershipController } from "./controllers/membership";
 import { mkdirSync, existsSync } from "fs";
 import { join } from "path";
 
@@ -48,19 +43,7 @@ export class Staart extends Server {
   }
 
   private setupControllers() {
-    const authController = new AuthController();
-    const userController = new UserController();
-    const organizationController = new OrganizationController();
-    const membershipController = new MembershipController();
-    const adminController = new AdminController();
-
-    super.addControllers([
-      authController,
-      userController,
-      organizationController,
-      membershipController,
-      adminController
-    ]);
+    // staart:setup/controllers
   }
 
   public start(port: number): void {