<script setup lang="ts">
import { ref, onMounted, computed, watch } from 'vue'
import { useRouter } from 'vue-router'
import { dataService } from '../services/dataService'
import type { Asset } from '@/models/asset'
import Button from 'primevue/button'
import Skeleton from 'primevue/skeleton'
import Dialog from 'primevue/dialog'
import Paginator from 'primevue/paginator'
import ConfirmDialog from 'primevue/confirmdialog'
import { useConfirm } from "primevue/useconfirm"
import Toast from 'primevue/toast'
import { useToast } from "primevue/usetoast"
import { useAlbumStore } from '../stores/albums'
import { storeToRefs } from 'pinia'
import InputText from 'primevue/inputtext'
import Textarea from 'primevue/textarea'
import Image from 'primevue/image'

const router = useRouter()
const confirm = useConfirm()
const toast = useToast()
const assets = ref<Asset[]>([])
const loading = ref(true)
const activeFilter = ref('all')
// @ts-ignore
const API_URL = import.meta.env.VITE_API_URL

// Albums Logic
const albumStore = useAlbumStore()
const { albums, loading: albumsLoading } = storeToRefs(albumStore)
const viewMode = ref('assets') // 'assets' | 'albums'
const showCreateDialog = ref(false)
const newAlbum = ref({ name: '', description: '' })
const submittingAlbum = ref(false)

const selectedAsset = ref<Asset | null>(null)
const isModalVisible = ref(false)

const first = ref(0)
const rows = ref(18)
const totalRecords = ref(0)
const fileInput = ref<HTMLInputElement | null>(null)

const triggerFileUpload = () => {
    fileInput.value?.click()
}

const handleFileUpload = async (event: Event) => {
    const target = event.target as HTMLInputElement
    if (target.files && target.files[0]) {
        const file = target.files[0]
        try {
            loading.value = true
            await dataService.uploadAsset(file)
            toast.add({ severity: 'success', summary: 'Success', detail: 'Asset uploaded successfully', life: 3000 })
            // Reset to first page and reload
            first.value = 0
            activeFilter.value = 'uploaded' // Switch to uploaded view to see the new asset
            loadAssets()
        } catch (e) {
            console.error('Failed to upload asset', e)
            toast.add({ severity: 'error', summary: 'Error', detail: 'Failed to upload asset', life: 3000 })
        } finally {
            loading.value = false
            target.value = '' // Reset input
        }
    }
}

const openModal = (asset: Asset) => {
    selectedAsset.value = asset
    isModalVisible.value = true
}

const loadAssets = async () => {
    loading.value = true
    try {
        const response = await dataService.getAssets(rows.value, first.value, activeFilter.value)
        if (response && response.assets) {
            assets.value = response.assets
            totalRecords.value = response.total_count || 0
        } else {
            // Fallback for unexpected response structure
            assets.value = Array.isArray(response) ? response : []
            totalRecords.value = assets.value.length
        }
    } catch (e) {
        console.error('Failed to load assets', e)
    } finally {
        loading.value = false
    }
}

// Watch view mode to load albums
watch(viewMode, async (newMode) => {
    if (newMode === 'albums' && albums.value.length === 0) {
        await albumStore.fetchAlbums()
    }
})

onMounted(() => {
    loadAssets()
    // Preload albums if needed or wait for switch
})

const createAlbum = async () => {
    if (!newAlbum.value.name) return
    submittingAlbum.value = true
    try {
        await albumStore.createAlbum(newAlbum.value)
        showCreateDialog.value = false
        newAlbum.value = { name: '', description: '' }
        toast.add({ severity: 'success', summary: 'Success', detail: 'Album created', life: 3000 })
    } catch (e) {
        toast.add({ severity: 'error', summary: 'Error', detail: 'Failed to create album', life: 3000 })
    } finally {
        submittingAlbum.value = false
    }
}

const goToAlbumDetail = (id: string) => {
    router.push({ name: 'album-detail', params: { id } })
}

// Server-side filtering is now used, so we don't need a local filteredAssets computed property
// unless we want to do secondary filtering, but usually it's better to let the server handle it.

const paginatedAssets = computed(() => {
    // With server-side pagination, assets.value already contains only the current page
    return assets.value
})

const confirmDelete = (asset: Asset) => {
    confirm.require({
        message: 'Do you want to delete this asset?',
        header: 'Delete Confirmation',
        icon: 'pi pi-info-circle',
        rejectLabel: 'Cancel',
        acceptLabel: 'Delete',
        rejectClass: 'p-button-secondary p-button-outlined',
        acceptClass: 'p-button-danger',
        accept: async () => {
            try {
                await dataService.deleteAsset(asset.id)
                toast.add({ severity: 'success', summary: 'Confirmed', detail: 'Asset deleted', life: 3000 })
                loadAssets()
            } catch (e) {
                console.error('Failed to delete asset', e)
                toast.add({ severity: 'error', summary: 'Error', detail: 'Failed to delete asset', life: 3000 })
            }
        }
    })
}

const onPage = (event: any) => {
    first.value = event.first
    rows.value = event.rows
    loadAssets()
}

const handleFilterChange = (filter: string) => {
    activeFilter.value = filter
    first.value = 0 // Reset to first page when filter changes
    loadAssets()
}

const goBack = () => {
    router.push('/')
}

const formatDate = (dateString: string) => {
    if (!dateString) return ''
    return new Intl.DateTimeFormat('en-US', { month: 'short', day: 'numeric', year: 'numeric' }).format(new Date(dateString))
}

</script>

<template>
    <div class="flex flex-col h-full p-8 overflow-y-auto w-full text-slate-100">
        <!-- Main Content (Sidebar removed) -->
        <div class="flex-1 flex flex-col overflow-hidden">
            <!-- Top Bar -->
            <header class="flex justify-between items-center mb-4 px-1">
                <div>
                    <h1
                        class="text-xl font-bold bg-gradient-to-r from-white to-slate-400 bg-clip-text text-transparent m-0">
                        Library</h1>
                    <p class="text-xs text-slate-500 mt-1">Manage your generations and albums</p>
                </div>

                <div class="flex items-center gap-4">
                    <!-- View Switcher -->
                    <div class="glass-panel p-1 flex gap-1 rounded-xl bg-slate-800/50 border border-white/5">
                        <Button label="Assets" icon="pi pi-images"
                            :class="viewMode === 'assets' ? 'bg-white/10 text-slate-50 shadow-sm' : 'bg-transparent text-slate-400 hover:text-slate-200'"
                            class="px-3 py-1.5 rounded-lg text-xs font-medium transition-all duration-200" text
                            @click="viewMode = 'assets'" />
                        <Button label="Albums" icon="pi pi-book"
                            :class="viewMode === 'albums' ? 'bg-white/10 text-slate-50 shadow-sm' : 'bg-transparent text-slate-400 hover:text-slate-200'"
                            class="px-3 py-1.5 rounded-lg text-xs font-medium transition-all duration-200" text
                            @click="viewMode = 'albums'" />
                    </div>

                    <!-- Assets Toolbar -->
                    <div v-if="viewMode === 'assets'" class="flex items-center gap-2">
                        <input type="file" ref="fileInput" @change="handleFileUpload" class="hidden" accept="image/*" />
                        <Button label="Upload" icon="pi pi-upload" @click="triggerFileUpload"
                            class="!bg-violet-600 !border-none hover:!bg-violet-500 !text-xs !px-3 !py-1.5 !rounded-lg !font-medium shadow-lg shadow-violet-500/20" />

                        <div class="glass-panel p-1 flex gap-1 rounded-xl bg-slate-800/50 border border-white/5">
                            <Button v-for="filter in ['all', 'generated', 'uploaded']" :key="filter"
                                :label="filter.charAt(0).toUpperCase() + filter.slice(1)"
                                :class="activeFilter === filter ? 'bg-white/10 text-slate-50 shadow-sm' : 'bg-transparent text-slate-400 hover:text-slate-200'"
                                class="px-3 py-1.5 rounded-lg text-xs font-medium transition-all duration-200" text
                                @click="handleFilterChange(filter)" />
                        </div>
                    </div>
                    <!-- Albums Toolbar -->
                    <div v-if="viewMode === 'albums'" class="flex items-center gap-2">
                        <Button label="Create Album" icon="pi pi-plus" @click="showCreateDialog = true"
                            class="!bg-violet-600 !border-none hover:!bg-violet-500 !text-xs !px-3 !py-1.5 !rounded-lg !font-medium shadow-lg shadow-violet-500/20" />
                    </div>
                </div>
            </header>

            <!-- ASSETS VIEW -->
            <div v-if="viewMode === 'assets'" class="contents">
                <!-- Loading State -->
                <div v-if="loading"
                    class="grid grid-cols-2 md:grid-cols-4 lg:grid-cols-6 xl:grid-cols-6 gap-2 md:gap-1 pb-4 overflow-y-auto custom-scrollbar">
                    <Skeleton v-for="i in 12" :key="i" class="aspect-[9/16] !bg-slate-800 rounded-none sm:rounded-md" />
                </div>

                <!-- Assets Grid -->
                <div v-else class="flex-1 overflow-y-auto custom-scrollbar pb-20">
                    <div v-if="paginatedAssets.length > 0"
                        class="grid grid-cols-2 md:grid-cols-4 lg:grid-cols-6 xl:grid-cols-6 gap-2 md:gap-1">
                        <div v-for="asset in paginatedAssets" :key="asset.id"
                            class="aspect-[9/16] relative group overflow-hidden bg-slate-800 transition-all duration-300 rounded-none sm:rounded-md border border-white/5 hover:border-white/20"
                            @click="openModal(asset)">

                            <!-- Image -->
                            <img :src="(API_URL + asset.url + '?thumbnail=true') || 'https://via.placeholder.com/300'"
                                :alt="asset.name"
                                class="w-full h-full object-cover transition-transform duration-500 group-hover:scale-105" />

                            <!-- Type Badge -->
                            <div v-if="asset.type !== 'image'"
                                class="absolute top-2 right-2 bg-black/60 backdrop-blur-sm px-1.5 py-0.5 rounded text-[10px] uppercase font-bold text-white z-10 opacity-70 group-hover:opacity-100 transition-opacity">
                                {{ asset.type }}
                            </div>

                            <!-- Hover Overlay -->
                            <div
                                class="absolute inset-0 bg-black/60 opacity-0 group-hover:opacity-100 transition-opacity duration-200 flex flex-col justify-between p-3">

                                <!-- Top Actions -->
                                <div
                                    class="flex justify-between items-start translate-y-[-10px] group-hover:translate-y-0 transition-transform duration-200">
                                    <Button icon="pi pi-trash" v-tooltip.right="'Delete'"
                                        class="!w-7 !h-7 !rounded-full !bg-red-500/20 !border-none !text-red-400 text-xs hover:!bg-red-500 hover:!text-white transition-colors"
                                        @click.stop="confirmDelete(asset)" />

                                    <span v-if="asset.linked_char_id"
                                        class="text-[10px] bg-emerald-500/20 text-emerald-400 px-2 py-1 rounded-full border border-emerald-500/20"
                                        v-tooltip.left="'Linked to Character'">
                                        Linked
                                    </span>
                                </div>

                                <!-- Center View Button -->
                                <div class="absolute inset-0 flex items-center justify-center pointer-events-none">
                                    <Button icon="pi pi-eye" rounded text
                                        class="!bg-black/40 !text-white !w-10 !h-10 !rounded-full opacity-0 group-hover:opacity-100 hover:!bg-black/60 hover:!scale-110 transition-all duration-300 pointer-events-auto !border border-white/20"
                                        @click.stop="openModal(asset)" />
                                </div>

                                <!-- Bottom Info -->
                                <div
                                    class="translate-y-[10px] group-hover:translate-y-0 transition-transform duration-200">
                                    <p class="text-xs font-medium text-white line-clamp-1 mb-0.5">{{ asset.name }}</p>
                                    <p class="text-[10px] text-slate-400">{{ formatDate(asset.created_at) }}</p>
                                </div>
                            </div>
                        </div>
                    </div>

                    <div v-else class="flex flex-col items-center justify-center h-64 text-slate-500">
                        <i class="pi pi-images text-4xl mb-3 opacity-50"></i>
                        <p>No assets found</p>
                        <Button v-if="activeFilter !== 'all'" label="Clear Filters" text class="mt-2 !text-violet-400"
                            @click="handleFilterChange('all')" />
                    </div>

                    <!-- Pagination -->
                    <div v-if="totalRecords > rows" class="mt-6 flex justify-center">
                        <Paginator :first="first" :rows="rows" :totalRecords="totalRecords" @page="onPage" :template="{
                            default: 'FirstPageLink PrevPageLink PageLinks NextPageLink LastPageLink'
                        }" class="!bg-transparent !border-none !p-0" :pt="{
                            root: { class: '!bg-transparent' },
                            pages: { class: 'gap-1' },
                            pageButton: ({ context }) => ({
                                class: [
                                    '!min-w-[32px] !h-8 !rounded-lg !border-none !transition-all !duration-200 !text-xs',
                                    context.active ? '!bg-violet-600 !text-white !shadow-lg !shadow-violet-500/30' : '!bg-white/5 !text-slate-400 hover:!bg-white/10 hover:!text-slate-200'
                                ]
                            }),
                            firstPageButton: { root: { class: '!bg-white/5 !text-slate-400 !border-none !rounded-lg !min-w-[32px] !h-8 hover:!bg-white/10 hover:!text-slate-200 transition-all !mr-1' } },
                            previousPageButton: { root: { class: '!bg-white/5 !text-slate-400 !border-none !rounded-lg !min-w-[32px] !h-8 hover:!bg-white/10 hover:!text-slate-200 transition-all !mr-2' } },
                            nextPageButton: { root: { class: '!bg-white/5 !text-slate-400 !border-none !rounded-lg !min-w-[32px] !h-8 hover:!bg-white/10 hover:!text-slate-200 transition-all !ml-2' } },
                            lastPageButton: { root: { class: '!bg-white/5 !text-slate-400 !border-none !rounded-lg !min-w-[32px] !h-8 hover:!bg-white/10 hover:!text-slate-200 transition-all !ml-1' } }
                        }" />
                    </div>
                </div>
            </div>

            <!-- ALBUMS VIEW -->
            <div v-else class="flex-1 overflow-y-auto custom-scrollbar pb-20">
                <!-- Empty State -->
                <div v-if="albums.length === 0 && !albumsLoading"
                    class="flex flex-col items-center justify-center h-64 text-slate-400">
                    <i class="pi pi-folder-open text-6xl mb-4 opacity-50"></i>
                    <p class="text-xl">No albums found</p>
                    <Button label="Create your first album" class="mt-4 p-button-text"
                        @click="showCreateDialog = true" />
                </div>
                <!-- Loading State -->
                <div v-else-if="albumsLoading" class="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
                    <div v-for="i in 6" :key="i" class="glass-panel rounded-2xl p-6 flex flex-col gap-4">
                        <Skeleton height="12rem" class="w-full rounded-xl" />
                        <Skeleton width="60%" height="1.5rem" />
                        <Skeleton width="40%" height="1rem" />
                    </div>
                </div>

                <!-- Albums Grid -->
                <div v-else class="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
                    <div v-for="album in albums" :key="album.id"
                        class="glass-panel rounded-2xl p-6 flex flex-col gap-4 transition-all duration-300 cursor-pointer border border-white/5 hover:-translate-y-1 hover:bg-white/5 hover:border-white/10 group"
                        @click="goToAlbumDetail(album.id)">

                        <!-- Album Cover Placeholder -->
                        <div v-if="album.cover_asset_id"
                            class="aspect-video w-full bg-slate-800 rounded-xl flex items-center justify-center overflow-hidden border border-white/5">
                            <Image :src="API_URL + '/assets/' + album.cover_asset_id + '?thumbnail=true'" preview
                                class="w-full h-full object-cover" imageClass="w-full h-full object-cover"
                                @click.stop />
                        </div>
                        <div v-else
                            class="aspect-video w-full bg-slate-800 rounded-xl flex items-center justify-center overflow-hidden border border-white/5">
                            <i
                                class="pi pi-images text-4xl text-slate-600 group-hover:text-slate-400 transition-colors"></i>
                        </div>

                        <div>
                            <h3 class="m-0 mb-1 text-xl font-semibold truncate">{{ album.name }}</h3>
                            <p class="m-0 text-sm text-slate-400 line-clamp-2 min-h-[2.5em]">
                                {{ album.description || 'No description' }}
                            </p>
                            <div class="mt-3 flex items-center text-xs text-slate-500">
                                <i class="pi pi-image mr-1"></i>
                                <span>{{ album.generation_ids?.length || 0 }} items</span>
                            </div>
                        </div>
                    </div>
                </div>
            </div>
        </div>

        <Dialog v-model:visible="isModalVisible" modal dismissableMask header="Asset View"
            :style="{ width: '90vw', maxWidth: '800px' }" class="glass-panel rounded-2xl">
            <div v-if="selectedAsset" class="flex flex-col items-center">
                <img :src="selectedAsset.link ? API_URL + selectedAsset.link : (selectedAsset.url ? API_URL + selectedAsset.url : 'https://via.placeholder.com/800')"
                    :alt="selectedAsset.name" class="max-w-full max-h-[70vh] rounded-xl object-contain shadow-2xl" />
                <div class="mt-6 text-center">
                    <h2 class="text-2xl font-bold mb-2">{{ selectedAsset.name }}</h2>
                    <p class="text-slate-400">{{ formatDate(selectedAsset.created_at) }}</p>
                </div>
            </div>
        </Dialog>

        <!-- Create Album Dialog -->
        <Dialog v-model:visible="showCreateDialog" modal header="Create New Album" :style="{ width: '500px' }"
            :breakpoints="{ '960px': '75vw', '641px': '90vw' }" class="p-dialog-custom">
            <div class="flex flex-col gap-4 pt-4">
                <div class="flex flex-col gap-2">
                    <label for="name" class="font-semibold">Name</label>
                    <InputText id="name" v-model="newAlbum.name" class="w-full" placeholder="My Awesome Album"
                        autofocus />
                </div>
                <div class="flex flex-col gap-2">
                    <label for="description" class="font-semibold">Description</label>
                    <Textarea id="description" v-model="newAlbum.description" rows="3" class="w-full"
                        placeholder="Optional description..." />
                </div>
            </div>
            <template #footer>
                <Button label="Cancel" icon="pi pi-times" @click="showCreateDialog = false" text />
                <Button label="Create" icon="pi pi-check" @click="createAlbum" :loading="submittingAlbum" autofocus />
            </template>
        </Dialog>

        <ConfirmDialog></ConfirmDialog>
        <Toast />
    </div>
</template>

<style scoped>
/* Additional custom styles if needed */
</style>
