wg-easy/src/server/utils/config.ts

import debug from 'debug';
import packageJson from '@@/package.json';
import { z } from 'zod';
import type { Database } from '~~/services/database/repositories/database';
import { parseCidr } from 'cidr-tools';
import { stringifyIp } from 'ip-bigint';

export const RELEASE = 'v' + packageJson.version;

export const SERVER_DEBUG = debug('Server');

export async function migrateConfig(input: unknown) {
  const schema = z.object({
    server: z.object({
      privateKey: z.string(),
      publicKey: z.string(),
      address: z.string(),
    }),
    clients: z.record(
      z.string(),
      z.object({
        name: z.string(),
        address: z.string(),
        privateKey: z.string(),
        publicKey: z.string(),
        preSharedKey: z.string(),
        createdAt: z.string(),
        updatedAt: z.string(),
        enabled: z.boolean(),
      })
    ),
  });
  const res = await schema.safeParseAsync(input);
  if (!res.success) {
    throw new Error('Invalid Config');
  }
  const system = await Database.system.get();
  const oldConfig = res.data;
  const oldCidr = parseCidr(oldConfig.server.address + '/24');
  const db = {
    system: {
      ...system,
      interface: {
        ...system.interface,
        address4: oldConfig.server.address,
        privateKey: oldConfig.server.privateKey,
        publicKey: oldConfig.server.publicKey,
      },
      userConfig: {
        ...system.userConfig,
        address4Range:
          stringifyIp({ number: oldCidr.start, version: 4 }) + '/24',
      },
    } satisfies Partial<Database['system']>,
    clients: {} as Database['clients'],
  };
  for (const [oldId, oldClient] of Object.entries(oldConfig.clients)) {
    const address6 = nextIPv6(db.system, db.clients);
    db.clients[oldId] = {
      id: oldId,
      address4: oldClient.address,
      createdAt: oldClient.createdAt,
      enabled: oldClient.enabled,
      name: oldClient.name,
      preSharedKey: oldClient.preSharedKey,
      privateKey: oldClient.privateKey,
      publicKey: oldClient.publicKey,
      updatedAt: oldClient.updatedAt,
      expiresAt: null,
      oneTimeLink: null,
      allowedIPs: db.system.userConfig.allowedIps,
      serverAllowedIPs: [],
      persistentKeepalive: 0,
      address6: address6,
    };
  }
}