@hapico/cli 0.0.18 → 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";
@@ -12,10 +12,10 @@ import * as Babel from "@babel/standalone";
 import QRCode from "qrcode-terminal";
 import open from "open";
 import { randomUUID } from "crypto";
-import inquirer from "inquirer";
 import { exec } from "child_process";
 import { promisify } from "util";
 import chalk from "chalk";
+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,18 +323,19 @@ class RoomState {
       clearTimeout(this.reconnectTimeout);
     }
 
-    this.ws = new WebSocket(
-      `https://ws2.myworkbeast.com/ws?room=${this.roomId}`
-    );
+    this.ws = new WebSocket(`wss://ws3.myworkbeast.com/ws?room=${this.roomId}`);
 
-    this.ws.onopen = () => {
+    // Set binaryType to 'arraybuffer' to handle binary data
+    this.ws.binaryType = "arraybuffer";
+
+    this.ws.on("open", () => {
       console.log(chalk.green(`Connected to room: ${this.roomId}`));
       this.isConnected = true;
       this.reconnectAttempts = 0;
       onConnected?.(); // Call the onConnected callback if provided
-    };
+    });
 
-    this.ws.onclose = () => {
+    this.ws.on("close", () => {
       console.log(chalk.yellow(`⚠ Disconnected from room: ${this.roomId}`));
       this.isConnected = false;
 
@@ -328,11 +349,20 @@ class RoomState {
         () => this.connect(onConnected),
         delay
       );
-    };
+    });
 
-    this.ws.on("message", (data: Buffer | string) => {
+    this.ws.on("message", (data: any) => {
       try {
-        const jsonStr = typeof data === "string" ? data : data.toString();
+        let jsonStr: string;
+        if (data instanceof ArrayBuffer) {
+          jsonStr = pako.inflate(data, { to: "string" });
+        } else if (typeof data === "string") {
+          jsonStr = data;
+        } else if (Buffer.isBuffer(data)) {
+          jsonStr = pako.inflate(data, { to: "string" });
+        } else {
+          jsonStr = data.toString(); // Fallback nếu không nén
+        }
         const parsedData = JSON.parse(jsonStr);
         const includes = [
           "view",
@@ -389,12 +419,12 @@ class RoomState {
 
   updateState(key: string, value: any): void {
     if (this.ws && this.ws.readyState === WebSocket.OPEN) {
-      this.ws.send(
-        JSON.stringify({
-          type: "update",
-          state: { [key]: value },
-        })
-      );
+      const message = JSON.stringify({
+        type: "update",
+        state: { [key]: value },
+      });
+      const compressed = pako.deflate(message); // Nén dữ liệu
+      this.ws.send(compressed); // Gửi binary
     }
   }
 
@@ -403,7 +433,7 @@ class RoomState {
       clearTimeout(this.reconnectTimeout);
     }
     if (this.ws) {
-      this.ws.onclose = null; // Prevent reconnect on intentional close
+      this.ws.removeAllListeners("close"); // Prevent reconnect on intentional close
       this.ws.close();
     }
   }
@@ -417,7 +447,7 @@ class RoomState {
   }
 }
 
-program.version("0.0.18").description("Hapico CLI for project management");
+program.version("0.0.20").description("Hapico CLI for project management");
 
 program
   .command("clone <id>")
@@ -443,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(
@@ -566,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
@@ -584,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) ?? "";
@@ -607,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."
@@ -617,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}`,
@@ -637,7 +707,7 @@ program
 
       if (projectType === "zalominiapp") {
         QRCode.generate(
-          `https://zalo.me/s/3218692650896662017/player/${projectId}`,
+          `https://zalo.me/s/3218692650896662017/player/${projectId}?env=TESTING&version=75`,
           { small: true },
           (qrcode) => {
             console.log(
@@ -659,23 +729,95 @@ program
     });
   });
 
+// Hàm tái sử dụng để push mã nguồn lên server
+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 srcDir: string = path.join(pwd, "src");
+  if (!fs.existsSync(srcDir)) {
+    spinner.fail(
+      chalk.red(
+        "✗ Source directory 'src' does not exist. Please clone a project first."
+      )
+    );
+    return false;
+  }
+  const fileManager = new FileManager(srcDir);
+  const files = fileManager.listFiles();
+
+  // 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(
+      apiUrl,
+      {
+        code: JSON.stringify({
+          files,
+        }),
+      },
+      {
+        headers: {
+          Authorization: `Bearer ${accessToken}`,
+          "Content-Type": "application/json",
+        },
+      }
+    );
+    return true;
+  } catch (error) {
+    spinner.fail(
+      chalk.red(
+        `✗ Error saving project ${projectId}: ${(error as Error).message}`
+      )
+    );
+    return false;
+  }
+}
+
 program
   .command("push")
   .description("Push the project source code to the server")
-  .action(() => {
-    const data = getStoredToken();
-    const { accessToken } = data || {};
-    if (!accessToken) {
-      console.error(
-        chalk.red("✗ You need to login first. Use 'hapico login' command.")
-      );
-      return;
-    }
+  .action(async () => {
     const saveSpinner: Ora = ora(
       chalk.blue("Saving project source code...")
     ).start();
     const pwd = process.cwd();
-    const projectId = getStoredProjectId(pwd);
+    const { projectId, replicate } = getStoredProjectId(pwd);
+
     if (!projectId) {
       saveSpinner.fail(
         chalk.red(
@@ -684,53 +826,40 @@ program
       );
       return;
     }
-    const srcDir: string = path.join(pwd, "src");
-    if (!fs.existsSync(srcDir)) {
-      saveSpinner.fail(
-        chalk.red(
-          "✗ Source directory 'src' does not exist. Please clone a project first."
-        )
-      );
-      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}.`)
+          );
+        }
+      }
     }
-    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,
-      });
+
+    if (allSuccess) {
+      saveSpinner.succeed(
+        chalk.green("Project source code saved successfully to all projects!")
+      );
+    } else {
+      saveSpinner.warn(
+        chalk.yellow("Project source code saved with some errors.")
+      );
     }
-    const apiUrl = `https://base.myworkbeast.com/api/views/${projectId}`;
-    axios
-      .put(
-        apiUrl,
-        {
-          code: JSON.stringify({
-            files,
-          }),
-        },
-        {
-          headers: {
-            Authorization: `Bearer ${accessToken}`,
-            "Content-Type": "application/json",
-          },
-        }
-      )
-      .then(() => {
-        saveSpinner.succeed(
-          chalk.green("Project source code saved successfully!")
-        );
-      })
-      .catch((error) => {
-        saveSpinner.fail(chalk.red(`✗ Error saving project: ${error.message}`));
-      });
   });
 
 program
@@ -821,7 +950,7 @@ program
       return;
     }
     const pwd: string = process.cwd();
-    const projectId = getStoredProjectId(pwd);
+    const { projectId } = getStoredProjectId(pwd);
     if (!projectId) {
       console.error(
         chalk.red(
@@ -835,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!"));
@@ -982,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!"));
@@ -994,7 +1132,6 @@ program
       ).start();
 
       try {
-        files = response.data.files || [];
         apiSpinner.succeed(chalk.green("Project data fetched successfully!"));
 
         const saveSpinner: Ora = ora(
@@ -1213,4 +1350,142 @@ program
     }
   );
 
+// hapico publish
+program.command("publish").action(async () => {
+  const publishSpinner: Ora = ora(
+    chalk.blue("Publishing to Hapico...")
+  ).start();
+  const pwd = process.cwd();
+  const { projectId, replicate } = getStoredProjectId(pwd);
+
+  if (!projectId) {
+    publishSpinner.fail(
+      chalk.red(
+        "✗ Project ID not found. Please ensure hapico.config.json exists in the project directory."
+      )
+    );
+    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) },
+      {
+        headers: {
+          Authorization: `Bearer ${accessToken}`,
+          "Content-Type": "application/json",
+        },
+      }
+    );
+    publishSpinner.succeed(
+      chalk.green(`Project ${projectId} published successfully!`)
+    );
+  } catch (error) {
+    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);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hapico/cli",
-  "version": "0.0.18",
+  "version": "0.0.20",
   "description": "A simple CLI tool for project management",
   "main": "index.js",
   "bin": {
@@ -27,6 +27,7 @@
     "lodash": "^4.17.21",
     "open": "^10.2.0",
     "ora": "^8.2.0",
+    "pako": "^2.1.0",
     "prettier": "^3.6.2",
     "qrcode-terminal": "^0.12.0",
     "unzipper": "^0.12.3",
@@ -38,6 +39,7 @@
     "@types/commander": "^2.12.5",
     "@types/lodash": "^4.17.20",
     "@types/node": "^24.1.0",
+    "@types/pako": "^2.0.4",
     "@types/qrcode-terminal": "^0.12.2",
     "@types/unzipper": "^0.10.11",
     "@types/ws": "^8.18.1",