<?php if ( ! defined('BASEPATH')) exit('No direct script access allowed');

/*
	Callbook library
	Instead of implementing functions for each callbook, we should call one library, which then calls the correct callbook according to config.
	This makes it easy to implement other callbooks as well.
*/

class Callbook {

	private $ci;

	public function __construct() {
		$this->ci = & get_instance();
	}

	// TODO:
	// Implement the following:
	// - Implement callsign reduced logic
	public function getCallbookData($callsign) {
		// Load callbook configuration from config.php
		$source_callbooks = $this->ci->config->item('callbook');
		$callbook_errors = array();

		// Check if the source callbook is a single element or an array
		if (is_array($source_callbooks)) {
			// Parse each callbook in the array until we get a valid result
			foreach ($source_callbooks as $source) {
				$callbook = $this->queryCallbook($callsign, $source);
				if (!isset($callbook['error']) || $callbook['error'] == '') {
					break;
				} else {
					$callbook_errors['error_'.$source] = $callbook['error'];
					$callbook_errors['error_'.$source.'_name'] = $callbook['source'];
				}
			}
		} else {
			// Single callbook lookup (default behavior)
			$callbook = $this->queryCallbook($callsign, $source_callbooks);
		}

		// Handle callbook specific fields
		if (! array_key_exists('geoloc', $callbook)) {
			$callbook['geoloc'] = '';
		}

		// qrz.com gives AA00aa if the user deleted his grid from the profile
		$this->ci->load->library('qra');
		if (!array_key_exists('gridsquare', $callbook) || !$this->ci->qra->validate_grid($callbook['gridsquare'])) {
			$callbook['gridsquare'] = '';
		}

		if (isset($callbook['error']) && $callbook['error'] != '') {
			if (is_array($source_callbooks)) {
				foreach ($source_callbooks as $source) {
					if (isset($callbook_errors['error_'.$source])) {
						$callbook['error_'.$source] = $callbook_errors['error_'.$source];
						$callbook['error_'.$source.'_name'] = $callbook_errors['error_'.$source.'_name'];
					}
				}
			}
		}
		return $callbook;
	}

	function queryCallbook($callsign, $source) {
		switch ($source) {
			case 'qrz':
				$callbook = $this->qrz($callsign, $this->ci->config->item('use_fullname'));
				break;
			case 'qrzcq':
				$callbook = $this->qrzcq($callsign);
				break;
			case 'hamqth':
				$callbook = $this->hamqth($callsign);
				break;
			case 'qrzru':
				$callbook = $this->qrzru($callsign);
				break;
			default:
				$callbook['error'] = 'No callbook defined. Please review configuration.';
		}

		log_message('debug', 'Callbook lookup for '.$callsign.' using '.$source.': '.((($callbook['error'] ?? '' ) != '') ? $callbook['error'] : 'Success'));
		return $callbook;
	}

	function qrz($callsign, $fullname) {
		if (!$this->ci->load->is_loaded('qrz')) {
			$this->ci->load->library('qrz');
		}
		if ($this->ci->config->item('qrz_username') == null || $this->ci->config->item('qrz_password') == null) {
			$callbook['error'] = 'Lookup not configured. Please review configuration.';
			$callbook['source'] = $this->ci->qrz->sourcename();
		} else {
			$username = $this->ci->config->item('qrz_username');
			$password = $this->ci->config->item('qrz_password');

			if (!$this->ci->session->userdata('qrz_session_key')) {
				$qrz_session_key = $this->ci->qrz->session($username, $password);
				$this->ci->session->set_userdata('qrz_session_key', $qrz_session_key);
			}

			$callbook = $this->ci->qrz->search($callsign, $this->ci->session->userdata('qrz_session_key'), $fullname);

			if ($callbook['error'] ?? '' == 'Invalid session key') {
				$qrz_session_key = $this->ci->qrz->session($username, $password);
				$this->ci->session->set_userdata('qrz_session_key', $qrz_session_key);
				$callbook = $this->ci->qrz->search($callsign, $this->ci->session->userdata('qrz_session_key'), $fullname);
			}

			if (strpos($callbook['error'] ?? '', 'Not found') !== false && strpos($callsign, "/") !== false) {
				$plaincall = $this->get_plaincall($callsign);
				// Now try again but give back reduced data, as we can't validate location and stuff (true at the end)
				$callbook = $this->ci->qrz->search($plaincall, $this->ci->session->userdata('qrz_session_key'), $fullname, true);
			}
		}
		$callbook['source'] = $this->ci->qrz->sourcename();

		return $callbook;
	}

	function qrzcq($callsign) {
		if (!$this->ci->load->is_loaded('qrzcq')) {
			$this->ci->load->library('qrzcq');
		}
		if ($this->ci->config->item('qrzcq_username') == null || $this->ci->config->item('qrzcq_password') == null) {
			$callbook['error'] = 'Lookup not configured. Please review configuration.';
			$callbook['source'] = $this->ci->qrzcq->sourcename();
		} else {
			$username = $this->ci->config->item('qrzcq_username');
			$password = $this->ci->config->item('qrzcq_password');

			if (!$this->ci->session->userdata('qrzcq_session_key')) {
				$result = $this->ci->qrzcq->session($username, $password);
				if ($result[0] == 0) {
					$this->ci->session->set_userdata('qrzcq_session_key', $result[1]);
				} else {
					$data['error'] = __("QRZCQ Error").": ".$result[1];
					$data['source'] = $this->ci->qrzcq->sourcename();
					return $data;
				}
			}

			$callbook = $this->ci->qrzcq->search($callsign, $this->ci->session->userdata('qrzcq_session_key'));

			if ($callbook['error'] ?? '' == 'Invalid session key') {
				$qrzcq_session_key = $this->ci->qrzcq->session($username, $password);
				$this->ci->session->set_userdata('qrzcq_session_key', $qrzcq_session_key);
				$callbook = $this->ci->qrzcq->search($callsign, $this->ci->session->userdata('qrzcq_session_key'));
			}

			if (strpos($callbook['error'] ?? '', 'Not found') !== false && strpos($callsign, "/") !== false) {
				$plaincall = $this->get_plaincall($callsign);
				// Now try again but give back reduced data, as we can't validate location and stuff (true at the end)
				$callbook = $this->ci->qrzcq->search($plaincall, $this->ci->session->userdata('qrzcq_session_key'), true);
			}
		}
		$callbook['source'] = $this->ci->qrzcq->sourcename();

		return $callbook;
	}

	function hamqth($callsign) {
		// Load the HamQTH library
		if (!$this->ci->load->is_loaded('hamqth')) {
			$this->ci->load->library('hamqth');
		}
		if ($this->ci->config->item('hamqth_username') == null || $this->ci->config->item('hamqth_password') == null) {
			$callbook['error'] = 'Lookup not configured. Please review configuration.';
			$callbook['source'] = $this->ci->hamqth->sourcename();
		} else {
			$username = $this->ci->config->item('hamqth_username');
			$password = $this->ci->config->item('hamqth_password');

			if (!$this->ci->session->userdata('hamqth_session_key')) {
				$hamqth_session_key = $this->ci->hamqth->session($username, $password);
				if ($hamqth_session_key == false) {
					$callbook['error'] = __("Error obtaining a session key for HamQTH query");
					$callbook['source'] = $this->ci->hamqth->sourcename();
					return $callbook;
				} else {
					$this->ci->session->set_userdata('hamqth_session_key', $hamqth_session_key);
				}
			}

			$callbook = $this->ci->hamqth->search($callsign, $this->ci->session->userdata('hamqth_session_key'));

			// If HamQTH session has expired, start a new session and retry the search.
			if ($callbook['error'] == "Session does not exist or expired") {
				$hamqth_session_key = $this->ci->hamqth->session($username, $password);
				$this->ci->session->set_userdata('hamqth_session_key', $hamqth_session_key);
				$callbook = $this->ci->hamqth->search($callsign, $this->ci->session->userdata('hamqth_session_key'));
			}

			if (strpos($callbook['error'] ?? '', 'Not found') !== false && strpos($callsign, "/") !== false) {
				$plaincall = $this->get_plaincall($callsign);
				// Now try again but give back reduced data, as we can't validate location and stuff (true at the end)
				$callbook = $this->ci->hamqth->search($plaincall, $this->ci->session->userdata('hamqth_session_key'), true);
			}
		}
		$callbook['source'] = $this->ci->hamqth->sourcename();

		return $callbook;
	}

	function qrzru($callsign) {
		if (!$this->ci->load->is_loaded('qrzru')) {
			$this->ci->load->library('qrzru');
		}
		if ($this->ci->config->item('qrzru_username') == null || $this->ci->config->item('qrzru_password') == null) {
			$callbook['error'] = 'Lookup not configured. Please review configuration.';
			$callbook['source'] = $this->ci->qrzru->sourcename();
		} else {
			$username = $this->ci->config->item('qrzru_username');
			$password = $this->ci->config->item('qrzru_password');

			if (!$this->ci->session->userdata('qrzru_session_key')) {
				$result = $this->ci->qrzru->session($username, $password);
				$this->ci->session->set_userdata('qrzru_session_key', $result);
			}

			$callbook = $this->ci->qrzru->search($callsign, $this->ci->session->userdata('qrzru_session_key'));

			if ($callbook['error'] ?? '' == 'Session does not exist or expired') {
				$qrzru_session_key = $this->ci->qrzru->session($username, $password);
				$this->ci->session->set_userdata('qrzru_session_key', $qrzru_session_key);
				$callbook = $this->ci->qrzru->search($callsign, $this->ci->session->userdata('qrzru_session_key'));
			}

			if (strpos($callbook['error'] ?? '', 'Callsign not found') !== false && strpos($callsign, "/") !== false) {
				$plaincall = $this->get_plaincall($callsign);
				// Now try again but give back reduced data, as we can't validate location and stuff (true at the end)
				$callbook = $this->ci->qrzru->search($plaincall, $this->ci->session->userdata('qrzru_session_key'), true);
			}
		}
		$callbook['source'] = $this->ci->qrzru->sourcename();

		return $callbook;
	}

	function get_plaincall($callsign) {
		$split_callsign = explode('/', $callsign);
		if (count($split_callsign) == 1) {				// case of plain callsign --> return callsign
			return $callsign;
		}

		// Case of known suffixes that are not part of the callsign
		if (in_array(strtoupper($split_callsign[1]), array('LGT', 'AM', 'LH', 'A', 'B', 'R', 'T', 'X', 'D', 'P', 'M', 'MM', 'QRP', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9'))) {
			return $split_callsign[0];
		}
		// case EA/FOABC --> return 1
		if (strlen($split_callsign[1]) > 3) {	// Last Element longer than 3 chars? Take that as call
			return $split_callsign[1];
		}
		// case F0ABC/KH6 --> return cell 0
		return $split_callsign[0];
	}

}