e16a2fa978
- WebSocket relay service (FastAPI) bridges agents and viewers - Python agent with screen capture (mss), input control (pynput), script execution, and auto-reconnect - Windows service wrapper, PyInstaller spec, NSIS installer for silent mass deployment (RemoteLink-Setup.exe /S /SERVER= /ENROLL=) - Enrollment token system: admin generates tokens, agents self-register - Real WebSocket viewer replaces simulated canvas - Linux agent binary served from /downloads/remotelink-agent-linux - DB migration 0002: viewer_token on sessions, enrollment_tokens table - Sign-up pages cleaned up (invite-only redirect) Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
94 lines
4.1 KiB
TypeScript
94 lines
4.1 KiB
TypeScript
import {
|
|
pgTable,
|
|
uuid,
|
|
text,
|
|
boolean,
|
|
timestamp,
|
|
integer,
|
|
} from 'drizzle-orm/pg-core'
|
|
import { sql } from 'drizzle-orm'
|
|
|
|
export const users = pgTable('users', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
email: text('email').unique().notNull(),
|
|
passwordHash: text('password_hash').notNull(),
|
|
fullName: text('full_name'),
|
|
company: text('company'),
|
|
role: text('role').default('user').notNull(),
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
|
|
updatedAt: timestamp('updated_at', { withTimezone: true }).defaultNow().notNull(),
|
|
})
|
|
|
|
export const machines = pgTable('machines', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
userId: uuid('user_id').references(() => users.id, { onDelete: 'cascade' }).notNull(),
|
|
name: text('name').notNull(),
|
|
hostname: text('hostname'),
|
|
os: text('os'),
|
|
osVersion: text('os_version'),
|
|
agentVersion: text('agent_version'),
|
|
ipAddress: text('ip_address'),
|
|
accessKey: text('access_key').unique().notNull(),
|
|
isOnline: boolean('is_online').default(false).notNull(),
|
|
lastSeen: timestamp('last_seen', { withTimezone: true }),
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
|
|
updatedAt: timestamp('updated_at', { withTimezone: true }).defaultNow().notNull(),
|
|
})
|
|
|
|
export const sessionCodes = pgTable('session_codes', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
code: text('code').notNull(),
|
|
machineId: uuid('machine_id').references(() => machines.id, { onDelete: 'cascade' }).notNull(),
|
|
createdBy: uuid('created_by').references(() => users.id, { onDelete: 'set null' }),
|
|
expiresAt: timestamp('expires_at', { withTimezone: true }).notNull(),
|
|
isActive: boolean('is_active').default(true).notNull(),
|
|
usedAt: timestamp('used_at', { withTimezone: true }),
|
|
usedBy: uuid('used_by').references(() => users.id, { onDelete: 'set null' }),
|
|
})
|
|
|
|
export const sessions = pgTable('sessions', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
machineId: uuid('machine_id').references(() => machines.id, { onDelete: 'set null' }),
|
|
machineName: text('machine_name'),
|
|
viewerUserId: uuid('viewer_user_id').references(() => users.id, { onDelete: 'set null' }),
|
|
connectionType: text('connection_type'),
|
|
sessionCode: text('session_code'),
|
|
viewerToken: uuid('viewer_token').default(sql`gen_random_uuid()`),
|
|
startedAt: timestamp('started_at', { withTimezone: true }).defaultNow().notNull(),
|
|
endedAt: timestamp('ended_at', { withTimezone: true }),
|
|
durationSeconds: integer('duration_seconds'),
|
|
notes: text('notes'),
|
|
})
|
|
|
|
export const enrollmentTokens = pgTable('enrollment_tokens', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
token: uuid('token').unique().notNull().default(sql`gen_random_uuid()`),
|
|
label: text('label'),
|
|
createdBy: uuid('created_by').references(() => users.id, { onDelete: 'set null' }),
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
|
|
expiresAt: timestamp('expires_at', { withTimezone: true }),
|
|
usedCount: integer('used_count').default(0).notNull(),
|
|
maxUses: integer('max_uses'),
|
|
revokedAt: timestamp('revoked_at', { withTimezone: true }),
|
|
})
|
|
|
|
export const invites = pgTable('invites', {
|
|
id: uuid('id').primaryKey().default(sql`gen_random_uuid()`),
|
|
token: uuid('token').unique().notNull().default(sql`gen_random_uuid()`),
|
|
email: text('email').notNull(),
|
|
createdBy: uuid('created_by').references(() => users.id, { onDelete: 'set null' }),
|
|
createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
|
|
expiresAt: timestamp('expires_at', { withTimezone: true })
|
|
.notNull()
|
|
.default(sql`now() + interval '7 days'`),
|
|
usedAt: timestamp('used_at', { withTimezone: true }),
|
|
usedBy: uuid('used_by').references(() => users.id, { onDelete: 'set null' }),
|
|
})
|
|
|
|
export type User = typeof users.$inferSelect
|
|
export type Machine = typeof machines.$inferSelect
|
|
export type SessionCode = typeof sessionCodes.$inferSelect
|
|
export type Session = typeof sessions.$inferSelect
|
|
export type Invite = typeof invites.$inferSelect
|
|
export type EnrollmentToken = typeof enrollmentTokens.$inferSelect
|