import { createLogger } from '@sim/logger' import { getErrorMessage } from '@sim/utils/errors' import { generateId } from '@sim/utils/id' import { type NextRequest, NextResponse } from 'next/server' import type { Client, SFTPWrapper, Stats } from 'ssh2' import { sshCheckFileExistsContract } from '@/lib/api/contracts/storage-transfer' import { parseRequest } from '@/lib/api/server' import { checkInternalAuth } from '@/lib/auth/hybrid' import { withRouteHandler } from '@/lib/core/utils/with-route-handler' import { createSSHConnection, getFileType, parsePermissions, sanitizePath, } from '@/app/api/tools/ssh/utils' const logger = createLogger('SSHCheckFileExistsAPI') function getSFTP(client: Client): Promise { return new Promise((resolve, reject) => { client.sftp((err, sftp) => { if (err) { reject(err) } else { resolve(sftp) } }) }) } export const POST = withRouteHandler(async (request: NextRequest) => { const requestId = generateId().slice(0, 8) try { const auth = await checkInternalAuth(request) if (!auth.success || !auth.userId) { logger.warn(`[${requestId}] Unauthorized SSH check file exists attempt`) return NextResponse.json({ error: auth.error || 'Unauthorized' }, { status: 401 }) } const parsed = await parseRequest(sshCheckFileExistsContract, request, {}) if (!parsed.success) return parsed.response const params = parsed.data.body logger.info( `[${requestId}] Checking if path exists: ${params.path} on ${params.host}:${params.port}` ) const client = await createSSHConnection({ host: params.host, port: params.port, username: params.username, password: params.password, privateKey: params.privateKey, passphrase: params.passphrase, }) try { const sftp = await getSFTP(client) const filePath = sanitizePath(params.path) const stats = await new Promise((resolve) => { sftp.stat(filePath, (err, stats) => { if (err) { resolve(null) } else { resolve(stats) } }) }) if (!stats) { logger.info(`[${requestId}] Path does not exist: ${filePath}`) return NextResponse.json({ exists: false, type: 'not_found', message: `Path does not exist: ${filePath}`, }) } const fileType = getFileType(stats) // Check if the type matches the expected type if (params.type !== 'any' && fileType !== params.type) { logger.info(`[${requestId}] Path exists but is not a ${params.type}: ${filePath}`) return NextResponse.json({ exists: false, type: fileType, size: stats.size, permissions: parsePermissions(stats.mode), modified: new Date((stats.mtime || 0) * 1000).toISOString(), message: `Path exists but is a ${fileType}, not a ${params.type}`, }) } logger.info(`[${requestId}] Path exists: ${filePath} (${fileType})`) return NextResponse.json({ exists: true, type: fileType, size: stats.size, permissions: parsePermissions(stats.mode), modified: new Date((stats.mtime || 0) * 1000).toISOString(), message: `Path exists: ${filePath} (${fileType})`, }) } finally { client.end() } } catch (error) { const errorMessage = getErrorMessage(error, 'Unknown error occurred') logger.error(`[${requestId}] SSH check file exists failed:`, error) return NextResponse.json( { error: `SSH check file exists failed: ${errorMessage}` }, { status: 500 } ) } })