Add math helper

Author: nkwwk

src/helpers/math.helper.spec.ts

@@ -0,0 +1,11 @@
+import { describe, expect, test } from '@jest/globals';
+import mathHelper from './math.helper';
+
+describe('Test math helper js', () => {
+  test('generate random string', () => {
+    expect(mathHelper.random(10, { lower: true, upper: false, numeric: false, symbol: false })).toMatch(/^[a-z]{10}$/);
+    expect(mathHelper.random(10, { lower: false, upper: true, numeric: false, symbol: false })).toMatch(/^[A-Z]{10}$/);
+    expect(mathHelper.random(10, { lower: false, upper: false, numeric: true, symbol: false })).toMatch(/^[0-9]{10}$/);
+    expect(mathHelper.random(10, { lower: false, upper: false, numeric: false, symbol: true })).toMatch(/^[~`!@#$%^&*()_+-={}[\]:";'<>?,./|\\]{10}$/);
+  });
+});

src/helpers/math.helper.ts

@@ -0,0 +1,12 @@
+export default {
+  random: function (length: number, { lower = true, upper = true, numeric = true, symbol = false } = {}) {
+    let mask = '';
+    if (lower) mask += 'abcdefghijklmnopqrstuvwxyz';
+    if (upper) mask += 'ABCDEFGHIJKLMNOPQRSTUVWXYZ';
+    if (numeric) mask += '0123456789';
+    if (symbol) mask += '~`!@#$%^&*()_+-={}[]:";\'<>?,./|\\';
+    let result = '';
+    for (var i = length; i > 0; --i) result += mask[Math.floor(Math.random() * mask.length)];
+    return result;
+  }
+}

src/helpers/wp.helper.ts

@@ -1,32 +1,30 @@
 import { Engine } from 'php-parser';
 
-function parseWpConfig(content: string, options: any = {}) {
-  const engine = new Engine(options);
-  const { children } = engine.parseCode(content, '');
-  let database, dbHost = 'localhost', dbPort = 3306, dbUser;
-  for (const child of children as any[]) {
-    if (child.expression?.what?.name === 'define') {
-      const [ name, value ] = child.expression.arguments;
-      if (name.kind === 'string' && value.kind === 'string') {
-        if (name.value === 'DB_NAME') {
-          database = value.value as string;
-        }
-        if (name.value === 'DB_HOST') {
-          const [ host, port ] = value.value.split(':');
-          dbHost = host ?? 'localhost';
-          if (port !== undefined) {
-            dbPort = parseInt(port);
+export default {
+  parseWpConfig: function (content: string, options: any = {}) {
+    const engine = new Engine(options);
+    const { children } = engine.parseCode(content, '');
+    let database, dbHost = 'localhost', dbPort = 3306, dbUser;
+    for (const child of children as any[]) {
+      if (child.expression?.what?.name === 'define') {
+        const [ name, value ] = child.expression.arguments;
+        if (name.kind === 'string' && value.kind === 'string') {
+          if (name.value === 'DB_NAME') {
+            database = value.value as string;
+          }
+          if (name.value === 'DB_HOST') {
+            const [ host, port ] = value.value.split(':');
+            dbHost = host ?? 'localhost';
+            if (port !== undefined) {
+              dbPort = parseInt(port);
+            }
+          }
+          if (name.value === 'DB_USER') {
+            dbUser = value.value as string;
           }
-        }
-        if (name.value === 'DB_USER') {
-          dbUser = value.value as string;
         }
       }
     }
-  }
-  return { database, dbHost, dbPort, dbUser };
-}
-
-export default {
-  parseWpConfig,
+    return { database, dbHost, dbPort, dbUser };
+  },
 }

src/services/install.service.ts

@@ -4,26 +4,16 @@ import shell from 'shelljs';
 import inquirer from 'inquirer';
 import { Sequelize, QueryTypes } from 'sequelize';
 import wpHelper from '@/helpers/wp.helper.js';
+import mathHelper from '@/helpers/math.helper';
 import { MySQLCredential } from '@/types/common.js';
 
-function random(length: number, { lower = true, upper = true, numeric = true, symbol = false } = {}) {
-  let mask = '';
-  if (lower) mask += 'abcdefghijklmnopqrstuvwxyz';
-  if (upper) mask += 'ABCDEFGHIJKLMNOPQRSTUVWXYZ';
-  if (numeric) mask += '0123456789';
-  if (symbol) mask += '~`!@#$%^&*()_+-={}[]:";\'<>?,./|\\';
-  let result = '';
-  for (var i = length; i > 0; --i) result += mask[Math.floor(Math.random() * mask.length)];
-  return result;
-}
-
 async function configSetup(wpDir: string, domain: string, nickname: string, database: MySQLCredential) {
   console.log('Setup OS and MySQL database');
 
   let { host, port, username, password } = database;
   let wpConfigDbUser = undefined;
   let wpDb = nickname;
-  let wpDbPassword = random(20, { symbol: true });
+  let wpDbPassword = mathHelper.random(20, { symbol: true });
 
   // check php version for default
   const matches = shell.exec('php -v').stdout.match(/^PHP ([0-9]\.[0-9])/);
@@ -63,7 +53,7 @@ async function configSetup(wpDir: string, domain: string, nickname: string, data
   }
 
   // setting wordpress config
-  shell.sed('-i', 'put your unique phrase here', random(32, { symbol: true }), wpConfig);
+  shell.sed('-i', 'put your unique phrase here', mathHelper.random(32, { symbol: true }), wpConfig);
   shell.sed('-i', 'database_name_here', wpDb, wpConfig);
   shell.sed('-i', 'username_here', nickname, wpConfig);
   shell.sed('-i', 'password_here', wpDbPassword, wpConfig);