Seewo-HugoAura/src/aura/init/shared/registryManager.js

// @ts-check

const { exec, execSync } = require("child_process");

// Constants

const LOG_PREFIX = "[HugoAura / Init / Reg";
const LOG_PREFIX_FUNC = "[HugoAura / Reg";
const AURA_REGISTRY_PATH = [
  "HKEY_USERS",
  ".DEFAULT",
  "SOFTWARE",
  "HugoAura",
].join("\\");

class RegistryManager {
  /**
   * @param {string} [path]
   * @returns {Promise<boolean>}
   */
  async handleCreateReg(path) {
    try {
      const { stdout } = await new Promise((resolve, reject) => {
        exec(
          ["reg", "add", path].join(" "),
          { encoding: "utf8" },
          (error, stdout, stderr) => {
            if (error) reject(error);
            else resolve({ stdout, stderr });
          }
        );
      });

      console.log(
        `${LOG_PREFIX} / SUCCESS] Registry path ${path} successfully created.`
      );
      console.debug(`${LOG_PREFIX} / DEBUG] Reg add command stdout:`, stdout);
      return true;
    } catch (e) {
      console.error(
        `${LOG_PREFIX} / ERROR] Failed creating registry path, error:`,
        e
      );
      return false;
    }
  }

  /**
   * @param {string} [path]
   * @returns {boolean}
   */
  handleCreateRegSync(path) {
    try {
      const createResult = execSync(["reg", "add", path].join(" "), {
        encoding: "utf8",
      });

      if (createResult) {
        console.log(
          `${LOG_PREFIX} / SUCCESS] Registry path ${path} successfully created.`
        );
        console.debug(
          `${LOG_PREFIX} / DEBUG] Reg add command stdout:`,
          createResult
        );
        return true;
      } else {
        return false;
      }
    } catch (e) {
      console.error(
        `${LOG_PREFIX} / ERROR] Failed creating registry path, error:`,
        e
      );
      return false;
    }
  }

  /**
   * @returns {Promise<boolean>}
   */
  async initRegistry() {
    try {
      const { stdout } = await new Promise((resolve, reject) => {
        exec(
          ["reg", "query", AURA_REGISTRY_PATH].join(" "),
          { encoding: "utf8" },
          (error, stdout, stderr) => {
            if (error) reject(error);
            else resolve({ stdout, stderr });
          }
        );
      });

      console.log(`${LOG_PREFIX}] Registry check up success.`);
      console.debug(`${LOG_PREFIX}] Command stdout:`, stdout);
      return true;
    } catch (e) {
      console.warn(`${LOG_PREFIX} / WARN] Failed to query registry, error:`, e);
      return await this.handleCreateReg(AURA_REGISTRY_PATH);
    }
  }

  /**
   * @returns {boolean}
   */
  initRegistrySync() {
    try {
      const queryResult = execSync(
        ["reg", "query", AURA_REGISTRY_PATH].join(" "),
        { encoding: "utf8" }
      );

      if (queryResult) {
        console.log(`${LOG_PREFIX}] Registry check up success.`);
        console.debug(`${LOG_PREFIX}] Command stdout:`, queryResult);
        return true;
      } else {
        return false;
      }
    } catch (e) {
      console.warn(`${LOG_PREFIX} / WARN] Failed to query registry, error:`, e);
      return this.handleCreateRegSync(AURA_REGISTRY_PATH);
    }
  }

  /**
   * @param {string} relativePath
   * @param {string} keyName
   * @param {string} keyVal
   * @param {boolean | undefined} silent
   * @returns {Promise<{ success: boolean, error: Error | null }>}
   */
  async createOrUpdateRegKey(relativePath, keyName, keyVal, silent = false) {
    try {
      const { stdout } = await new Promise((resolve, reject) => {
        exec(
          [
            "reg",
            "add",
            [AURA_REGISTRY_PATH, relativePath].join("\\"),
            "/v",
            keyName,
            "/t",
            "REG_SZ",
            "/d",
            `\"${keyVal}\"`,
            "/f",
          ].join(" "),
          { encoding: "utf8" },
          (error, stdout, stderr) => {
            if (error) reject(error);
            else resolve({ stdout, stderr });
          }
        );
      });

      if (!silent) {
        console.debug(
          `${LOG_PREFIX_FUNC} / SUCCESS] Successfully created / updated reg key ${relativePath}/${keyName} with data: ${keyVal}`
        );
        console.debug(
          `${LOG_PREFIX_FUNC} / SUCCESS] Add key command stdout:`,
          stdout
        );
      }
      return {
        success: true,
        error: null,
      };
    } catch (e) {
      console.error(
        `${LOG_PREFIX_FUNC} / ERROR] Failed to create / update reg key, error:`,
        silent ? "<Hidden>" : e
      );
      return {
        success: false,
        error: e,
      };
    }
  }

  /**
   * @param {string} relativePath
   * @param {string} keyName
   * @param {string} keyVal
   * @param {boolean | undefined} silent
   * @returns {{ success: boolean, error: Error | null }}
   */
  createOrUpdateRegKeySync(relativePath, keyName, keyVal, silent = false) {
    try {
      const result = execSync(
        [
          "reg",
          "add",
          [AURA_REGISTRY_PATH, relativePath].join("\\"),
          "/v",
          keyName,
          "/t",
          "REG_SZ",
          "/d",
          `\"${keyVal}\"`,
          "/f",
        ].join(" "),
        { encoding: "utf8" }
      );

      if (result) {
        if (!silent) {
          console.debug(
            `${LOG_PREFIX} / SUCCESS] Successfully created / updated reg key ${relativePath}/${keyName} with data: ${keyVal}`,
            result
          );
          console.error("");
        }
      }
      return {
        success: true,
        error: null,
      };
    } catch (e) {
      console.error(
        `${LOG_PREFIX_FUNC} / ERROR] Failed to create / update reg key error:`,
        e
      );
      return {
        success: false,
        error: e,
      };
    }
  }

  /**
   * @param {string} relativePath
   * @param {string | null} keyName
   * @param {boolean | undefined} silent
   * @returns {Promise<{ success: boolean, error: Error | null }>}
   */
  async delRegKey(relativePath, keyName, silent = false) {
    if (keyName === undefined) {
      throw new Error(
        `${LOG_PREFIX_FUNC} / CRITICAL] Arg \"keyName\" for function \"delRegKey\" cannot be undefined. Only null or null accepted.`
      );
    }

    try {
      const { stdout } = await new Promise((resolve, reject) => {
        exec(
          [
            "reg",
            "delete",
            [AURA_REGISTRY_PATH, relativePath].join("\\"),
            keyName ? "/v" : "",
            keyName ? keyName : "",
            "/f",
          ].join(" "),
          { encoding: "utf8" },
          (error, stdout, stderr) => {
            if (error) reject(error);
            else resolve({ stdout, stderr });
          }
        );
      });

      if (!silent) {
        console.debug(
          `${LOG_PREFIX_FUNC} / SUCCESS] Successfully deleted reg key ${relativePath}/${keyName}`
        );
        console.debug(
          `${LOG_PREFIX_FUNC} / SUCCESS] Delete key command stdout:`,
          stdout
        );
      }
      return {
        success: true,
        error: null,
      };
    } catch (e) {
      console.error(
        `${LOG_PREFIX_FUNC} / ERROR] Failed to delete reg key, error:`,
        silent ? "<Hidden>" : e
      );
      return {
        success: false,
        error: e,
      };
    }
  }

  /**
   * @param {string} relativePath
   * @param {string | null} keyName
   * @param {boolean | undefined} silent
   * @returns {{ success: boolean, error: Error | null }}
   */
  delRegKeySync(relativePath, keyName, silent = false) {
    if (keyName === undefined) {
      throw new Error(
        `${LOG_PREFIX_FUNC} / CRITICAL] Arg \"keyName\" for function \"delRegKeySync\" cannot be undefined. Only null or string accepted.`
      );
    }

    try {
      const result = execSync(
        [
          "reg",
          "delete",
          [AURA_REGISTRY_PATH, relativePath].join("\\"),
          keyName ? "/v" : "",
          keyName ? keyName : "",
          "/f",
        ].join(" "),
        { encoding: "utf8" }
      );

      if (result) {
        if (!silent) {
          console.debug(
            `${LOG_PREFIX_FUNC} / SUCCESS] Successfully deleted reg key ${relativePath}/${keyName}`
          );
          console.debug(
            `${LOG_PREFIX_FUNC} / SUCCESS] Delete key command stdout:`,
            result
          );
        }
        return {
          success: true,
          error: null,
        };
      } else {
        return {
          success: false,
          error: null,
        };
      }
    } catch (e) {
      console.error(
        `${LOG_PREFIX_FUNC} / ERROR] Failed to delete reg key, error:`,
        silent ? "<Hidden>" : e
      );
      return {
        success: false,
        error: e,
      };
    }
  }

  /**
   * @param {string} keyPath
   * @param {string} keyName
   * @param {boolean | undefined} silent
   * @param {boolean} absolute
   * @param {any} regex
   * @returns {Promise<{ success: boolean, data: string | null, error: Error | null }>}
   */
  async readRegKey(
    keyPath,
    keyName,
    silent = false,
    absolute = false,
    regex = null
  ) {
    try {
      const { stdout } = await new Promise((resolve, reject) => {
        exec(
          [
            "reg",
            "query",
            absolute ? keyPath : [AURA_REGISTRY_PATH, keyPath].join("\\"),
            "/v",
            `\"${keyName}\"`,
          ].join(" "),
          { encoding: "utf8" },
          (error, stdout, stderr) => {
            if (error) reject(error);
            else resolve({ stdout, stderr });
          }
        );
      });

      if (!silent) {
        console.debug(
          `${LOG_PREFIX_FUNC} / SUCCESS] Successfully read reg key ${keyPath}/${keyName}, stdout:`,
          stdout
        );
      }

      const match = regex ? stdout.match(regex) : stdout.match(/REG_SZ\s+(.+)/);

      if (!match) {
        console.warn(`${LOG_PREFIX} / WARN] Data not found in stdout`);
        return {
          success: false,
          data: null,
          error: new Error("Data not found"),
        };
      }

      const data = match[1].trim();
      return {
        success: true,
        data,
        error: null,
      };
    } catch (e) {
      console.error(`${LOG_PREFIX} / ERROR] Failed to read reg key, error:`, e);
      return {
        success: false,
        data: null,
        error: e,
      };
    }
  }

  /**
   * @param {string} keyPath
   * @param {string} keyName
   * @param {boolean | undefined} silent
   * @param {boolean} absolute
   * @param {any} regex
   * @returns {{ success: boolean, data: string | null, error: Error | null }}
   */
  readRegKeySync(
    keyPath,
    keyName,
    silent = false,
    absolute = false,
    regex = null
  ) {
    try {
      const readResult = execSync(
        [
          "reg",
          "query",
          absolute ? keyPath : [AURA_REGISTRY_PATH, keyPath].join("\\"),
          "/v",
          `\"${keyName}\"`,
        ].join(" "),
        { encoding: "utf8" }
      );

      if (readResult) {
        if (!silent) {
          console.debug(
            `${LOG_PREFIX_FUNC} / SUCCESS] Successfully read reg key ${keyPath}/${keyName}, stdout:`,
            readResult
          );
        }
        const match = regex
          ? readResult.match(regex)
          : readResult.match(/REG_SZ\s+(.+)/);

        if (!match) {
          console.warn(`${LOG_PREFIX} / WARN] Data not found in stdout`);
          return {
            success: false,
            data: null,
            error: new Error("Data not found"),
          };
        }

        const data = match[1].trim();
        return {
          success: true,
          data,
          error: null,
        };
      } else {
        return {
          success: false,
          data: null,
          error: null,
        };
      }
    } catch (e) {
      console.error(
        `${LOG_PREFIX} / ERROR] Failed to read reg key, error:`,
        silent ? "<Hidden>" : e
      );
      return {
        success: false,
        data: null,
        error: e,
      };
    }
  }
}

module.exports = RegistryManager;