<template>
	<Dialog
		:options="{
			title: title,
			size: '2xl',
		}"
		v-model="show"
	>
		<template #body-content>
			<div
				v-if="
					$resources?.databaseServerStorageBreakdown?.loading ||
					$resources?.applicationServerStorageBreakdown?.loading
				"
				class="flex h-80 w-full items-center justify-center gap-2 text-base text-gray-700"
			>
				<Spinner class="w-4" /> Analyzing ...
			</div>
			<div
				v-else-if="
					$resources?.databaseServerStorageBreakdown?.error ||
					$resources?.applicationServerStorageBreakdown?.error
				"
				class="flex h-80 w-full items-center justify-center gap-2 text-base text-gray-700"
			>
				<ErrorMessage
					:message="
						$resources.databaseServerStorageBreakdown.error ||
						$resources?.applicationServerStorageBreakdown?.error
					"
				/>
			</div>
			<div v-else>
				<StorageBreakupChart
					:colorPalette="colorPalette"
					:data="
						serverType == 'Database Server'
							? databaseStorageBreakdown
							: applicationServerBreakDown
					"
					:keyFormatter="keyFormatter"
					:valueFormatter="(key, value) => formatSizeInKB(value)"
					:stickyKeys="['free', 'os']"
					:hiddenKeysInSlider="['free']"
					:isTree="serverType === 'Server'"
				/>

				<div v-if="serverType === 'Database Server' && noOfDatabases">
					<div
						v-if="noOfDatabases > 1"
						class="my-3 flex flex-row items-center justify-between px-1.5"
					>
						<div class="flex flex-row items-center gap-1">
							<p class="text-base font-semibold text-gray-800">
								Usage of
								{{
									noOfDatabases > topNDatabases && !showAllDatabases
										? `Top ${topNDatabases} Databases`
										: `${noOfDatabases} Databases`
								}}
							</p>
						</div>
						<Button
							variant="outline"
							@click="showAllDatabases = !showAllDatabases"
						>
							{{ showAllDatabases ? 'Show Less' : 'Show All' }}
						</Button>
					</div>
					<StorageBreakupChart
						:showSlider="false"
						:data="dbStorageUsage"
						:keyFormatter="keyFormatter"
						:valueFormatter="(key, value) => formatSizeInKB(value)"
						:showTopN="
							showAllDatabases
								? noOfDatabases
								: Math.min(noOfDatabases, topNDatabases)
						"
					/>
				</div>

				<div v-if="serverType === 'Database Server'" class="mt-4">
					<AlertBanner
						title="Are you looking to purge binary logs or reduce retention to free up space ?"
						type="info"
						:showIcon="false"
					>
						<Button
							class="ml-auto"
							variant="outline"
							link="https://docs.frappe.io/cloud/database-server-actions#view--purge-binlogs"
						>
							Docs
						</Button>
					</AlertBanner>
				</div>
			</div>
		</template>
	</Dialog>
</template>
<script>
import { Spinner } from 'frappe-ui';
import StorageBreakupChart from '../StorageBreakupChart.vue';
import { h } from 'vue';

export default {
	name: 'StorageBreakdown',
	components: {
		Spinner,
		StorageBreakupChart,
	},
	props: {
		title: {
			type: String,
			default: 'Storage Breakdown',
		},
		serverType: {
			type: String,
			required: true,
		},
		server: {
			type: String,
			required: true,
		},
	},
	data() {
		return {
			show: true,
			colorPalette: [
				'#2563eb',
				'#10b981',
				'#f59e42',
				'#a21caf',
				'#22d3ee',
				'#ef4444',
				'#fde047',
				'#fbbf24',
				'#6366f1',
				'#14b8a6',
				'#eab308',
				'#f472b6',
				'#64748b',
				'#84cc16',
			],
			topNDatabases: 5, // Default to showing top 10 databases
			showAllDatabases: false,
		};
	},
	mounted() {
		if (this.serverType == 'Database Server') {
			this.$resources.databaseServerStorageBreakdown.submit();
		} else {
			this.$resources.applicationServerStorageBreakdown.submit();
		}
	},
	resources: {
		applicationServerStorageBreakdown() {
			return {
				url: 'press.api.client.run_pg_method',
				makeParams() {
					return {
						dt: 'Server',
						dn: this.server,
						method: 'get_storage_usage',
					};
				},
				auto: false,
			};
		},
		databaseServerStorageBreakdown() {
			return {
				url: 'press.api.client.run_pg_method',
				makeParams() {
					return {
						dt: 'Database Server',
						dn: this.server,
						method: 'get_storage_usage',
					};
				},
				auto: false,
			};
		},
	},
	computed: {
		applicationServerBreakDown() {
			if (!this.$resources.applicationServerStorageBreakdown?.data?.message)
				return {};

			let message =
				this.$resources.applicationServerStorageBreakdown.data.message;

			const getDisplaySize = (formattedSize) => {
				var units = formattedSize.slice(-2);
				var sizeFormatted = formattedSize.replace(units, '');
				return `${sizeFormatted} ${units}`;
			};

			const transformNode = (node, isRoot = false) => {
				const transformed = {
					name: node.name,
					label: isRoot
						? `${node.name}`
						: `${node.name} (${getDisplaySize(node.size_formatted)})`,
					children: [],
				};

				if (node.children && node.children.length > 0) {
					transformed.children = node.children.map((child) =>
						transformNode(child),
					);
				}

				return transformed;
			};

			const additionalUsage = (
				(message.total.size - (message.benches.size + message.docker.size)) /
				1024 ** 3
			).toFixed(2);
			const totalCalculatedSize = (
				(message.benches.size + message.docker.size) /
				1024 ** 3
			).toFixed(2);

			const treeData = {
				name: 'server-storage',
				label: `Server Storage Breakdown (${totalCalculatedSize} GB)`,
				additionalUsage: `${additionalUsage} GB`,
				children: [],
			};

			if (message.benches) {
				treeData.children.push(transformNode(message.benches, true));
			}

			if (message.docker) {
				const dockerNode = {
					name: 'docker',
					label: 'Docker',
					children: [
						{
							name: 'docker-images',
							label: `Images (${getDisplaySize(message.docker.image)})`,
							children: [],
						},
						{
							name: 'docker-containers',
							label: `Containers (${getDisplaySize(message.docker.container)})`,
							children: [],
						},
					],
				};
				treeData.children.push(dockerNode);
			}

			return treeData;
		},
		databaseStorageBreakdown() {
			if (!this.$resources.databaseServerStorageBreakdown?.data?.message)
				return {};
			let message = this.$resources.databaseServerStorageBreakdown.data.message;
			let data = {
				free: message.disk_free,
				os: message.os_usage,
				bin_log: message.database.bin_log,
				slow_log: message.database.slow_log,
				error_log: message.database.error_log,
				db_other: message.database.other,
				db_core: message.database.core,
				binlog_indexes: message.binlog_indexes,
				db_data: Object.values(message.database.schema).reduce(
					(partialSum, a) => partialSum + a,
					0,
				),
			};
			return data;
		},
		noOfDatabases() {
			if (this.serverType !== 'Database Server') return 0;
			if (!this.$resources.databaseServerStorageBreakdown?.data?.message)
				return 0;
			let message = this.$resources.databaseServerStorageBreakdown.data.message;
			return Object.keys(message.database.schema || {}).length;
		},
		dbStorageUsage() {
			if (this.serverType !== 'Database Server') return {};
			if (!this.$resources.databaseServerStorageBreakdown?.data?.message)
				return {};
			let message = this.$resources.databaseServerStorageBreakdown.data.message;
			return message.database.schema || {};
		},
		dbNameSiteMapping() {
			if (this.serverType !== 'Database Server') return {};
			if (!this.$resources.databaseServerStorageBreakdown?.data?.message)
				return {};
			return (
				this.$resources.databaseServerStorageBreakdown.data.message
					?.db_name_site_map ?? {}
			);
		},
	},
	methods: {
		keyFormatter(key) {
			if (key in this.dbNameSiteMapping) {
				return h(
					'a',
					{
						href: `/dashboard/database-analyzer?site=${this.dbNameSiteMapping[key]}`,
						target: '_blank',
						rel: 'noopener noreferrer',
					},
					[
						`${this.dbNameSiteMapping[key]} (${key})`,
						h('span', { style: 'margin-left: 0.25em;' }, '↗'),
					],
				);
			}
			return (
				{
					free: 'Free Space',
					os: 'Operating System',
					bin_log: 'MariaDB Binary Log',
					slow_log: 'MariaDB Slow Log',
					error_log: 'MariaDB Error Log',
					db_data: `${this.noOfDatabases} Databases (including mysql, sys, perf_schema)`,
					db_core: 'MariaDB Core',
					db_other: 'MariaDB Owned System Files',
					binlog_indexes: 'MariaDB Binlog Indexes (Binlog Browser)',
				}[key] || key
			);
		},
		formatSizeInKB(kb) {
			try {
				let floatKB = parseFloat(kb);
				if (floatKB > 1024 * 512) {
					return `${Math.round(floatKB / 1024 / 1024).toFixed(1)} GB`;
				} else if (floatKB > 512) {
					return `${Math.round(floatKB / 1024).toFixed(1)} MB`;
				} else {
					return `${floatKB} KB`;
				}
			} catch (error) {
				return `${kb} KB`;
			}
		},
	},
};
</script>
