import { sql, relations } from 'drizzle-orm';
import { int, sqliteTable, text, uniqueIndex } from 'drizzle-orm/sqlite-core';

import { client } from '../../schema';

export const obfuscatorPreset = sqliteTable(
  'obfuscator_presets_table',
  {
    id: int().primaryKey({ autoIncrement: true }),
    name: text().notNull().unique(),
    isDefault: int('is_default', { mode: 'boolean' }).notNull().default(false),
    mode: text({ enum: ['WIREGUARD', 'SOCKS5'] })
      .notNull()
      .default('WIREGUARD'),
    extPort: int('ext_port').notNull().unique(),
    sourceIf: text('source_if').notNull().default('0.0.0.0'),
    target: text(),
    key: text().notNull(),
    masking: text({ enum: ['STUN', 'MEDIA', 'AUTO', 'NONE', 'TLS'] })
      .notNull()
      .default('STUN'),
    obfuscateBytes: int('obfuscate_bytes').notNull().default(0),
    dummy: int().notNull().default(40),
    verbose: text({ enum: ['error', 'warn', 'info', 'debug', 'trace'] })
      .notNull()
      .default('error'),
    clientWgLocalPort: int('client_wg_local_port').notNull().default(13255),
    mediaSsrc: int('media_ssrc'),
    clientLocalPort: int('client_local_port').notNull().default(1080),
    createdAt: text('created_at')
      .notNull()
      .default(sql`(CURRENT_TIMESTAMP)`),
    updatedAt: text('updated_at')
      .notNull()
      .default(sql`(CURRENT_TIMESTAMP)`)
      .$onUpdate(() => sql`(CURRENT_TIMESTAMP)`),
  },
  (t) => [
    uniqueIndex('uq_default_preset')
      .on(t.isDefault)
      .where(sql`${t.isDefault} = 1`),
  ]
);

export const obfuscatorPresetRelations = relations(
  obfuscatorPreset,
  ({ many }) => ({
    clients: many(client),
  })
);