✨ Add project name customization to CLI

aryanprince · Sep 19, 2024 · b21aecc · b21aecc
1 parent d6caab2
commit b21aecc
Show file tree

Hide file tree

Showing 2 changed files with 85 additions and 61 deletions.
diff --git a/.prettierrc b/.prettierrc
diff --git a/src/index.ts b/src/index.ts
@@ -10,90 +10,114 @@ program
   .description(
     "A CLI to easily create DBs for local development using Docker Compose",
   )
-  .version("0.0.1")
+  .version("0.0.2")
   .action(async () => {
+    const currentDir = path.basename(process.cwd());
+
     const answers = await inquirer.prompt([
       {
         type: "list",
         name: "dbType",
         message: "Which database would you like to use?",
         choices: ["PostgreSQL", "MySQL"],
       },
+      {
+        type: "input",
+        name: "projectName",
+        message: "Enter your project name:",
+        default: currentDir,
+      },
     ]);
 
-    const dockerComposeContent = generateDockerCompose(answers.dbType);
+    const dockerComposeContent = generateDockerCompose(
+      answers.dbType,
+      answers.projectName,
+    );
 
     fs.writeFileSync(
       path.join(process.cwd(), "docker-compose.yml"),
       dockerComposeContent,
     );
+
     console.log("Docker Compose file created successfully!");
+
+    const connectionUrl = getConnectionUrl(answers.dbType, answers.projectName);
+    console.log("\nConnection URL for your database:");
+    console.log(connectionUrl);
   });
 
-function generateDockerCompose(dbType: string): string {
+function generateDockerCompose(dbType: string, projectName: string): string {
   if (dbType === "MySQL") {
     return `
-version: "3.9"
-
-name: myprojectname-mysql
-
-services:
-  # This is your local MySQL database instance
-  mysql-db:
-    image: mysql
-    restart: always
-    environment:
-      MYSQL_DATABASE: myprojectname
-      MYSQL_ROOT_PASSWORD: root
-      MYSQL_USER: dev
-      MYSQL_PASSWORD: dev
-    volumes:
-      - myprojectname-data:/var/lib/mysql
-    ports:
-      - "6969:3306" # Access the DB at port 6969
-
-  # Use Adminer to quickly view the database at localhost:8069
-  adminer:
-    image: adminer
-    restart: always
-    ports:
-      - "8089:8080"
-
-volumes:
-  myprojectname-data:
-    driver: local
-`;
+  version: "3.9"
+  
+  name: ${projectName}-mysql
+  
+  services:
+    # This is your local MySQL database instance
+    mysql-db:
+      image: mysql
+      restart: always
+      environment:
+        MYSQL_DATABASE: ${projectName}
+        MYSQL_ROOT_PASSWORD: root
+        MYSQL_USER: dev
+        MYSQL_PASSWORD: dev
+      volumes:
+        - ${projectName}-data:/var/lib/mysql
+      ports:
+        - "6969:3306" # Access the DB at port 6969
+  
+    # Use Adminer to quickly view the database at localhost:8089
+    adminer:
+      image: adminer
+      restart: always
+      ports:
+        - "8089:8080"
+  
+  volumes:
+    ${projectName}-data:
+      driver: local
+  `;
   } else {
     return `
-version: "3.9"
-
-name: myprojectname-postgres
-
-services:
-  # This is your local Postgres database instance
-  postgres-db:
-    image: postgres
-    restart: always
-    environment:
-      POSTGRES_DB: myprojectname
-      POSTGRES_USER: dev
-      POSTGRES_PASSWORD: dev
-    volumes:
-      - myprojectname-data:/var/lib/postgresql/data
-    ports:
-      - "6969:5432" # Access the DB at port 6969
-
-  # Use Adminer to quickly view the database at localhost:8069
-  adminer:
-    image: adminer
-    restart: always
-    ports:
-      - "8069:8080"
+  version: "3.9"
+  
+  name: ${projectName}-postgres
+  
+  services:
+    # This is your local Postgres database instance
+    postgres-db:
+      image: postgres
+      restart: always
+      environment:
+        POSTGRES_DB: ${projectName}
+        POSTGRES_USER: dev
+        POSTGRES_PASSWORD: dev
+      volumes:
+        - ${projectName}-data:/var/lib/postgresql/data
+      ports:
+        - "6969:5432" # Access the DB at port 6969
+  
+    # Use Adminer to quickly view the database at localhost:8069
+    adminer:
+      image: adminer
+      restart: always
+      ports:
+        - "8069:8080"
+  
+  volumes:
+    ${projectName}-data:
+      driver: local
+  `;
+  }
+}
 
-volumes:
-  myprojectname-data:
-    driver: local
-`;
+function getConnectionUrl(dbType: string, projectName: string): string {
+  if (dbType === "MySQL") {
+    return `mysql://dev:dev@localhost:6969/${projectName}`;
+  } else {
+    return `postgresql://dev:dev@localhost:6969/${projectName}`;
   }
 }