mstate-cli 0.2.9 → 0.4.0

package/src/handlers/company.handler.ts

@@ -1,274 +1,202 @@
-import { MSTATE_DOMAIN, MSTATE_URL } from '../common/constant';
-import {
-  customLog,
-  getSecretKey,
-  handleUpdateSecretKey,
-} from '../common/utils';
-import { getValueFromArgs } from '../common/helpers';
-
-import open from 'open';
-import express from 'express';
-import cors from 'cors';
-
-enum Key {
-  PERMISSIONS = 'permissions=',
-  SECRET_KEY = 'secret=',
-  USER_ID = 'user=',
-  NAME = 'name=',
-  EMAIL = 'email=',
-  PHONE = 'phone=',
-  COMPANY_ID = 'company=',
-}
-
-class CompanyHandler {
-  constructor() {}
-
-  async addToken() {
-    const args = process.argv;
-
-    const pString = getValueFromArgs(args, Key.PERMISSIONS).trim();
-    const permissions = pString.split(',').map((s) => s.trim());
-    const secretKey = getSecretKey();
-    const user = getValueFromArgs(args, Key.USER_ID);
-    const companyID = getValueFromArgs(args, Key.COMPANY_ID);
-
-    if (!permissions.length && permissions[0]?.length) {
-      customLog.changeAndThrow(`Parameter permissions is required`);
-      return;
-    }
-
-    if (!secretKey) {
-      customLog.changeAndThrow(`Parameter secret is required`);
-      return;
-    }
-
-    if (!user) {
-      customLog.changeAndThrow(`Parameter user is required`);
-      return;
-    }
-
-    if (!companyID) {
-      customLog.changeAndThrow(`Parameter company is required`);
-      return;
-    }
-
-    const url = `${MSTATE_URL}/company/permission`;
-    const responseJSON = await fetch(url, {
-      method: 'POST',
-      headers: {
-        'secret-key': secretKey,
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({
-        user,
-        permissions,
-        company: companyID,
-      }),
-    });
-
-    const response = await responseJSON.json();
-    if (response?.errors) {
-      customLog.changeAndThrow(
-        'Invalid Parameters for Company: ',
-        ...response?.errors,
-      );
-    } else {
-      customLog.success('Company updated successfully \n', response?.data);
-    }
-  }
-
-  async addUser() {
-    const args = process.argv;
-
-    const secretKey = getSecretKey();
-    const name = getValueFromArgs(args, Key.NAME);
-    const email = getValueFromArgs(args, Key.EMAIL);
-    const phone = getValueFromArgs(args, Key.PHONE);
-    const companyID = getValueFromArgs(args, Key.COMPANY_ID);
-
-    if (!secretKey) {
-      customLog.changeAndThrow(`Parameter secret is required`);
-      return;
-    }
-
-    if (!name) {
-      customLog.changeAndThrow(`Parameter name is required`);
-      return;
-    }
-
-    if (!email) {
-      customLog.changeAndThrow(`Parameter email is required`);
-      return;
-    }
-
-    if (!phone) {
-      customLog.changeAndThrow(`Parameter phone is required`);
-      return;
-    }
-
-    if (!companyID) {
-      customLog.changeAndThrow(`Parameter company is required`);
-      return;
-    }
-
-    const url = `${MSTATE_URL}/um/user`;
-    const responseJSON = await fetch(url, {
-      method: 'POST',
-      headers: {
-        'secret-key': secretKey,
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({
-        companyNickName: companyID,
-        name,
-        username: name,
-        email,
-        password: '123',
-        phone,
-      }),
-    });
-
-    const response = await responseJSON.json();
-    if (response?.errors) {
-      customLog.changeAndThrow(
-        'Invalid Parameters for Company: ',
-        ...response?.errors,
-      );
-    } else {
-      customLog.success('User Updated successfully \n', response?.data);
-    }
-  }
-
-  async revokePermission() {
-    const args = process.argv;
-
-    const pString = getValueFromArgs(args, Key.PERMISSIONS);
-    const permissions = pString.split(',').map((s) => s.trim());
-    const secretKey = getSecretKey();
-    const user = getValueFromArgs(args, Key.USER_ID);
-    const companyID = getValueFromArgs(args, Key.COMPANY_ID);
-
-    if (!permissions.length && permissions[0]?.length) {
-      customLog.changeAndThrow(`Parameter permissions is required`);
-      return;
-    }
-
-    if (!secretKey) {
-      customLog.changeAndThrow(`Parameter secret is required`);
-      return;
-    }
-
-    if (!user) {
-      customLog.changeAndThrow(`Parameter user is required`);
-      return;
-    }
-
-    if (!companyID) {
-      customLog.changeAndThrow(`Parameter company is required`);
-      return;
-    }
-
-    const url = `${MSTATE_URL}/company/permission`;
-    const responseJSON = await fetch(url, {
-      method: 'DELETE',
-      headers: {
-        'secret-key': secretKey,
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify({
-        company: companyID,
-        user,
-        permissions,
-      }),
-    });
-
-    const response = await responseJSON.json();
-    if (response?.errors) {
-      customLog.changeAndThrow(
-        'Invalid Parameters for Company: ',
-        ...response?.errors,
-      );
-    } else {
-      customLog.success('Company updated successfully \n', response?.data);
-    }
-  }
-
-  async handleLogin() {
-    // Start the login process
-    let port = 49152;
-    const app = express();
-    app.use(cors({ origin: '*' }));
-
-    async function login() {
-      console.log('Opening browser for authentication...');
-
-      // Start a local server to capture the token
-
-      const startServer = (port: number) => {
-        return app.listen(port, () => {
-          console.log(`Waiting for browser response...`);
-        });
-      };
-
-      const server = startServer(port);
-
-      server.on('error', (err: any) => {
-        if (err.code === 'EADDRINUSE') {
-          console.log(`Port ${port} is in use. Trying another port...`);
-
-          startServer(++port);
-        } else {
-          console.error('Server error:', err);
-        }
-      });
-
-      // Capture the OAuth callback
-      app.get('/cli/mstate/login', (req, res) => {
-        try {
-          const { token } = req.query;
-
-          if (token) {
-            console.log('Authentication successful!');
-
-            // Save the token locally
-            handleUpdateSecretKey(token as string);
-
-            res.send('<h1>Login successful! You can close this window.</h1>');
-          } else {
-            console.error('Authentication failed.');
-            res.send('<h1>Authentication failed. Please try again.</h1>');
-          }
-        } catch (error) {
-        } finally {
-          server.close(); // Stop the server after handling the response
-        }
-      });
-    }
-
-    // Command handling
-    login()
-      .then(async () => {
-        await open(
-          MSTATE_DOMAIN +
-            `/oauth?redirect=${encodeURIComponent(
-              'cli/mstate/login',
-            )}&open=${port}&scope=profile`,
-        );
-      })
-      .catch((err) => {
-        console.error('Error during login:', err.message);
-        process.exit(1);
-      });
-  }
-
-  handleUpdateSavedSecretKey() {
-    handleUpdateSecretKey(getSecretKey());
-  }
-
-  logToken() {
-    customLog.info('Token', getSecretKey());
-    customLog.info('MSTATE_URL', MSTATE_URL);
-  }
-}
-
-export const companyHandler = new CompanyHandler();
+import { MSTATE_DOMAIN, MSTATE_URL } from '../common/constant';
+import {
+  customLog,
+  getSecretKey,
+  handleUpdateSecretKey,
+} from '../common/utils';
+import { getValueFromArgs } from '../common/helpers';
+
+import open from 'open';
+import express from 'express';
+import cors from 'cors';
+import { Variables } from 'src/common/enum';
+
+class CompanyHandler {
+  constructor() {}
+
+  async addToken() {
+    const pString = getValueFromArgs(Variables.PERMISSIONS).trim();
+    const permissions = pString.split(',').map((s) => s.trim());
+    const secretKey = getSecretKey();
+    const user = getValueFromArgs(Variables.USER_ID, true);
+    const companyID = getValueFromArgs(Variables.COMPANY_ID, true);
+
+    if (!permissions.length && permissions[0]?.length) {
+      customLog.changeAndThrow(`Parameter permissions is required`);
+      return;
+    }
+
+    const url = `${MSTATE_URL}/company/permission`;
+    const responseJSON = await fetch(url, {
+      method: 'POST',
+      headers: {
+        'secret-key': secretKey,
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        user,
+        permissions,
+        company: companyID,
+      }),
+    });
+
+    const response = await responseJSON.json();
+    if (response?.errors) {
+      customLog.changeAndThrow(
+        'Invalid Parameters for Company: ',
+        ...response?.errors,
+      );
+    } else {
+      customLog.success('Company updated successfully \n', response?.data);
+    }
+  }
+
+  async addUser() {
+    const secretKey = getSecretKey();
+    const name = getValueFromArgs(Variables.USER_ID, true);
+    const email = getValueFromArgs(Variables.EMAIL, true);
+    const phone = getValueFromArgs(Variables.PHONE, true);
+
+    const url = `${MSTATE_URL}/um/user`;
+    const responseJSON = await fetch(url, {
+      method: 'POST',
+      headers: {
+        'secret-key': secretKey,
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        name,
+        username: name,
+        email,
+        password: '123',
+        phone,
+      }),
+    });
+
+    const response = await responseJSON.json();
+    if (response?.errors) {
+      customLog.changeAndThrow(
+        'Invalid Parameters for Company: ',
+        ...response?.errors,
+      );
+    } else {
+      customLog.success('User Updated successfully \n', response?.data);
+    }
+  }
+
+  async revokePermission() {
+    const pString = getValueFromArgs(Variables.PERMISSIONS);
+    const permissions = pString.split(',').map((s) => s.trim());
+    const secretKey = getSecretKey();
+    const user = getValueFromArgs(Variables.USER_ID, true);
+    const companyID = getValueFromArgs(Variables.COMPANY_ID, true);
+
+    if (!permissions.length && permissions[0]?.length) {
+      customLog.changeAndThrow(`Parameter permissions is required`);
+      return;
+    }
+
+    const url = `${MSTATE_URL}/company/permission`;
+    const responseJSON = await fetch(url, {
+      method: 'DELETE',
+      headers: {
+        'secret-key': secretKey,
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        company: companyID,
+        user,
+        permissions,
+      }),
+    });
+
+    const response = await responseJSON.json();
+    if (response?.errors) {
+      customLog.changeAndThrow(
+        'Invalid Parameters for Company: ',
+        ...response?.errors,
+      );
+    } else {
+      customLog.success('Company updated successfully \n', response?.data);
+    }
+  }
+
+  async handleLogin() {
+    // Start the login process
+    let port = 49152;
+    const app = express();
+    app.use(cors({ origin: '*' }));
+
+    async function login() {
+      console.log('Opening browser for authentication...');
+
+      // Start a local server to capture the token
+
+      const startServer = (port: number) => {
+        return app.listen(port, () => {
+          console.log(`Waiting for browser response...`);
+        });
+      };
+
+      const server = startServer(port);
+
+      server.on('error', (err: any) => {
+        if (err.code === 'EADDRINUSE') {
+          console.log(`Port ${port} is in use. Trying another port...`);
+
+          startServer(++port);
+        } else {
+          console.error('Server error:', err);
+        }
+      });
+
+      // Capture the OAuth callback
+      app.get('/cli/mstate/login', (req, res) => {
+        try {
+          const { token } = req.query;
+
+          if (token) {
+            console.log('Authentication successful!');
+
+            // Save the token locally
+            handleUpdateSecretKey(token as string);
+
+            res.send('<h1>Login successful! You can close this window.</h1>');
+          } else {
+            console.error('Authentication failed.');
+            res.send('<h1>Authentication failed. Please try again.</h1>');
+          }
+        } catch (error) {
+        } finally {
+          server.close(); // Stop the server after handling the response
+        }
+      });
+    }
+
+    // Command handling
+    login()
+      .then(async () => {
+        await open(
+          MSTATE_DOMAIN +
+            `/oauth?redirect=${encodeURIComponent(
+              'cli/mstate/login',
+            )}&open=${port}&scope=profile`,
+        );
+      })
+      .catch((err) => {
+        console.error('Error during login:', err.message);
+        process.exit(1);
+      });
+  }
+
+  handleUpdateSavedSecretKey() {
+    handleUpdateSecretKey(getSecretKey());
+  }
+
+  logToken() {
+    customLog.info('MSTATE_URL', MSTATE_URL);
+    customLog.info('Token', getSecretKey());
+  }
+}
+
+export const companyHandler = new CompanyHandler();

package/src/handlers/environment.handler.ts

@@ -1,111 +1,77 @@
-import * as fs from 'fs';
-import path from 'path';
-import { MSTATE_URL } from '../common/constant';
-import { customLog, getSecretKey } from '../common/utils';
-import { getValueFromArgs } from '../common/helpers';
-
-enum Key {
-  WORKFLOW = 'workflow=',
-  SECRET_KEY = 'secret=',
-  TYPE = 'type=',
-}
-
-class EnvironmentHandler {
-  constructor() {}
-
-  async cloneEnvironments() {
-    const args = process.argv;
-    const workflowName = getValueFromArgs(args, Key.WORKFLOW);
-    const secretKey = getSecretKey();
-    const type = getValueFromArgs(args, Key.TYPE);
-
-    if (!secretKey) {
-      customLog.changeAndThrow(`Parameter secret is required`);
-      return;
-    }
-
-    if (!workflowName) {
-      customLog.changeAndThrow(`Parameter workflow is required`);
-      return;
-    }
-
-    let query = `?workflow=${workflowName}`;
-    if (type) {
-      query += `&type=${type}`;
-    }
-    const url = `${MSTATE_URL}/env/${query}`;
-
-    const responseJSON = await fetch(url, {
-      headers: {
-        'Content-Type': 'application/json',
-        'secret-key': secretKey,
-      },
-    });
-
-    const response = await responseJSON.json();
-    if (response?.errors) {
-      customLog.changeAndThrow(
-        'Invalid Parameters for Environment: ',
-        ...response?.errors,
-      );
-    } else {
-      const config = response?.data;
-
-      const filePath = path.resolve(workflowName, 'environment.json');
-      fs.writeFileSync(filePath, JSON.stringify(config, null, 2));
-    }
-  }
-
-  async saveToDB() {
-    const args = process.argv;
-
-    const workflowName = getValueFromArgs(args, Key.WORKFLOW);
-    const secretKey = getSecretKey();
-    const type = getValueFromArgs(args, Key.TYPE) || undefined;
-
-    if (!secretKey) {
-      customLog.changeAndThrow(`Parameter secret is required`);
-      return;
-    }
-
-    if (!workflowName) {
-      customLog.changeAndThrow(`Parameter workflow is required`);
-      return;
-    }
-
-    const filePath = path.resolve(workflowName, 'environment.json');
-
-    const data = fs.readFileSync(filePath, 'utf8');
-    const environments = JSON.parse(data);
-
-    try {
-      const url = `${MSTATE_URL}/env`;
-      const responseJSON = await fetch(url, {
-        method: 'POST',
-        headers: {
-          'secret-key': secretKey,
-          'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({
-          workflow: workflowName,
-          env: environments,
-          type,
-        }),
-      });
-
-      const response = await responseJSON.json();
-      if (response?.errors) {
-        customLog.changeAndThrow(
-          `Invalid Parameters for Environment: `,
-          ...response?.errors,
-        );
-      } else {
-        customLog.success(`Environments Updated successfully`);
-      }
-    } catch (error: any) {
-      customLog.changeAndThrow(error.message);
-    }
-  }
-}
-
-export const environmentHandler = new EnvironmentHandler();
+import * as fs from 'fs';
+import path from 'path';
+import { MSTATE_URL } from '../common/constant';
+import { customLog, getSecretKey } from '../common/utils';
+import { getValueFromArgs } from '../common/helpers';
+import { Variables } from 'src/common/enum';
+
+class EnvironmentHandler {
+  constructor() {}
+
+  async cloneEnvironments() {
+    const workflowName = getValueFromArgs(Variables.WORKFLOW, true);
+    const secretKey = getSecretKey();
+
+    let query = `?workflow=${workflowName}`;
+    const url = `${MSTATE_URL}/env/${query}`;
+
+    const responseJSON = await fetch(url, {
+      headers: {
+        'Content-Type': 'application/json',
+        'secret-key': secretKey,
+      },
+    });
+
+    const response = await responseJSON.json();
+    if (response?.errors) {
+      customLog.changeAndThrow(
+        'Invalid Parameters for Environment: ',
+        ...response?.errors,
+      );
+    } else {
+      const config = response?.data;
+
+      const filePath = path.resolve(workflowName, 'environment.json');
+      fs.writeFileSync(filePath, JSON.stringify(config, null, 2));
+    }
+  }
+
+  async saveToDB() {
+    const workflowName = getValueFromArgs(Variables.WORKFLOW, true);
+    const secretKey = getSecretKey();
+
+    const filePath = path.resolve(workflowName, 'environment.json');
+
+    const data = fs.readFileSync(filePath, 'utf8');
+    const environments = JSON.parse(data);
+
+    try {
+      const url = `${MSTATE_URL}/env`;
+      const responseJSON = await fetch(url, {
+        method: 'POST',
+        headers: {
+          'secret-key': secretKey,
+          'Content-Type': 'application/json',
+        },
+        body: JSON.stringify({
+          workflow: workflowName,
+          env: environments,
+        }),
+      });
+
+      const response = await responseJSON.json();
+      if (response?.errors) {
+        customLog.changeAndThrow(
+          `Invalid Parameters for Environment: `,
+          ...response?.errors,
+        );
+      } else {
+        customLog.success(`Environments Updated successfully`);
+      }
+    } catch (error: any) {
+      customLog.changeAndThrow(error.message);
+    }
+  }
+}
+
+export const environmentHandler = new EnvironmentHandler();