<?php

namespace App\Http\Controllers\Api;

use App\Http\Controllers\Controller;
use Illuminate\Http\Request;
use App\Api\Plugins\PluginLoader;
use App\Models\ApiProvider;
use App\Models\Style;
use App\Models\Image;
use Illuminate\Support\Facades\File;
use Illuminate\Support\Facades\Log;
use Carbon\Carbon;
use App\Models\Setting;

class ImageController extends Controller
{
    public function index(Request $request)
    {
        $publicUploadsPath = public_path('storage/uploads');
        
        // Ensure the directory exists
        if (!File::exists($publicUploadsPath)) {
            File::makeDirectory($publicUploadsPath, 0755, true);
        }

        // Get files from the public/storage/uploads directory
        $diskFiles = File::files($publicUploadsPath);
        $diskImagePaths = [];
        foreach ($diskFiles as $file) {
            // Store path relative to public/storage/
            $diskImagePaths[] = 'uploads/' . $file->getFilename();
        }

        $dbImagePaths = Image::pluck('path')->toArray();

        // Add images from disk that are not in the database
        $imagesToAdd = array_diff($diskImagePaths, $dbImagePaths);
        foreach ($imagesToAdd as $path) {
            Image::create(['path' => $path, 'is_public' => true]);
        }

        // Remove images from database that are not on disk
        $imagesToRemove = array_diff($dbImagePaths, $diskImagePaths);
        Image::whereIn('path', $imagesToRemove)->delete();

        // Fetch images from the database after synchronization
        $query = Image::orderBy('updated_at', 'desc');

        // If user is not authenticated, filter by is_public, but also include their temporary images
        if (!auth()->check()) {
            $query->where(function ($q) {
                $q->where('is_public', true)->orWhere('is_temp', true);
            });
        } else {
            // If user is authenticated, show all their images
        }

        $newImageTimespanMinutes = Setting::where('key', 'new_image_timespan_minutes')->first()->value ?? 60; // Default to 60 minutes

        $images = $query->get()->map(function ($image) use ($newImageTimespanMinutes) {
            $image->is_new = Carbon::parse($image->created_at)->diffInMinutes(Carbon::now()) <= $newImageTimespanMinutes;
            return $image;
        });

        $formattedImages = [];
        foreach ($images as $image) {
            $formattedImages[] = [
                'image_id' => $image->id,
                'path' => asset('storage/' . $image->path),
                'name' => basename($image->path),
                'is_temp' => (bool) $image->is_temp,
                'is_public' => (bool) $image->is_public,
                'is_new' => (bool) $image->is_new,
            ];
        }
        return response()->json($formattedImages);
    }

    public function upload(Request $request)
    {
        $request->validate([
            'image' => 'required|image|mimes:jpeg,png,bmp,gif,webp|max:10240', // Max 10MB
        ]);

        $file = $request->file('image');
        $fileName = uniqid() . '.' . $file->getClientOriginalExtension();
        $destinationPath = public_path('storage/uploads');

        // Ensure the directory exists
        if (!File::exists($destinationPath)) {
            File::makeDirectory($destinationPath, 0755, true);
        }

        $file->move($destinationPath, $fileName);
        $relativePath = 'uploads/' . $fileName; // Path relative to public/storage/

        $image = Image::create([
            'path' => $relativePath,
            'is_public' => true,
        ]);

        return response()->json([
            'message' => __('api.image_uploaded_successfully'),
            'image_id' => $image->id,
            'path' => asset('storage/' . $relativePath),
        ]);
    }

    public function styleChangeRequest(Request $request)
    {
        // Log the incoming request for debugging
        \Illuminate\Support\Facades\Log::info('styleChangeRequest called', [
            'image_id' => $request->image_id,
            'style_id' => $request->style_id,
            'all_params' => $request->all()
        ]);
        
        // Same-origin check
        $appUrl = config('app.url');
        $referer = $request->headers->get('referer');

        if ($referer && parse_url($referer, PHP_URL_HOST) !== parse_url($appUrl, PHP_URL_HOST)) {
            \Illuminate\Support\Facades\Log::warning('Unauthorized styleChangeRequest', [
                'referer' => $referer,
                'app_url' => $appUrl
            ]);
            return response()->json(['error' => 'Unauthorized: Request must originate from the same domain.'], 403);
        }

        $request->validate([
            'image_id' => 'required|exists:images,id',
            'style_id' => 'nullable|exists:styles,id',
        ]);

        $image = Image::find($request->image_id);
        $style = null;

        if ($request->style_id) {
            $style = Style::with(['aiModel' => function ($query) {
                $query->where('enabled', true)->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                    $query->where('enabled', true);
                }]);
            }])->find($request->style_id);
        } else {
            // Attempt to get default style from settings
            $defaultStyleSetting = \App\Models\Setting::where('key', 'default_style_id')->first();
            if ($defaultStyleSetting && $defaultStyleSetting->value) {
                $style = Style::with(['aiModel' => function ($query) {
                    $query->where('enabled', true)->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                        $query->where('enabled', true);
                    }]);
                }])->find($defaultStyleSetting->value);
            }
        }

        if (!$style || !$style->aiModel || $style->aiModel->apiProviders->isEmpty()) {
            \Illuminate\Support\Facades\Log::warning('Style or provider not found', [
                'style' => $style ? $style->toArray() : null,
                'ai_model' => $style && $style->aiModel ? $style->aiModel->toArray() : null
            ]);
            return response()->json(['error' => __('api.style_or_provider_not_found')], 404);
        }

        try {
            // Use the primary API provider for this AI model if available
            $apiProvider = $style->aiModel->primaryApiProvider;
            if (!$apiProvider) {
                // Fallback to the first enabled API provider from the many-to-many relationship
                $apiProvider = $style->aiModel->apiProviders->where('enabled', true)->first();
            }
            if (!$apiProvider) {
                // If no enabled provider found, try any provider
                $apiProvider = $style->aiModel->apiProviders->first();
            }
            if (!$apiProvider) {
                \Illuminate\Support\Facades\Log::error('No API provider found for style', [
                    'style_id' => $style->id,
                    'ai_model_id' => $style->aiModel->id
                ]);
                return response()->json(['error' => __('api.style_or_provider_not_found')], 404);
            }
            
            \Illuminate\Support\Facades\Log::info('Selected API provider for style change', [
                'api_provider_id' => $apiProvider->id,
                'api_provider_name' => $apiProvider->name,
                'plugin' => $apiProvider->plugin
            ]);
            
            $plugin = PluginLoader::getPlugin($apiProvider->plugin, $apiProvider);

            $result = $plugin->processImageStyleChange($image, $style);

            // Update the image model with the ComfyUI prompt_id and style_id
            $image->comfyui_prompt_id = $result['prompt_id'];
            $image->style_id = $style->id;
            $image->save();

            // Return the prompt_id for WebSocket tracking
            \Illuminate\Support\Facades\Log::info('Style change request completed', [
                'prompt_id' => $result['prompt_id'],
                'image_uuid' => $image->uuid
            ]);
            
            return response()->json([
                'message' => 'Style change request sent.',
                'prompt_id' => $result['prompt_id'],
                'image_uuid' => $image->uuid, // Pass image UUID for frontend tracking
                'plugin' => $apiProvider->plugin, // Pass plugin name for frontend handling
            ]);
        } catch (\Exception $e) {
            \Illuminate\Support\Facades\Log::error('Error in styleChangeRequest', [
                'error' => $e->getMessage(),
                'trace' => $e->getTraceAsString()
            ]);
            return response()->json(['error' => $e->getMessage()], 500);
        }
    }

    public function keepImage(Request $request)
    {
        $request->validate([
            'image_id' => 'required|exists:images,id',
        ]);

        $image = Image::find($request->image_id);

        if (!$image) {
            return response()->json(['error' => __('api.image_not_found')], 404);
        }

        $image->is_temp = false;
        $image->save();

        return response()->json(['message' => __('api.image_kept_successfully')]);
    }

    public function deleteImage(Image $image)
    {
        try {
            // Delete from the public/storage directory
            File::delete(public_path('storage/' . $image->path));
            $image->delete();
            return response()->json(['message' => __('api.image_deleted_successfully')]);
        } catch (\Exception $e) {
            return response()->json(['error' => $e->getMessage()], 500);
        }
    }

    public function getStatus(Request $request)
    {
        $request->validate([
            'image_id' => 'required|exists:images,id',
            'api_provider_name' => 'required|string',
        ]);

        $image = Image::find($request->image_id);
        $apiProvider = ApiProvider::where('name', $request->api_provider_name)->first();

        if (!$image || !$apiProvider) {
            return response()->json(['error' => __('api.image_or_provider_not_found')], 404);
        }

        try {
            $plugin = PluginLoader::getPlugin($apiProvider->name);
            $status = $plugin->getStatus($image->uuid); // Annahme: Image Model hat eine UUID
            return response()->json($status);
        } catch (\Exception $e) {
            return response()->json(['error' => $e->getMessage()], 500);
        }
    }

    public function fetchStyledImage(string $promptId)
    {
        Log::info('fetchStyledImage called.', ['prompt_id' => $promptId]);
        try {
            // Find the image associated with the prompt_id, eagerly loading relationships
            $image = Image::with(['style.aiModel' => function ($query) {
                $query->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                    $query->where('enabled', true);
                }]);
            }])->where('comfyui_prompt_id', $promptId)->first();

            if (!$image) {
                Log::warning('fetchStyledImage: Image not found for prompt_id.', ['prompt_id' => $promptId]);
                return response()->json(['error' => __('api.image_not_found')], 404);
            }

            Log::info('fetchStyledImage: Image found.', ['image_id' => $image->id, 'image_uuid' => $image->uuid, 'comfyui_prompt_id' => $image->comfyui_prompt_id]);

            // Get the style and API provider associated with the image
            $style = $image->style;
            if (!$style) {
                Log::warning('fetchStyledImage: Style not found for image.', ['image_id' => $image->id]);
                return response()->json(['error' => __('api.style_or_provider_not_found')], 404);
            }
            Log::info('fetchStyledImage: Style found.', ['style_id' => $style->id, 'style_name' => $style->title]);

            if (!$style->aiModel) {
                Log::warning('fetchStyledImage: AI Model not found for style.', ['style_id' => $style->id]);
                return response()->json(['error' => __('api.style_or_provider_not_found')], 404);
            }
            Log::info('fetchStyledImage: AI Model found.', ['ai_model_id' => $style->aiModel->id, 'ai_model_name' => $style->aiModel->name]);

            if ($style->aiModel->apiProviders->isEmpty()) {
                Log::warning('fetchStyledImage: No enabled API Providers found for AI Model.', ['ai_model_id' => $style->aiModel->id]);
                return response()->json(['error' => __('api.style_or_provider_not_found')], 404);
            }
            // Use the primary API provider for this AI model if available
            $apiProvider = $style->aiModel->primaryApiProvider;
            if (!$apiProvider) {
                // Fallback to the first enabled API provider from the many-to-many relationship
                $apiProvider = $style->aiModel->apiProviders->where('enabled', true)->first();
            }
            if (!$apiProvider) {
                // If no enabled provider found, try any provider
                $apiProvider = $style->aiModel->apiProviders->first();
            }
            Log::info('fetchStyledImage: API Provider found.', ['api_provider_id' => $apiProvider->id, 'api_provider_name' => $apiProvider->name]);

            Log::info('Fetching base64 image from plugin.', ['prompt_id' => $promptId, 'api_provider' => $apiProvider->name]);
            // Use the plugin to get the final image data (e.g., from ComfyUI's history/view)
            $plugin = PluginLoader::getPlugin($apiProvider->plugin, $apiProvider);
            $base64Image = $plugin->waitForResult($promptId); // Re-purpose waitForResult for final fetch

            if (empty($base64Image)) {
                Log::error('Received empty base64 image from plugin.', ['prompt_id' => $promptId]);
                return response()->json(['error' => 'Received empty image data.'], 500);
            }

            Log::info('Base64 image received. Decoding and saving.');
            $decodedImage = base64_decode(preg_replace('#^data:image/\w+;base64, #i', '', $base64Image));

            $newImageName = 'styled_' . uniqid() . '.png';
            $newImagePathRelative = 'uploads/' . $newImageName;
            $newImageFullPath = public_path('storage/' . $newImagePathRelative);

            if (!File::exists(public_path('storage/uploads'))) {
                File::makeDirectory(public_path('storage/uploads'), 0755, true);
                Log::info('Created uploads directory.', ['path' => public_path('storage/uploads')]);
            }

            File::put($newImageFullPath, $decodedImage); // Save using File facade
            Log::info('Image saved to disk.', ['path' => $newImageFullPath]);

            $newImage = Image::create([
                'path' => $newImagePathRelative, // Store relative path
                'original_image_id' => $image->id,
                'style_id' => $style->id,
                'is_temp' => true,
            ]);

            Log::info('New image record created in database.', ['image_id' => $newImage->id, 'path' => $newImage->path]);

            return response()->json([
                'message' => 'Styled image fetched successfully',
                'styled_image' => [
                    'id' => $newImage->id,
                    'path' => asset('storage/' . $newImage->path),
                    'is_temp' => $newImage->is_temp,
                ],
            ]);
        } catch (\Exception $e) {
            Log::error('Error in fetchStyledImage: ' . $e->getMessage(), ['exception' => $e]);
            return response()->json(['error' => $e->getMessage()], 500);
        }
    }

    public function getComfyUiUrl(Request $request)
    {
        $styleId = $request->query('style_id');
        $imageUuid = $request->query('image_uuid');
        
        $apiProvider = null;
        
        // If style_id is provided, get the API provider for that style
        if ($styleId) {
            $style = Style::with(['aiModel' => function ($query) {
                $query->where('enabled', true)->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                    $query->where('enabled', true);
                }]);
            }])->find($styleId);
            
            if ($style && $style->aiModel) {
                // Use the primary API provider for this AI model if available
                $apiProvider = $style->aiModel->primaryApiProvider;
                if (!$apiProvider) {
                    // Fallback to the first enabled API provider from the many-to-many relationship
                    $apiProvider = $style->aiModel->apiProviders->where('enabled', true)->first();
                }
                if (!$apiProvider) {
                    // If no enabled provider found, try any provider
                    $apiProvider = $style->aiModel->apiProviders->first();
                }
            }
        }
        // If image_uuid is provided, get the API provider for that image's style
        elseif ($imageUuid) {
            $image = Image::with(['style.aiModel' => function ($query) {
                $query->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                    $query->where('enabled', true);
                }]);
            }])->where('uuid', $imageUuid)->first();
            
            if ($image && $image->style && $image->style->aiModel) {
                // Use the primary API provider for this AI model if available
                $apiProvider = $image->style->aiModel->primaryApiProvider;
                if (!$apiProvider) {
                    // Fallback to the first enabled API provider from the many-to-many relationship
                    $apiProvider = $image->style->aiModel->apiProviders->where('enabled', true)->first();
                }
                if (!$apiProvider) {
                    // If no enabled provider found, try any provider
                    $apiProvider = $image->style->aiModel->apiProviders->first();
                }
            }
        }
        // Fallback to the old behavior if no style_id or image_uuid is provided
        else {
            // Try to get a default style if none is provided
            $defaultStyleSetting = \App\Models\Setting::where('key', 'default_style_id')->first();
            if ($defaultStyleSetting && $defaultStyleSetting->value) {
                $style = Style::with(['aiModel' => function ($query) {
                    $query->where('enabled', true)->with(['primaryApiProvider', 'apiProviders' => function ($query) {
                        $query->where('enabled', true);
                    }]);
                }])->find($defaultStyleSetting->value);
                
                if ($style && $style->aiModel) {
                    // Use the primary API provider for this AI model if available
                    $apiProvider = $style->aiModel->primaryApiProvider;
                    if (!$apiProvider) {
                        // Fallback to the first enabled API provider from the many-to-many relationship
                        $apiProvider = $style->aiModel->apiProviders->where('enabled', true)->first();
                    }
                    if (!$apiProvider) {
                        // If no enabled provider found, try any provider
                        $apiProvider = $style->aiModel->apiProviders->first();
                    }
                }
            }
            
            // If still no API provider, use the first available ComfyUI provider
            if (!$apiProvider) {
                $apiProvider = ApiProvider::where('plugin', 'ComfyUi')->where('enabled', true)->first();
            }
        }

        if (!$apiProvider) {
            return response()->json(['error' => 'No enabled ComfyUI API provider found.'], 404);
        }

        return response()->json(['comfyui_url' => rtrim($apiProvider->api_url, '/')]);
    }
}