index.mjs

import { readFile, writeFile } from 'node:fs/promises';
import { join } from 'node:path';
import * as nt from 'node-smb-server/lib/ntlm.js';
import SMBServer from 'node-smb-server/lib/smbserver.js';
import { init } from '@cloud-cli/cli';

const defaults = {
  allowAnonymous: false,
  smb2Support: false,
  extendedSecurity: true,
};

const configPath = join(process.cwd(), 'configuration', 'smb.json');

const manager = {
  async start() {
    const config = await manager.loadConfig();

    return new Promise((resolve, reject) => {
      if (manager.server) {
        manager.stop();
      }

      const port = config.port || 445;
      const host = config.host || '0.0.0.0';
      const server = new SMBServer(config, null);

      server.start(port, host, () => resolve(true));
      server.on('error', (e) => {
        console.log(e);
        reject(e)
      });

      manager.server = server;
    });
  },

  stop() {
    if (manager.server) {
      manager.server.stop();
      manager.server = null;
    }
    return true;
  },

  async loadConfigFile() {
    const config = JSON.parse(await readFile(configPath, 'utf-8'));
    config.shares = config.shares || {};
    config.users = config.users || {};
    return { ...config, ...defaults };
  },

  async loadConfig() {
    const config = await this.loadConfigFile();

    Object.keys(config.shares).forEach((name) => {
      const share = config.shares[name];
      share.path = join(config.rootDir || process.cwd(), share.path);
    });

    return config;
  },

  async saveConfig(config) {
    await writeFile(configPath, JSON.stringify(config, null, 2));
    return true;
  },

  async addUser({ name, password }) {
    const user = {
      [name]: this.createHash({ password }),
    };

    const config = await this.loadConfigFile();
    Object.assign(config.users, user);
    return this.saveConfig(config);
  },

  async removeUser({ name }) {
    const config = await this.loadConfigFile();
    delete config.users[name];
    return this.saveConfig(config);
  },

  async addShare({ name, path }) {
    const share = {
      [name]: {
        backend: 'fs',
        description: 'folder',
        path,
      },
    };

    const config = await this.loadConfigFile();
    Object.assign(config.shares, share);
    return this.saveConfig(config);
  },

  async removeShare({ name }) {
    const config = await this.loadConfigFile();
    delete config.shares[name];
    return this.saveConfig(config);
  },

  async configure({ host, port, domainName = 'USERS' }) {
    const config = await this.loadConfigFile();
    Object.assign(config, { host, port, domainName });
    return this.saveConfig(config);
  },

  createHash({ password }) {
    return {
      lmHash: nt.lm.createHash(password).toString('hex'),
      ntlmHash: nt.ntlm.createHash(password).toString('hex'),
    };
  },
};

export default {
  start() {
    return manager.start();
  },

  stop() {
    return manager.stop();
  },

  reload() {
    this.stop();
    return this.start();
  },

  createHash(args) {
    return manager.createHash(args);
  },

  addUser(args) {
    return manager.addUser(args);
  },

  removeUser(args) {
    return manager.removeUser(args);
  },

  addShare(args) {
    return manager.addShare(args);
  },

  removeShare(args) {
    return manager.removeShare(args);
  },

  configure(args) {
    return manager.configure(args);
  },

  [init]() {
    return manager.start();
  }
};