AstralRinth/apps/app-frontend/src/helpers/worlds.ts

import type { GameVersion } from '@modrinth/ui'
import { autoToHTML } from '@sfirew/minecraft-motd-parser'
import { invoke } from '@tauri-apps/api/core'
import dayjs from 'dayjs'

import { get_full_path } from '@/helpers/profile'
import { openPath } from '@/helpers/utils'

type BaseWorld = {
	name: string
	last_played?: string
	icon?: string
	display_status: DisplayStatus
	type: WorldType
}

export type WorldType = 'singleplayer' | 'server'
export type DisplayStatus = 'normal' | 'hidden' | 'favorite'

export type SingleplayerWorld = BaseWorld & {
	type: 'singleplayer'
	path: string
	game_mode: SingleplayerGameMode
	hardcore: boolean
	locked: boolean
}

export type ServerWorld = BaseWorld & {
	type: 'server'
	index: number
	address: string
	pack_status: ServerPackStatus
}

export type World = SingleplayerWorld | ServerWorld

export type WorldWithProfile = {
	profile: string
} & World

export type SingleplayerGameMode = 'survival' | 'creative' | 'adventure' | 'spectator'
export type ServerPackStatus = 'enabled' | 'disabled' | 'prompt'

export type ServerStatus = {
	// https://minecraft.wiki/w/Text_component_format
	description?: string | Chat
	players?: {
		max: number
		online: number
		sample: { name: string; id: string }[]
	}
	version?: {
		name: string
		protocol: number
		legacy: boolean
	}
	favicon?: string
	enforces_secure_chat: boolean
	ping?: number
}

export interface Chat {
	text: string
	bold: boolean
	italic: boolean
	underlined: boolean
	strikethrough: boolean
	obfuscated: boolean
	color?: string
	extra: Chat[]
}

export type ServerData = {
	refreshing: boolean
	lastSuccessfulRefresh?: number
	status?: ServerStatus
	rawMotd?: string | Chat
	renderedMotd?: string
}

export type ProtocolVersion = {
	version: number
	legacy: boolean
}

export async function get_recent_worlds(
	limit: number,
	displayStatuses?: DisplayStatus[],
): Promise<WorldWithProfile[]> {
	return await invoke('plugin:worlds|get_recent_worlds', { limit, displayStatuses })
}

export async function get_profile_worlds(path: string): Promise<World[]> {
	return await invoke('plugin:worlds|get_profile_worlds', { path })
}

export async function get_singleplayer_world(
	instance: string,
	world: string,
): Promise<SingleplayerWorld> {
	return await invoke('plugin:worlds|get_singleplayer_world', { instance, world })
}

export async function set_world_display_status(
	instance: string,
	worldType: WorldType,
	worldId: string,
	displayStatus: DisplayStatus,
): Promise<void> {
	return await invoke('plugin:worlds|set_world_display_status', {
		instance,
		worldType,
		worldId,
		displayStatus,
	})
}

export async function rename_world(
	instance: string,
	world: string,
	newName: string,
): Promise<void> {
	return await invoke('plugin:worlds|rename_world', { instance, world, newName })
}

export async function reset_world_icon(instance: string, world: string): Promise<void> {
	return await invoke('plugin:worlds|reset_world_icon', { instance, world })
}

export async function backup_world(instance: string, world: string): Promise<number> {
	return await invoke('plugin:worlds|backup_world', { instance, world })
}

export async function delete_world(instance: string, world: string): Promise<void> {
	return await invoke('plugin:worlds|delete_world', { instance, world })
}

export async function add_server_to_profile(
	path: string,
	name: string,
	address: string,
	packStatus: ServerPackStatus,
): Promise<number> {
	return await invoke('plugin:worlds|add_server_to_profile', { path, name, address, packStatus })
}

export async function edit_server_in_profile(
	path: string,
	index: number,
	name: string,
	address: string,
	packStatus: ServerPackStatus,
): Promise<void> {
	return await invoke('plugin:worlds|edit_server_in_profile', {
		path,
		index,
		name,
		address,
		packStatus,
	})
}

export async function remove_server_from_profile(path: string, index: number): Promise<void> {
	return await invoke('plugin:worlds|remove_server_from_profile', { path, index })
}

export async function get_profile_protocol_version(path: string): Promise<ProtocolVersion | null> {
	return await invoke('plugin:worlds|get_profile_protocol_version', { path })
}

export async function get_server_status(
	address: string,
	protocolVersion: ProtocolVersion | null = null,
): Promise<ServerStatus> {
	return await invoke('plugin:worlds|get_server_status', { address, protocolVersion })
}

export async function start_join_singleplayer_world(path: string, world: string): Promise<unknown> {
	return await invoke('plugin:worlds|start_join_singleplayer_world', { path, world })
}

export async function start_join_server(path: string, address: string): Promise<unknown> {
	return await invoke('plugin:worlds|start_join_server', { path, address })
}

export async function showWorldInFolder(instancePath: string, worldPath: string) {
	const fullPath = await get_full_path(instancePath)
	return await openPath(fullPath + '/saves/' + worldPath)
}

export function getWorldIdentifier(world: World) {
	return world.type === 'singleplayer' ? world.path : world.address
}

export function sortWorlds(worlds: World[]) {
	worlds.sort((a, b) => {
		if (!a.last_played) {
			return 1
		}
		if (!b.last_played) {
			return -1
		}
		return dayjs(b.last_played).diff(dayjs(a.last_played))
	})
}

export function isSingleplayerWorld(world: World): world is SingleplayerWorld {
	return world.type === 'singleplayer'
}

export function isServerWorld(world: World): world is ServerWorld {
	return world.type === 'server'
}

export async function refreshServerData(
	serverData: ServerData,
	protocolVersion: ProtocolVersion | null,
	address: string,
): Promise<void> {
	const refreshTime = Date.now()
	serverData.refreshing = true
	await get_server_status(address, protocolVersion)
		.then((status) => {
			if (serverData.lastSuccessfulRefresh && serverData.lastSuccessfulRefresh > refreshTime) {
				// Don't update if there was a more recent successful refresh
				return
			}
			serverData.lastSuccessfulRefresh = Date.now()
			serverData.status = status
			if (status.description) {
				serverData.rawMotd = status.description
				serverData.renderedMotd = autoToHTML(status.description)
			}
		})
		.finally(() => {
			serverData.refreshing = false
		})
		.catch((err) => {
			console.error(`Refreshing addr ${address}`, protocolVersion, err)
			if (!protocolVersion?.legacy) {
				refreshServerData(serverData, { version: 74, legacy: true }, address)
			}
		})
}

export function refreshServers(
	worlds: World[],
	serverData: Record<string, ServerData>,
	protocolVersion: ProtocolVersion | null,
) {
	const servers = worlds.filter(isServerWorld)
	servers.forEach((server) => {
		if (!serverData[server.address]) {
			serverData[server.address] = {
				refreshing: true,
			}
		} else {
			serverData[server.address].refreshing = true
		}
	})

	// noinspection ES6MissingAwait - handled with .then by refreshServerData already
	Object.keys(serverData).forEach((address) =>
		refreshServerData(serverData[address], protocolVersion, address),
	)
}

export async function refreshWorld(worlds: World[], instancePath: string, worldPath: string) {
	const index = worlds.findIndex((w) => w.type === 'singleplayer' && w.path === worldPath)
	const newWorld = await get_singleplayer_world(instancePath, worldPath)
	if (index !== -1) {
		worlds[index] = newWorld
	} else {
		console.info(`Adding new world at path: ${worldPath}.`)
		worlds.push(newWorld)
	}
	sortWorlds(worlds)
}

export async function handleDefaultProfileUpdateEvent(
	worlds: World[],
	instancePath: string,
	e: ProfileEvent,
) {
	if (e.event === 'world_updated') {
		await refreshWorld(worlds, instancePath, e.world)
	}

	if (e.event === 'server_joined') {
		const world = worlds.find(
			(w) =>
				w.type === 'server' &&
				(w.address === `${e.host}:${e.port}` || (e.port == 25565 && w.address == e.host)),
		)
		if (world) {
			world.last_played = e.timestamp
			sortWorlds(worlds)
		} else {
			console.error(`Could not find world for server join event: ${e.host}:${e.port}`)
		}
	}
}

export async function refreshWorlds(instancePath: string): Promise<World[]> {
	const worlds = await get_profile_worlds(instancePath).catch((err) => {
		console.error(`Error refreshing worlds for instance: ${instancePath}`, err)
	})
	if (worlds) {
		sortWorlds(worlds)
	}

	return worlds ?? []
}

export function hasServerQuickPlaySupport(gameVersions: GameVersion[], currentVersion: string) {
	if (!gameVersions.length) {
		return true
	}

	const versionIndex = gameVersions.findIndex((v) => v.version === currentVersion)
	const targetIndex = gameVersions.findIndex((v) => v.version === 'a1.0.5_01')

	return versionIndex === -1 || targetIndex === -1 || versionIndex <= targetIndex
}

export function hasWorldQuickPlaySupport(gameVersions: GameVersion[], currentVersion: string) {
	if (!gameVersions.length) {
		return false
	}

	const versionIndex = gameVersions.findIndex((v) => v.version === currentVersion)
	const targetIndex = gameVersions.findIndex((v) => v.version === '23w14a')

	return versionIndex !== -1 && targetIndex !== -1 && versionIndex <= targetIndex
}

export type ProfileEvent = { profile_path_id: string } & (
	| {
			event: 'servers_updated'
	  }
	| {
			event: 'world_updated'
			world: string
	  }
	| {
			event: 'server_joined'
			host: string
			port: number
			timestamp: string
	  }
)