// Prevents additional console window on Windows in release, DO NOT REMOVE!!
#![cfg_attr(not(debug_assertions), windows_subsystem = "windows")]

#[cfg(not(target_os = "linux"))]
use border::WebviewWindowExt as WebviewWindowExtAlt;
use nostr_sdk::prelude::*;
use std::{collections::HashMap, fs, time::Duration};
use tauri::{async_runtime::Mutex, Manager};
#[cfg(not(target_os = "linux"))]
use tauri_plugin_decorum::WebviewWindowExt;

use commands::{account::*, chat::*};

mod commands;
mod common;

pub struct Nostr {
	client: Client,
	contact_list: Mutex<Vec<Contact>>,
	inbox_relays: Mutex<HashMap<PublicKey, Vec<String>>>,
}

fn main() {
	let invoke_handler = {
		let builder = tauri_specta::ts::builder().commands(tauri_specta::collect_commands![
			login,
			create_account,
			import_key,
			connect_account,
			get_accounts,
			get_metadata,
			get_contact_list,
			get_chats,
			get_chat_messages,
			get_inbox,
			set_inbox,
			connect_inbox,
			disconnect_inbox,
			send_message,
		]);

		#[cfg(debug_assertions)]
		let builder = builder.path("../src/commands.ts");

		builder.build().unwrap()
	};

	#[cfg(debug_assertions)]
	let builder = tauri::Builder::default().plugin(tauri_plugin_devtools::init());
	#[cfg(not(debug_assertions))]
	let builder = tauri::Builder::default();

	builder
		.setup(|app| {
			let handle = app.handle();
			let main_window = app.get_webview_window("main").unwrap();

			// Open devtools
			#[cfg(debug_assertions)]
			main_window.open_devtools();

			// Set custom decoration
			#[cfg(target_os = "windows")]
			main_window.create_overlay_titlebar().unwrap();

			// Set traffic light inset
			#[cfg(target_os = "macos")]
			main_window.set_traffic_lights_inset(12.0, 18.0).unwrap();

			// Restore native border
			#[cfg(target_os = "macos")]
			main_window.add_border(None);

			let client = tauri::async_runtime::block_on(async move {
				// Create data folder if not exist
				let dir = handle.path().app_config_dir().expect("App config directory not found.");
				let _ = fs::create_dir_all(dir.clone());

				// Setup database
				let database = SQLiteDatabase::open(dir.join("nostr.db")).await.expect("Error.");

				// Setup nostr client
				let opts = Options::new()
					.timeout(Duration::from_secs(30))
					.send_timeout(Some(Duration::from_secs(5)))
					.connection_timeout(Some(Duration::from_secs(5)));

				let client = ClientBuilder::default().opts(opts).database(database).build();

				// Add bootstrap relay
				let _ = client
					.add_relays(["wss://relay.poster.place/", "wss://bostr.nokotaro.com/"])
					.await;

				// Connect
				client.connect().await;

				client
			});

			// Create global state
			app.manage(Nostr {
				client,
				contact_list: Mutex::new(Vec::new()),
				inbox_relays: Mutex::new(HashMap::new()),
			});

			Ok(())
		})
		.enable_macos_default_menu(false)
		.plugin(tauri_plugin_prevent_default::init())
		.plugin(tauri_plugin_os::init())
		.plugin(tauri_plugin_clipboard_manager::init())
		.plugin(tauri_plugin_dialog::init())
		.plugin(tauri_plugin_decorum::init())
		.plugin(tauri_plugin_shell::init())
		.invoke_handler(invoke_handler)
		.run(tauri::generate_context!())
		.expect("error while running tauri application");
}