wg-easy/src/server/database/repositories/interface/service.ts

import type { DBType } from '#db/sqlite';
import isCidr from 'is-cidr';
import { eq, sql } from 'drizzle-orm';
import { wgInterface } from './schema';
import type { InterfaceCidrUpdateType, InterfaceUpdateType } from './types';
import { client as clientSchema } from '#db/schema';
import { parseCidr } from 'cidr-tools';

function createPreparedStatement(db: DBType) {
  return {
    get: db.query.wgInterface
      .findFirst({ where: eq(wgInterface.name, sql.placeholder('interface')) })
      .prepare(),
    getAll: db.query.wgInterface.findMany().prepare(),
    updateKeyPair: db
      .update(wgInterface)
      .set({
        privateKey: sql.placeholder('privateKey') as never as string,
        publicKey: sql.placeholder('publicKey') as never as string,
      })
      .where(eq(wgInterface.name, sql.placeholder('interface')))
      .prepare(),
  };
}

export class InterfaceService {
  #db: DBType;
  #statements: ReturnType<typeof createPreparedStatement>;

  constructor(db: DBType) {
    this.#db = db;
    this.#statements = createPreparedStatement(db);
  }

  get(infName: string) {
    return this.#statements.get.execute({ interface: infName });
  }

  getAll() {
    return this.#statements.getAll.execute();
  }

  updateKeyPair(infName: string, privateKey: string, publicKey: string) {
    return this.#statements.updateKeyPair.execute({
      interface: infName,
      privateKey,
      publicKey,
    });
  }

  update(infName: string, data: InterfaceUpdateType) {
    return this.#db
      .update(wgInterface)
      .set(data)
      .where(eq(wgInterface.name, infName))
      .execute();
  }

  updateCidr(infName: string, data: InterfaceCidrUpdateType) {
    if (!isCidr(data.ipv4Cidr) || !isCidr(data.ipv6Cidr)) {
      throw new Error('Invalid CIDR');
    }
    return this.#db.transaction(async (tx) => {
      await tx
        .update(wgInterface)
        .set(data)
        .where(eq(wgInterface.name, infName))
        .execute();

      const clients = await tx.query.client.findMany().execute();

      for (const client of clients) {
        // TODO: optimize
        const clients = await tx.query.client.findMany().execute();

        const nextIpv4 = nextIP(4, parseCidr(data.ipv4Cidr), clients);
        const nextIpv6 = nextIP(6, parseCidr(data.ipv6Cidr), clients);

        await tx
          .update(clientSchema)
          .set({
            ipv4Address: nextIpv4,
            ipv6Address: nextIpv6,
          })
          .where(eq(clientSchema.id, client.id))
          .execute();
      }
    });
  }
}