#!/bin/sh
# PhotoPrism RPCD Handler for LuCI
# Copyright (C) 2026 CyberMind.fr

. /usr/share/libubox/jshn.sh

CONFIG="photoprism"

# Method: status
method_status() {
	/usr/sbin/photoprismctl status 2>/dev/null || echo '{"installed":false,"running":false}'
}

# Method: get_config
method_get_config() {
	json_init

	json_add_boolean "enabled" "$([ "$(uci -q get ${CONFIG}.main.enabled)" = "1" ] && echo true || echo false)"
	json_add_string "data_path" "$(uci -q get ${CONFIG}.main.data_path || echo '/srv/photoprism')"
	json_add_string "originals_path" "$(uci -q get ${CONFIG}.main.originals_path || echo '/srv/photoprism/originals')"
	json_add_string "http_port" "$(uci -q get ${CONFIG}.main.http_port || echo '2342')"
	json_add_string "memory_limit" "$(uci -q get ${CONFIG}.main.memory_limit || echo '2G')"
	json_add_string "admin_user" "$(uci -q get ${CONFIG}.admin.username || echo 'admin')"
	json_add_string "domain" "$(uci -q get ${CONFIG}.network.domain || echo '')"
	json_add_boolean "face_recognition" "$([ "$(uci -q get ${CONFIG}.features.face_recognition)" = "1" ] && echo true || echo false)"
	json_add_boolean "object_detection" "$([ "$(uci -q get ${CONFIG}.features.object_detection)" = "1" ] && echo true || echo false)"
	json_add_boolean "places" "$([ "$(uci -q get ${CONFIG}.features.places)" = "1" ] && echo true || echo false)"

	json_dump
}

# Method: set_config
method_set_config() {
	local originals_path="$1"

	if [ -n "$originals_path" ]; then
		uci set ${CONFIG}.main.originals_path="$originals_path"
		uci commit ${CONFIG}
	fi

	json_init
	json_add_boolean "success" true
	json_dump
}

# Method: get_stats
method_get_stats() {
	local data_path=$(uci -q get ${CONFIG}.main.data_path || echo '/srv/photoprism')
	local originals_path=$(uci -q get ${CONFIG}.main.originals_path || echo '/srv/photoprism/originals')
	local originals_size="0"
	local storage_size="0"
	local photo_count=0
	local video_count=0

	if [ -d "${originals_path}" ]; then
		originals_size=$(du -sh "${originals_path}" 2>/dev/null | cut -f1 || echo "0")
		photo_count=$(find "${originals_path}" -type f \( -iname "*.jpg" -o -iname "*.jpeg" -o -iname "*.png" -o -iname "*.heic" -o -iname "*.raw" -o -iname "*.dng" \) 2>/dev/null | wc -l || echo 0)
		video_count=$(find "${originals_path}" -type f \( -iname "*.mp4" -o -iname "*.mov" -o -iname "*.avi" -o -iname "*.mkv" \) 2>/dev/null | wc -l || echo 0)
	fi

	if [ -d "${data_path}/storage" ]; then
		storage_size=$(du -sh "${data_path}/storage" 2>/dev/null | cut -f1 || echo "0")
	fi

	json_init
	json_add_int "photo_count" "$photo_count"
	json_add_int "video_count" "$video_count"
	json_add_string "originals_size" "$originals_size"
	json_add_string "storage_size" "$storage_size"
	json_dump
}

# Method: get_index_progress - Live indexing stats
method_get_index_progress() {
	local data_path=$(uci -q get ${CONFIG}.main.data_path || echo '/srv/photoprism')
	local storage_path="${data_path}/storage"

	# Count indexed files
	local sidecar_count=0
	local thumbnail_count=0
	local db_size="0"
	local indexing=false
	local current_file=""
	local last_activity=""

	if [ -d "${storage_path}/sidecar" ]; then
		sidecar_count=$(find "${storage_path}/sidecar" -type f 2>/dev/null | wc -l || echo 0)
	fi

	if [ -d "${storage_path}/cache/thumbnails" ]; then
		thumbnail_count=$(find "${storage_path}/cache/thumbnails" -type f 2>/dev/null | wc -l || echo 0)
	fi

	if [ -f "${storage_path}/photoprism.db" ]; then
		db_size=$(ls -lh "${storage_path}/photoprism.db" 2>/dev/null | awk '{print $5}')
	fi

	# Check if indexing process is running
	if lxc-attach -n photoprism -- ps aux 2>/dev/null | grep -q "photoprism index"; then
		indexing=true
	fi

	# Get last log entry
	local log_file="${storage_path}/index.log"
	if [ -f "$log_file" ]; then
		local last_line=$(tail -1 "$log_file" 2>/dev/null)
		# Extract filename from log
		current_file=$(echo "$last_line" | grep -oE '[0-9A-F]{8}-[0-9A-F]{4}-[0-9A-F]{4}-[0-9A-F]{4}-[0-9A-F]{12}\.[a-z]+' | head -1)
		# Extract timestamp
		last_activity=$(echo "$last_line" | grep -oE 'time="[^"]+"' | cut -d'"' -f2)
	fi

	cat <<EOF
{
	"indexing": $indexing,
	"sidecar_count": $sidecar_count,
	"thumbnail_count": $thumbnail_count,
	"db_size": "$db_size",
	"current_file": "$current_file",
	"last_activity": "$last_activity"
}
EOF
}

# Method: start
method_start() {
	/etc/init.d/photoprism start >/dev/null 2>&1
	json_init
	json_add_boolean "success" true
	json_dump
}

# Method: stop
method_stop() {
	/etc/init.d/photoprism stop >/dev/null 2>&1
	json_init
	json_add_boolean "success" true
	json_dump
}

# Method: restart
method_restart() {
	/etc/init.d/photoprism restart >/dev/null 2>&1
	json_init
	json_add_boolean "success" true
	json_dump
}

# Method: install
method_install() {
	local result
	result=$(/usr/sbin/photoprismctl install 2>&1)
	local success=$?

	json_init
	json_add_boolean "success" "$([ $success -eq 0 ] && echo true || echo false)"
	json_add_string "output" "$result"
	json_dump
}

# Method: uninstall
method_uninstall() {
	local result
	result=$(/usr/sbin/photoprismctl uninstall 2>&1)

	json_init
	json_add_boolean "success" true
	json_add_string "output" "$result"
	json_dump
}

# Method: index
method_index() {
	local result
	result=$(/usr/sbin/photoprismctl index 2>&1)

	json_init
	json_add_boolean "success" true
	json_add_string "output" "$result"
	json_dump
}

# Method: import
method_import() {
	local result
	result=$(/usr/sbin/photoprismctl import 2>&1)

	json_init
	json_add_boolean "success" true
	json_add_string "output" "$result"
	json_dump
}

# Method: emancipate
method_emancipate() {
	local domain="$1"

	if [ -z "$domain" ]; then
		json_init
		json_add_boolean "success" false
		json_add_string "error" "Domain required"
		json_dump
		return
	fi

	local result
	result=$(/usr/sbin/photoprismctl emancipate "$domain" 2>&1)

	json_init
	json_add_boolean "success" true
	json_add_string "output" "$result"
	json_add_string "url" "https://$domain"
	json_dump
}

# Method: logs
method_logs() {
	local lines="${1:-50}"
	local output
	output=$(/usr/sbin/photoprismctl logs "$lines" 2>&1)

	json_init
	json_add_string "logs" "$output"
	json_dump
}

# RPCD interface
case "$1" in
	list)
		echo '{
			"status": {},
			"get_config": {},
			"set_config": {"originals_path": "string"},
			"get_stats": {},
			"get_index_progress": {},
			"start": {},
			"stop": {},
			"restart": {},
			"install": {},
			"uninstall": {},
			"index": {},
			"import": {},
			"emancipate": {"domain": "string"},
			"logs": {"lines": "number"}
		}'
		;;
	call)
		case "$2" in
			status) method_status ;;
			get_config) method_get_config ;;
			set_config)
				read -r input
				originals_path=$(echo "$input" | jsonfilter -e '@.originals_path' 2>/dev/null)
				method_set_config "$originals_path"
				;;
			get_stats) method_get_stats ;;
			get_index_progress) method_get_index_progress ;;
			start) method_start ;;
			stop) method_stop ;;
			restart) method_restart ;;
			install) method_install ;;
			uninstall) method_uninstall ;;
			index) method_index ;;
			import) method_import ;;
			emancipate)
				read -r input
				domain=$(echo "$input" | jsonfilter -e '@.domain' 2>/dev/null)
				method_emancipate "$domain"
				;;
			logs)
				read -r input
				lines=$(echo "$input" | jsonfilter -e '@.lines' 2>/dev/null)
				method_logs "$lines"
				;;
		esac
		;;
esac