47 lines
1.4 KiB
TypeScript
47 lines
1.4 KiB
TypeScript
import { db } from '@/lib/db'
|
|
import { machines } from '@/lib/db/schema'
|
|
import { eq } from 'drizzle-orm'
|
|
import { NextRequest, NextResponse } from 'next/server'
|
|
|
|
export async function POST(request: NextRequest) {
|
|
try {
|
|
const body = await request.json()
|
|
const { accessKey, name, hostname, os, osVersion, agentVersion, ipAddress } = body
|
|
|
|
if (!accessKey) {
|
|
return NextResponse.json({ error: 'Access key required' }, { status: 400 })
|
|
}
|
|
|
|
const result = await db
|
|
.select()
|
|
.from(machines)
|
|
.where(eq(machines.accessKey, accessKey))
|
|
.limit(1)
|
|
|
|
const machine = result[0]
|
|
if (!machine) {
|
|
return NextResponse.json({ error: 'Invalid access key' }, { status: 401 })
|
|
}
|
|
|
|
await db
|
|
.update(machines)
|
|
.set({
|
|
name: name || machine.name,
|
|
hostname: hostname || machine.hostname,
|
|
os: os || machine.os,
|
|
osVersion: osVersion || machine.osVersion,
|
|
agentVersion: agentVersion || machine.agentVersion,
|
|
ipAddress: ipAddress || machine.ipAddress,
|
|
isOnline: true,
|
|
lastSeen: new Date(),
|
|
updatedAt: new Date(),
|
|
})
|
|
.where(eq(machines.id, machine.id))
|
|
|
|
return NextResponse.json({ success: true, machineId: machine.id })
|
|
} catch (error) {
|
|
console.error('[Agent Register] Error:', error)
|
|
return NextResponse.json({ error: 'Internal server error' }, { status: 500 })
|
|
}
|
|
}
|