@hapico/cli 0.0.19 → 0.0.20

package/index.ts

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 /* eslint-disable @typescript-eslint/no-explicit-any */
-import { isEqual, find } from "lodash";
+import { isEqual, find, map } from "lodash";
 import { program } from "commander";
 import axios from "axios";
 import * as fs from "fs";
@@ -15,7 +15,7 @@ import { randomUUID } from "crypto";
 import { exec } from "child_process";
 import { promisify } from "util";
 import chalk from "chalk";
-import pako from 'pako';
+import pako from "pako";
 
 // Promisify exec for async usage
 const execPromise = promisify(exec);
@@ -74,14 +74,25 @@ const saveProjectId = (projectDir: string, id: string) => {
 };
 
 // Function to get stored project ID
-const getStoredProjectId = (projectDir: string): string | null => {
+const getStoredProjectId = (
+  projectDir: string
+): {
+  projectId: string | null;
+  replicate?: string[];
+} => {
   const configFile = path.join(projectDir, "hapico.config.json");
   if (fs.existsSync(configFile)) {
     const data = fs.readFileSync(configFile, { encoding: "utf8" });
     const json = JSON.parse(data);
-    return json.projectId || null;
+    return {
+      projectId: json.projectId || null,
+      replicate: json.replicate || [],
+    };
   }
-  return null;
+  return {
+    projectId: null,
+    replicate: [],
+  };
 };
 
 // Khởi tạo cache bằng Map để lưu trữ kết quả compile
@@ -95,7 +106,7 @@ interface FileContent {
 
 interface ApiResponse {
   data: {
-    files?: FileContent[];
+    code: string;
     type: "view" | "zalominiapp" | string;
     dbCode: string;
     db_code: string;
@@ -111,6 +122,14 @@ interface WebSocketMessage {
   state?: RoomStateData;
 }
 
+export const tryJSONParse = (str: string): any => {
+  try {
+    return JSON.parse(str);
+  } catch {
+    return null;
+  }
+};
+
 export const compileES5 = (code: string, filePath?: string) => {
   if (filePath && !filePath.endsWith(".ts") && !filePath.endsWith(".tsx")) {
     return code;
@@ -131,7 +150,8 @@ export const compileES5 = (code: string, filePath?: string) => {
 
     compileCache.set(code, result.code || "");
     return result.code;
-  } catch (error) {
+  } catch (error: any) {
+    console.log(chalk.red(`Error compiling code: ${error?.message}`));
     return "";
   }
 };
@@ -303,12 +323,10 @@ class RoomState {
       clearTimeout(this.reconnectTimeout);
     }
 
-    this.ws = new WebSocket(
-      `wss://ws3.myworkbeast.com/ws?room=${this.roomId}`
-    );
+    this.ws = new WebSocket(`wss://ws3.myworkbeast.com/ws?room=${this.roomId}`);
 
     // Set binaryType to 'arraybuffer' to handle binary data
-    this.ws.binaryType = 'arraybuffer';
+    this.ws.binaryType = "arraybuffer";
 
     this.ws.on("open", () => {
       console.log(chalk.green(`Connected to room: ${this.roomId}`));
@@ -337,11 +355,11 @@ class RoomState {
       try {
         let jsonStr: string;
         if (data instanceof ArrayBuffer) {
-          jsonStr = pako.inflate(data, { to: 'string' });
+          jsonStr = pako.inflate(data, { to: "string" });
         } else if (typeof data === "string") {
           jsonStr = data;
         } else if (Buffer.isBuffer(data)) {
-          jsonStr = pako.inflate(data, { to: 'string' });
+          jsonStr = pako.inflate(data, { to: "string" });
         } else {
           jsonStr = data.toString(); // Fallback nếu không nén
         }
@@ -429,7 +447,7 @@ class RoomState {
   }
 }
 
-program.version("0.0.19").description("Hapico CLI for project management");
+program.version("0.0.20").description("Hapico CLI for project management");
 
 program
   .command("clone <id>")
@@ -455,7 +473,12 @@ program
       const response: ApiResponse = await axios.get(
         `https://base.myworkbeast.com/api/views/${id}`
       );
-      files = response.data.files || [];
+      const code = response?.data?.code;
+      const decompressedCode = pako.inflate(
+        Uint8Array.from(atob(code), (c) => c.charCodeAt(0)),  
+        { to: "string" }
+      );
+      files = tryJSONParse(decompressedCode)?.files || [];
       apiSpinner.succeed(chalk.green("Project data fetched successfully!"));
 
       const templateSpinner: Ora = ora(
@@ -578,7 +601,7 @@ program
     const info = JSON.stringify({
       id: sessionId,
       createdAt: new Date().toISOString(),
-      viewId: getStoredProjectId(pwd),
+      viewId: getStoredProjectId(pwd)?.projectId,
     });
 
     // Convert info to base64
@@ -596,12 +619,47 @@ program
     const room = new RoomState(`view_${projectId}`, []);
     const fileManager = new FileManager(srcDir);
     const initialFiles = fileManager.listFiles();
-    room.files = initialFiles;
+    // get tsconfig.json
+    const tsconfigPath = path.join(srcDir, "..", "tsconfig.json");
+    if (fs.existsSync(tsconfigPath)) {
+      const content = fs.readFileSync(tsconfigPath, { encoding: "utf8" });
+      initialFiles.push({
+        path: "./tsconfig.json",
+        content,
+      });
+    }
+    // Remove All binary files
+    const supportExtensions = [
+      ".ts",
+      ".tsx",
+      ".js",
+      ".jsx",
+      ".json",
+      ".css",
+      ".html",
+      ".env",
+      ".env.local",
+      ".env.development",
+      ".env.production",
+      ".jsonc",
+      ".yml",
+      ".yaml",
+      ".md",
+      ".markdown",
+      ".txt",
+      ".xml",
+      ".config",
+    ];
+    const filteredFiles = initialFiles.filter((file) => {
+      return supportExtensions.some((ext) => file.path.endsWith(ext));
+    });
+
+    room.files = filteredFiles;
 
     room.connect(async () => {
       devSpinner.succeed(chalk.green("Project started in development mode!"));
 
-      room.updateState("view", initialFiles);
+      room.updateState("view", filteredFiles);
 
       fileManager.setOnFileChange((filePath, content) => {
         const es5 = compileES5(content, filePath) ?? "";
@@ -619,8 +677,8 @@ program
       });
 
       // Fetch project info
-      const projectInfo = getStoredProjectId(pwd);
-      if (!projectInfo) {
+      const store = getStoredProjectId(pwd);
+      if (!store.projectId) {
         console.error(
           chalk.red(
             "✗ Project ID not found. Please ensure hapico.config.json exists in the project directory."
@@ -629,7 +687,7 @@ program
         return;
       }
       const project = await axios.get(
-        `https://base.myworkbeast.com/api/views/${projectInfo}`,
+        `https://base.myworkbeast.com/api/views/${store.projectId}`,
         {
           headers: {
             Authorization: `Bearer ${accessToken}`,
@@ -672,23 +730,16 @@ program
   });
 
 // Hàm tái sử dụng để push mã nguồn lên server
-async function pushProject(spinner: Ora): Promise<boolean> {
+async function pushProject(spinner: Ora, projectId: string): Promise<boolean> {
   const data = getStoredToken();
   const { accessToken } = data || {};
   if (!accessToken) {
-    spinner.fail(chalk.red("✗ You need to login first. Use 'hapico login' command."));
-    return false;
-  }
-  const pwd = process.cwd();
-  const projectId = getStoredProjectId(pwd);
-  if (!projectId) {
     spinner.fail(
-      chalk.red(
-        "✗ Project ID not found. Please ensure hapico.config.json exists in the project directory."
-      )
+      chalk.red("✗ You need to login first. Use 'hapico login' command.")
     );
     return false;
   }
+  const pwd = process.cwd();
   const srcDir: string = path.join(pwd, "src");
   if (!fs.existsSync(srcDir)) {
     spinner.fail(
@@ -701,17 +752,35 @@ async function pushProject(spinner: Ora): Promise<boolean> {
   const fileManager = new FileManager(srcDir);
   const files = fileManager.listFiles();
 
-  // Nếu có file .env
-  const envFile = path.join(pwd, ".env");
-  console.log(chalk.cyan(`🔍 Checking for .env file at ${envFile}`));
-  if (fs.existsSync(envFile)) {
-    console.log(chalk.green(".env file found, including in push."));
-    const content = fs.readFileSync(envFile, { encoding: "utf8" });
-    files.push({
-      path: "./.env",
-      content,
-    });
-  }
+  // Supported files
+  const SUPPORT_FILES = [
+    "./.env",
+    "./.env.local",
+    "./.env.development",
+    "./.env.production",
+    "./package.json",
+    "./tsconfig.json",
+  ];
+
+  // Include supported files
+  SUPPORT_FILES.forEach((relativePath) => {
+    const fullPath = path.join(pwd, relativePath);
+    if (fs.existsSync(fullPath)) {
+      console.log(
+        chalk.green(
+          `Including ${relativePath} in push for project ${projectId}.`
+        )
+      );
+      const content = fs.readFileSync(fullPath, { encoding: "utf8" });
+      files.push({
+        path: relativePath,
+        content,
+        es5: compileES5(content, fullPath) ?? "",
+      });
+    }
+  });
+  spinner.text = `Pushing project source code to server for project ${projectId}...`;
+
   const apiUrl = `https://base.myworkbeast.com/api/views/${projectId}`;
   try {
     await axios.put(
@@ -730,7 +799,11 @@ async function pushProject(spinner: Ora): Promise<boolean> {
     );
     return true;
   } catch (error) {
-    spinner.fail(chalk.red(`✗ Error saving project: ${(error as Error).message}`));
+    spinner.fail(
+      chalk.red(
+        `✗ Error saving project ${projectId}: ${(error as Error).message}`
+      )
+    );
     return false;
   }
 }
@@ -742,10 +815,49 @@ program
     const saveSpinner: Ora = ora(
       chalk.blue("Saving project source code...")
     ).start();
-    const success = await pushProject(saveSpinner);
-    if (success) {
+    const pwd = process.cwd();
+    const { projectId, replicate } = getStoredProjectId(pwd);
+
+    if (!projectId) {
+      saveSpinner.fail(
+        chalk.red(
+          "✗ Project ID not found. Please ensure hapico.config.json exists in the project directory."
+        )
+      );
+      return;
+    }
+
+    // Push to the main project
+    const mainSuccess = await pushProject(saveSpinner, projectId);
+    let allSuccess = mainSuccess;
+
+    // Push to replicated projects if replicate array exists
+    if (replicate && Array.isArray(replicate) && replicate.length > 0) {
+      saveSpinner.text = chalk.blue("Pushing to replicated projects...");
+      for (const repId of replicate) {
+        const success = await pushProject(saveSpinner, repId);
+        if (!success) {
+          allSuccess = false;
+          console.warn(
+            chalk.yellow(
+              `⚠ Failed to push to replicated project ${repId}. Continuing...`
+            )
+          );
+        } else {
+          console.log(
+            chalk.green(`✓ Successfully pushed to replicated project ${repId}.`)
+          );
+        }
+      }
+    }
+
+    if (allSuccess) {
       saveSpinner.succeed(
-        chalk.green("Project source code saved successfully!")
+        chalk.green("Project source code saved successfully to all projects!")
+      );
+    } else {
+      saveSpinner.warn(
+        chalk.yellow("Project source code saved with some errors.")
       );
     }
   });
@@ -838,7 +950,7 @@ program
       return;
     }
     const pwd: string = process.cwd();
-    const projectId = getStoredProjectId(pwd);
+    const { projectId } = getStoredProjectId(pwd);
     if (!projectId) {
       console.error(
         chalk.red(
@@ -852,15 +964,22 @@ program
     ).start();
     try {
       const response: ApiResponse = await axios.get(
-        `https://base.myworkbeast.com/api/views/${projectId}`,
+        `https://base.myworkbeast.com/api/views/v3/${projectId}`,
         {
           headers: {
-            Authorization: `Bearer ${token}`,
+            Authorization: `Bearer ${token.accessToken}`,
             "Content-Type": "application/json",
           },
         }
       );
-      const files: FileContent[] = response.data.files || [];
+
+      const code = response?.data?.code;
+      const decompressedCode = pako.inflate(
+        Uint8Array.from(atob(code), (c) => c.charCodeAt(0)),
+        { to: "string" }
+      );
+
+      const files: FileContent[] = tryJSONParse(decompressedCode)?.files || [];
       const fileManager = new FileManager(path.join(pwd, "src"));
       fileManager.syncFiles(files);
       apiSpinner.succeed(chalk.green("Project files updated successfully!"));
@@ -999,7 +1118,9 @@ program
           migrationsDir,
           `${index + 1}_migration_${timestamp}.sql`
         );
-        fs.writeFileSync(filename, (migration as any).sql, { encoding: "utf8" });
+        fs.writeFileSync(filename, (migration as any).sql, {
+          encoding: "utf8",
+        });
       });
 
       console.log(chalk.green("✓ Migrations saved successfully!"));
@@ -1011,7 +1132,6 @@ program
       ).start();
 
       try {
-        files = response.data.files || [];
         apiSpinner.succeed(chalk.green("Project data fetched successfully!"));
 
         const saveSpinner: Ora = ora(
@@ -1232,17 +1352,12 @@ program
 
 // hapico publish
 program.command("publish").action(async () => {
-  const publishSpinner: Ora = ora(chalk.blue("Publishing to Hapico...")).start();
-  // Bước 1: Push mã nguồn lên server (tái sử dụng hàm pushProject)
-  const pushSuccess = await pushProject(publishSpinner);
-  if (!pushSuccess) {
-    return; // Dừng nếu push thất bại
-  }
-
-  // Bước 2: Gọi API để publish
-  const { accessToken } = getStoredToken();
+  const publishSpinner: Ora = ora(
+    chalk.blue("Publishing to Hapico...")
+  ).start();
   const pwd = process.cwd();
-  const projectId = getStoredProjectId(pwd);
+  const { projectId, replicate } = getStoredProjectId(pwd);
+
   if (!projectId) {
     publishSpinner.fail(
       chalk.red(
@@ -1251,11 +1366,42 @@ program.command("publish").action(async () => {
     );
     return;
   }
+
+  // Step 1: Push source code to main project and replicas
+  const pushSuccess = await pushProject(publishSpinner, projectId);
+  if (!pushSuccess) {
+    return; // Stop if push to main project fails
+  }
+
+  // Push to replicated projects
+  let allPushSuccess = true;
+  if (replicate && Array.isArray(replicate) && replicate.length > 0) {
+    publishSpinner.text = chalk.blue("Pushing to replicated projects...");
+    for (const repId of replicate) {
+      const success = await pushProject(publishSpinner, repId);
+      if (!success) {
+        allPushSuccess = false;
+        console.warn(
+          chalk.yellow(
+            `⚠ Failed to push to replicated project ${repId}. Continuing...`
+          )
+        );
+      } else {
+        console.log(
+          chalk.green(`✓ Successfully pushed to replicated project ${repId}.`)
+        );
+      }
+    }
+  }
+
+  // Step 2: Publish main project
+  const { accessToken } = getStoredToken();
   const apiUrl = "https://base.myworkbeast.com/api/views/publish";
+  let allPublishSuccess = true;
   try {
     await axios.post(
       apiUrl,
-      { view_id: parseInt(projectId, 10) }, // Đảm bảo viewId là number
+      { view_id: parseInt(projectId, 10) },
       {
         headers: {
           Authorization: `Bearer ${accessToken}`,
@@ -1263,11 +1409,83 @@ program.command("publish").action(async () => {
         },
       }
     );
-    publishSpinner.succeed(chalk.green("Project published successfully!"));
+    publishSpinner.succeed(
+      chalk.green(`Project ${projectId} published successfully!`)
+    );
   } catch (error) {
-    console.log(error);
-    publishSpinner.fail(chalk.red(`✗ Error publishing project: ${(error as Error).message}`));
+    allPublishSuccess = false;
+    publishSpinner.fail(
+      chalk.red(
+        `✗ Error publishing project ${projectId}: ${(error as Error).message}`
+      )
+    );
+  }
+
+  // Step 3: Publish replicated projects
+  if (replicate && Array.isArray(replicate) && replicate.length > 0) {
+    publishSpinner.text = chalk.blue("Publishing replicated projects...");
+    for (const repId of replicate) {
+      try {
+        await axios.post(
+          apiUrl,
+          { view_id: parseInt(repId, 10) },
+          {
+            headers: {
+              Authorization: `Bearer ${accessToken}`,
+              "Content-Type": "application/json",
+            },
+          }
+        );
+        console.log(
+          chalk.green(`✓ Successfully published replicated project ${repId}.`)
+        );
+      } catch (error) {
+        allPublishSuccess = false;
+        console.warn(
+          chalk.yellow(
+            `⚠ Error publishing replicated project ${repId}: ${(error as Error).message}`
+          )
+        );
+      }
+    }
+  }
+
+  if (allPushSuccess && allPublishSuccess) {
+    publishSpinner.succeed(chalk.green("All projects published successfully!"));
+  } else {
+    publishSpinner.warn(chalk.yellow("Publishing completed with some errors."));
+  }
+});
+
+program.command("mirror").action(() => {
+  console.log(chalk.cyan("🌐 Starting mirror mode..."));
+  const pwd = process.cwd();
+  const srcDir = path.join(pwd, "src");
+  if (!fs.existsSync(srcDir)) {
+    console.error(
+      chalk.red(
+        "✗ Source directory 'src' does not exist. Please clone a project first."
+      )
+    );
+    return;
   }
+
+  const fileManager = new FileManager(srcDir);
+  const initialFiles = fileManager.listFiles();
+
+  // Lấy danh sách file và viết ra 1 file .txt
+  let content = ``;
+  map(initialFiles, (file) => {
+    content += `\`\`\`typescript
+// Path: ${file.path}
+${file.content}
+
+\`\`\`
+`;
+  });
+  const outputFile = path.join(pwd, "hapico_files.txt");
+  fs.writeFileSync(outputFile, content, { encoding: "utf8" });
+  console.log(chalk.green(`✓ File list saved to ${outputFile}`));
 });
 
-program.parse(process.argv);
+program.parse(process.argv);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hapico/cli",
-  "version": "0.0.19",
+  "version": "0.0.20",
   "description": "A simple CLI tool for project management",
   "main": "index.js",
   "bin": {