use std::{net::{SocketAddrV4, UdpSocket}, str::FromStr};

use rosc::OscPacket;
use toml;
use serde::Deserialize;
use inputbot::{KeybdKey, KeybdKey::*};

// Configuration struct with mapping of OSC addresses to keyboard keys, and a general server configuration.
#[derive(Deserialize)]
struct Config {
	server: ServerConfig,
	// Mappings is called mapping in config
	#[serde(rename = "mapping")]
	mappings: Vec<EventKeyMapping>,
}
#[derive(Deserialize)]
struct EventKeyMapping {
	event: String,
	key: String,
}
#[derive(Deserialize)]
struct ServerConfig {
	port: u16,
}

fn main() {
	let config = match load_config() {
		Ok(c) => c,
		Err(e) => {
			println!("Error loading config file: {}", e);
			return;
		}
	};

	let mut event_key_map = std::collections::HashMap::new();
	for mapping in config.mappings {
		// Convert the key string to a KeybdKey
		let key = match to_fkey(&mapping.key) {
			Some(k) => k,
			None => {
				println!("Invalid key: {}", mapping.key);
				continue;
			}
		};
		event_key_map.insert(mapping.event, key);
	}
	
	println!("Starting OSC server on port {}...", config.server.port);	
    let addr = match SocketAddrV4::from_str(format!("127.0.0.1:{}", config.server.port).as_str()) {
        Ok(addr) => addr,
        Err(_) => panic!("Unable to match address"),
    };
    let sock = UdpSocket::bind(addr).unwrap();
    println!("Listening to {}", addr);

    let mut buf = [0u8; rosc::decoder::MTU];

    loop {
        match sock.recv_from(&mut buf) {
            Ok((size, addr)) => {
                println!("Received packet with size {} from: {}", size, addr);
                let (_, packet) = rosc::decoder::decode_udp(&buf[..size]).unwrap();
                handle_packet(packet, &event_key_map);
            }
            Err(e) => {
                println!("Error receiving from socket: {}", e);
                break;
            }
        }
    }
}

fn load_config() -> Result<Config, String> {
	let config_file = std::fs::read_to_string("config.toml").map_err(|e| e.to_string())?;
	let config: Config = toml::from_str(&config_file).map_err(|e| e.to_string())?;
	Ok(config)
}

fn handle_packet(packet: OscPacket, mappings: &std::collections::HashMap<String, KeybdKey>) {
    match packet {
        OscPacket::Message(msg) => {
            // Match event to key
			let key = match mappings.get(&msg.addr) {
				Some(k) => k,
				None => {
					return;
				}
			};
			// Press key
			key.press();
        }
        OscPacket::Bundle(bundle) => {
            println!("OSC Bundle: {:?}", bundle);
        }
    }
}

// Translate "F1", "F2", "F3" etc to corresponding KeybdKey.
fn to_fkey(key: &String) -> Option<KeybdKey> {
	match key.as_str() {
		"F1" => Some(F1Key),
		"F2" => Some(F2Key),
		"F3" => Some(F3Key),
		"F4" => Some(F4Key),
		"F5" => Some(F5Key),
		"F6" => Some(F6Key),
		"F7" => Some(F7Key),
		"F8" => Some(F8Key),
		"F9" => Some(F9Key),
		"F10" => Some(F10Key),
		"F11" => Some(F11Key),
		"F12" => Some(F12Key),
		"F13" => Some(F13Key),
		"F14" => Some(F14Key),
		"F15" => Some(F15Key),
		"F16" => Some(F16Key),
		"F17" => Some(F17Key),
		"F18" => Some(F18Key),
		"F19" => Some(F19Key),
		"F20" => Some(F20Key),
		"F21" => Some(F21Key),
		"F22" => Some(F22Key),
		"F23" => Some(F23Key),
		"F24" => Some(F24Key),
		_ => None,
	}
}