feat: ssh auth, protocol management, and cleanup
This commit is contained in:
infosave2007
@@ -0,0 +1,378 @@
|
||||
<?php
|
||||
|
||||
class AIController
|
||||
{
|
||||
|
||||
private $openRouterService;
|
||||
|
||||
public function __construct()
|
||||
{
|
||||
$this->openRouterService = new OpenRouterService();
|
||||
}
|
||||
|
||||
/**
|
||||
* AI Assistant endpoint for generating installation scripts
|
||||
*/
|
||||
public function assist(): void
|
||||
{
|
||||
requireAdmin();
|
||||
|
||||
try {
|
||||
$input = json_decode(file_get_contents('php://input'), true);
|
||||
if (!$input) {
|
||||
throw new Exception('Invalid JSON input');
|
||||
}
|
||||
|
||||
$prompt = trim($input['prompt'] ?? '');
|
||||
$model = trim($input['model'] ?? 'openai/gpt-3.5-turbo');
|
||||
$protocolType = trim($input['protocol_type'] ?? '');
|
||||
$protocolId = isset($input['protocol_id']) ? (int) $input['protocol_id'] : null;
|
||||
$target = trim($input['target'] ?? 'install'); // install, uninstall, template
|
||||
|
||||
if (empty($prompt)) {
|
||||
throw new Exception('Prompt is required');
|
||||
}
|
||||
|
||||
// Generate enhanced prompt based on protocol type and target
|
||||
$enhancedPrompt = $this->enhancePrompt($prompt, $protocolType, $target);
|
||||
|
||||
// Call OpenRouter API
|
||||
$response = $this->openRouterService->generateScript($enhancedPrompt, $model);
|
||||
|
||||
// Save AI generation to database
|
||||
$generationId = $this->saveAIGeneration([
|
||||
'protocol_id' => $protocolId,
|
||||
'model_used' => $model,
|
||||
'prompt' => "[$target] " . $prompt,
|
||||
'generated_script' => $response['script'] ?? '',
|
||||
'suggestions' => json_encode($response['suggestions'] ?? []),
|
||||
'ubuntu_compatible' => $response['ubuntu_compatible'] ?? null,
|
||||
'created_at' => date('Y-m-d H:i:s')
|
||||
]);
|
||||
|
||||
header('Content-Type: application/json');
|
||||
echo json_encode([
|
||||
'success' => true,
|
||||
'data' => [
|
||||
'script' => $response['script'] ?? '',
|
||||
'suggestions' => $response['suggestions'] ?? [],
|
||||
'ubuntu_compatible' => $response['ubuntu_compatible'] ?? false,
|
||||
'estimated_time' => $response['estimated_time'] ?? '5 minutes',
|
||||
'model_used' => $model,
|
||||
'generation_id' => $generationId,
|
||||
'target' => $target
|
||||
]
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
error_log("Error in AIController::assist: " . $e->getMessage());
|
||||
header('Content-Type: application/json');
|
||||
http_response_code(500);
|
||||
echo json_encode([
|
||||
'success' => false,
|
||||
'error' => $e->getMessage()
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get available AI models
|
||||
*/
|
||||
public function getModels(): void
|
||||
{
|
||||
requireAdmin();
|
||||
|
||||
try {
|
||||
$models = $this->openRouterService->getAvailableModels();
|
||||
|
||||
header('Content-Type: application/json');
|
||||
echo json_encode([
|
||||
'success' => true,
|
||||
'data' => $models
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
error_log("Error in AIController::getModels: " . $e->getMessage());
|
||||
header('Content-Type: application/json');
|
||||
http_response_code(500);
|
||||
echo json_encode([
|
||||
'success' => false,
|
||||
'error' => $e->getMessage()
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
public function testModel(): void
|
||||
{
|
||||
requireAdmin();
|
||||
header('Content-Type: application/json');
|
||||
|
||||
try {
|
||||
$input = json_decode(file_get_contents('php://input'), true);
|
||||
if (!$input) {
|
||||
throw new Exception('Invalid JSON input');
|
||||
}
|
||||
|
||||
$model = trim($input['model'] ?? '');
|
||||
if ($model === '') {
|
||||
throw new Exception('Model id is required');
|
||||
}
|
||||
|
||||
$result = $this->openRouterService->testModelAvailability($model);
|
||||
|
||||
echo json_encode([
|
||||
'success' => $result['success'] ?? false,
|
||||
'message' => $result['message'] ?? null,
|
||||
'http_code' => $result['http_code'] ?? null
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
http_response_code(400);
|
||||
echo json_encode([
|
||||
'success' => false,
|
||||
'error' => $e->getMessage()
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get AI generation history for a protocol
|
||||
*/
|
||||
public function getGenerationHistory(int $protocolId): void
|
||||
{
|
||||
requireAdmin();
|
||||
|
||||
try {
|
||||
$generations = $this->getAIGenerationsByProtocol($protocolId);
|
||||
|
||||
header('Content-Type: application/json');
|
||||
echo json_encode([
|
||||
'success' => true,
|
||||
'data' => $generations
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
error_log("Error in AIController::getGenerationHistory: " . $e->getMessage());
|
||||
header('Content-Type: application/json');
|
||||
http_response_code(500);
|
||||
echo json_encode([
|
||||
'success' => false,
|
||||
'error' => $e->getMessage()
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Apply AI-generated script to protocol
|
||||
*/
|
||||
public function applyGeneration(int $generationId): void
|
||||
{
|
||||
requireAdmin();
|
||||
|
||||
try {
|
||||
$generation = $this->getAIGeneration($generationId);
|
||||
if (!$generation) {
|
||||
throw new Exception('AI generation not found');
|
||||
}
|
||||
|
||||
if (!$generation['protocol_id']) {
|
||||
throw new Exception('This generation is not associated with any protocol');
|
||||
}
|
||||
|
||||
// Determine target from prompt
|
||||
$target = 'install';
|
||||
if (preg_match('/^\[(uninstall|template)\]/', $generation['prompt'], $matches)) {
|
||||
$target = $matches[1];
|
||||
}
|
||||
|
||||
// Update protocol with generated script
|
||||
$pdo = DB::conn();
|
||||
|
||||
if ($target === 'uninstall') {
|
||||
$stmt = $pdo->prepare('
|
||||
UPDATE protocols
|
||||
SET uninstall_script = ?, updated_at = ?
|
||||
WHERE id = ?
|
||||
');
|
||||
$stmt->execute([
|
||||
$generation['generated_script'],
|
||||
date('Y-m-d H:i:s'),
|
||||
$generation['protocol_id']
|
||||
]);
|
||||
} elseif ($target === 'template') {
|
||||
$stmt = $pdo->prepare('
|
||||
UPDATE protocols
|
||||
SET output_template = ?, updated_at = ?
|
||||
WHERE id = ?
|
||||
');
|
||||
$stmt->execute([
|
||||
$generation['generated_script'],
|
||||
date('Y-m-d H:i:s'),
|
||||
$generation['protocol_id']
|
||||
]);
|
||||
} else {
|
||||
$stmt = $pdo->prepare('
|
||||
UPDATE protocols
|
||||
SET install_script = ?, ubuntu_compatible = ?, updated_at = ?
|
||||
WHERE id = ?
|
||||
');
|
||||
$stmt->execute([
|
||||
$generation['generated_script'],
|
||||
$generation['ubuntu_compatible'],
|
||||
date('Y-m-d H:i:s'),
|
||||
$generation['protocol_id']
|
||||
]);
|
||||
}
|
||||
|
||||
header('Content-Type: application/json');
|
||||
echo json_encode([
|
||||
'success' => true,
|
||||
'message' => 'AI-generated script applied to protocol successfully'
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
error_log("Error in AIController::applyGeneration: " . $e->getMessage());
|
||||
header('Content-Type: application/json');
|
||||
http_response_code(500);
|
||||
echo json_encode([
|
||||
'success' => false,
|
||||
'error' => $e->getMessage()
|
||||
]);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Preview AI-generated script with syntax highlighting
|
||||
*/
|
||||
public function previewGeneration(int $generationId): void
|
||||
{
|
||||
requireAdmin();
|
||||
|
||||
try {
|
||||
$generation = $this->getAIGeneration($generationId);
|
||||
if (!$generation) {
|
||||
throw new Exception('AI generation not found');
|
||||
}
|
||||
|
||||
View::render('ai/preview_generation.twig', [
|
||||
'generation' => $generation,
|
||||
'script' => $generation['generated_script'],
|
||||
'suggestions' => json_decode($generation['suggestions'], true) ?? []
|
||||
]);
|
||||
|
||||
} catch (Exception $e) {
|
||||
$_SESSION['protocol_error'] = $e->getMessage();
|
||||
redirect('/settings/protocols');
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Enhance user prompt with context and requirements
|
||||
*/
|
||||
private function enhancePrompt(string $prompt, string $protocolType, string $target = 'install'): string
|
||||
{
|
||||
$context = "";
|
||||
|
||||
if ($target === 'uninstall') {
|
||||
$context = "Create a bash uninstallation script for a VPN protocol. ";
|
||||
} elseif ($target === 'template') {
|
||||
$context = "Create a WireGuard-compatible client config template. ";
|
||||
} else {
|
||||
$context = "Create a bash installation script for a VPN protocol. ";
|
||||
}
|
||||
|
||||
if ($protocolType) {
|
||||
$context .= "This is for a $protocolType protocol. ";
|
||||
}
|
||||
|
||||
$context .= "Requirements:\n";
|
||||
|
||||
if ($target === 'template') {
|
||||
$context .= "- Use Mustache-style placeholders like {{private_key}}, {{client_ip}}, {{server_host}}, {{server_port}}\n";
|
||||
$context .= "- The template should be valid configuration format (e.g. .conf for WireGuard)\n";
|
||||
} elseif ($target === 'uninstall') {
|
||||
$context .= "- The script should be compatible with Ubuntu 22.04 and 24.04\n";
|
||||
$context .= "- Remove all docker containers, images, and networks created by the installation\n";
|
||||
$context .= "- Remove configuration files and directories\n";
|
||||
$context .= "- Clean up firewall rules if necessary\n";
|
||||
} else {
|
||||
$context .= "- The script should be compatible with Ubuntu 22.04 and 24.04\n";
|
||||
$context .= "- Use Docker containers where possible for isolation\n";
|
||||
$context .= "- Generate necessary keys and certificates automatically\n";
|
||||
$context .= "- Configure appropriate firewall rules\n";
|
||||
$context .= "- Provide clear output about installation progress\n";
|
||||
$context .= "- Handle errors gracefully\n";
|
||||
$context .= "- Include configuration validation\n";
|
||||
}
|
||||
|
||||
$context .= "\nUser requirements: $prompt\n";
|
||||
$context .= "\nReturn the response in this JSON format:\n";
|
||||
$context .= "{\n";
|
||||
if ($target === 'template') {
|
||||
$context .= ' "script": "[Interface]\\nPrivateKey = {{private_key}}...",' . "\n";
|
||||
} else {
|
||||
$context .= ' "script": "#!/bin/bash\\n# Complete script",' . "\n";
|
||||
}
|
||||
$context .= ' "suggestions": ["suggestion 1", "suggestion 2"],' . "\n";
|
||||
$context .= ' "ubuntu_compatible": true,' . "\n";
|
||||
$context .= ' "estimated_time": "5 minutes"' . "\n";
|
||||
$context .= "}\n";
|
||||
|
||||
return $context;
|
||||
}
|
||||
|
||||
/**
|
||||
* Database methods
|
||||
*/
|
||||
|
||||
private function saveAIGeneration(array $data): int
|
||||
{
|
||||
$pdo = DB::conn();
|
||||
$stmt = $pdo->prepare('
|
||||
INSERT INTO ai_generations (protocol_id, model_used, prompt, generated_script, suggestions, ubuntu_compatible, created_at)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?)
|
||||
');
|
||||
$stmt->execute([
|
||||
$data['protocol_id'],
|
||||
$data['model_used'],
|
||||
$data['prompt'],
|
||||
$data['generated_script'],
|
||||
$data['suggestions'],
|
||||
$data['ubuntu_compatible'],
|
||||
$data['created_at']
|
||||
]);
|
||||
|
||||
return (int) $pdo->lastInsertId();
|
||||
}
|
||||
|
||||
private function getAIGeneration(int $id): ?array
|
||||
{
|
||||
$pdo = DB::conn();
|
||||
$stmt = $pdo->prepare('
|
||||
SELECT ag.*, p.name as protocol_name, p.slug as protocol_slug
|
||||
FROM ai_generations ag
|
||||
LEFT JOIN protocols p ON ag.protocol_id = p.id
|
||||
WHERE ag.id = ?
|
||||
');
|
||||
$stmt->execute([$id]);
|
||||
$generation = $stmt->fetch(PDO::FETCH_ASSOC);
|
||||
|
||||
return $generation ?: null;
|
||||
}
|
||||
|
||||
private function getAIGenerationsByProtocol(int $protocolId): array
|
||||
{
|
||||
$pdo = DB::conn();
|
||||
$stmt = $pdo->prepare('
|
||||
SELECT ag.*, p.name as protocol_name
|
||||
FROM ai_generations ag
|
||||
LEFT JOIN protocols p ON ag.protocol_id = p.id
|
||||
WHERE ag.protocol_id = ?
|
||||
ORDER BY ag.created_at DESC
|
||||
LIMIT 50
|
||||
');
|
||||
$stmt->execute([$protocolId]);
|
||||
|
||||
return $stmt->fetchAll(PDO::FETCH_ASSOC);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user