<?php

/**
 * Server list manager
 * @package framework
 * @subpackage servers
 */
trait Hm_Server_Connect {

    /* list of server connections */
    protected static $server_list = [];

    /**
     * Connect to a server
     * @param int $id server id
     * @param mixed $cache cached server data
     * @param string $user username
     * @param string $pass password
     * @param bool $save_credentials true to save the username and password
     * @return object|false connection object on success, otherwise false
     */
    public static function connect($id, $cache = false, $user = false, $pass = false, $save_credentials = false) {
        if (!array_key_exists($id, self::$server_list)) {
            return false;
        }
        $server = self::$server_list[$id];

        if ($server['object']) {
            return $server['object'];
        }
        list($user, $pass) = self::user_and_pass($user, $pass, $server);
        if ($user === false || $pass === false) {
            return false;
        }

        if (self::service_connect($id, $server, $user, $pass, $cache)) {
            return self::enable_server($id, $user, $pass, $save_credentials);
        }
        return self::$server_list[$id]['object'];
    }

    /**
     * Enable server
     * @param int $id server id
     * @param string $user username
     * @param string $pass password
     * @param bool $save save user and pass
     */
    private static function enable_server($id, $user, $pass, $save) {
        self::$server_list[$id]['connected'] = true;
        if ($save) {
            self::$server_list[$id]['user'] = $user;
            self::$server_list[$id]['pass'] = $pass;
        }
        return self::$server_list[$id]['object'];
    }

    /**
     * Process user and pass args
     * @param string|false $user username
     * @param string|false $pass password
     * @param array $server server details
     * @return array
     */
    private static function user_and_pass($user, $pass, $server) {
        if (array_key_exists('user', $server) && array_key_exists('pass', $server)) {
            return array($server['user'], $server['pass']);
        }
        return array($user, $pass);
    }

    /**
     * Server lists must override this method to connect
     * @param int $id server id
     * @param string $server server hostname or ip
     * @param string $user username for authentication
     * @param string $pass password for authentication
     * @param mixed $cache cached connection data
     * @return bool true on success
     */
    public abstract static function service_connect($id, $server, $user, $pass, $cache);

    /**
     * Try to disconnect cleanly
     * @param int $id server id
     * @return void
     */
    public static function clean_up($id = false) {
        if ($id !== false && array_key_exists($id, self::$server_list)) {
            self::disconnect($id);
        } else {
            foreach (self::$server_list as $index => $server) {
                self::disconnect($index);
            }
        }
    }

    /**
     * Disconnect from a server
     * @param int $id the server id to disconnect
     * @return void
     */
    public static function disconnect($id) {
        if (self::$server_list[$id]['connected'] && self::$server_list[$id]['object']) {
            if (method_exists(self::$server_list[$id]['object'], 'disconnect')) {
                self::$server_list[$id]['object']->disconnect();
            }
            self::$server_list[$id]['connected'] = false;
        }
    }
}

/**
 * Server list manager
 * @package framework
 * @subpackage servers
 */
trait Hm_Server_Modify {

    use Hm_Server_Connect;

    /**
     * Update the oauth2 password and password expiration
     * @param int $id server id
     * @param string $pass new password
     * @param int $expiry new password expiration timestamp
     * @return bool
     */
    public static function update_oauth2_token($id, $pass, $expiry) {
        if (!array_key_exists($id, self::$server_list)) {
            return false;
        }
        if (!array_key_exists('auth', self::$server_list[$id])) {
            return false;
        }
        if (self::$server_list[$id]['auth'] != 'xoauth2') {
            return false;
        }
        self::$server_list[$id]['pass'] = $pass;
        self::$server_list[$id]['expiration'] = $expiry;
        self::$server_list[$id]['object'] = false;
        return true;
    }

    /**
     * Remove the username and password from a connection
     * @param int $id server id
     * @return void
     */
    public static function forget_credentials($id) {
        if (array_key_exists($id, self::$server_list)) {
            unset(self::$server_list[$id]['user']);
            unset(self::$server_list[$id]['pass']);
        }
    }
    /**
     * Toggle the hidden status of a server
     * @param int $id server id
     * @param int $hide bool
     * @return void
     */
    public static function toggle_hidden($id, $hide) {
        if (array_key_exists($id, self::$server_list)) {
            self::$server_list[$id]['hide'] = $hide;
        }
    }
}

/**
 * Struct that makes it easy for a module set to manage a list of server connections
 * @package framework
 * @subpackage servers
 */
trait Hm_Server_List {

    use Hm_Server_Modify;
    use Searchable;
    use Hm_Repository {
        Hm_Repository::add as repo_add;
        Hm_Repository::get as repo_get;
    }

    public static function init($name, $user_config, $session) {
        self::initRepo($name, $user_config, $session, self::$server_list);
    }

    /**
     * Add a server definition
     * @param array $atts server details
     * @param int $id server id
     * @return void
     */
    public static function add($atts, $save = true) {
        $atts['object'] = false;
        $atts['connected'] = false;
        return self::repo_add($atts, $save);
    }

    /**
     * Return all server details
     * @param int $id if not false, return details for this server only
     * @param bool $full true to return passwords for server connections. CAREFUL!
     * @return array server details
     */
    public static function dump($id = false, $full = false) {
        $list = [];
        if ($id !== false) {
            return self::get($id, $full);
        }
        foreach (self::$server_list as $index => $server) {
            if (!$full) {
                $server = self::clean($server);
            }
            $list[$index] = $server;
        }
        return $list;
    }

    /**
     * Get the dataset for the server list
     * @return array
     */
    protected static function getDataset() {
        return self::$server_list;
    }

    /**
     * @param int $id server id to fetch
     * @param bool $full true to return passwords for server connections. CAREFUL!
     * @return array
     */
    public static function get($id, $full) {
        $server = self::repo_get($id);
        if ($server && ! $full) {
            return self::clean($server);
        }
        return $server;
    }

    /*
     * @param array $server
     * @return array
     */
    public static function clean($server) {
        if (! array_key_exists('pass', $server) || ! $server['pass']) {
            $server['nopass'] = true;
        }
        unset($server['pass']);
        return $server;
    }

    /**
     * Fetch a server by the username and servername
     * @param string $username the user associated with the server
     * @param string $servername the host associated with the server
     * @return array|false
     */
    public static function fetch($username, $servername) {
        foreach (self::$server_list as $id => $server) {
            if (self::match($server, $username, $servername)) {
                if (array_key_exists('pass', $server)) {
                    unset($server['pass']);
                }
                $server['id'] = $id;
                return $server;
            }
        }
        return false;
    }

    /**
     * @param array $server server details
     * @param string $user username
     * @param string $name server name
     * @return boolean
     */
    private static function match($server, $user, $name) {
        if (array_key_exists('user', $server) && array_key_exists('server', $server)) {
            if ($user == $server['user'] && $name == $server['server']) {
                return true;
            }
        }
        return false;
    }

    private static function appendPasswordAndUsername(array $server) {
        $server['password'] = $server['pass'];
        $server['username'] = $server['user'];
        return $server;
    }

    public static function getForMailbox($id) {
        $server = self::get($id, true);
        if ($server) {
            return self::appendPasswordAndUsername($server);
        }
        return false;
    }

    public static function dumpForMailbox($id = false) {
        $list = self::dump($id, true);
        if ($id !== false) {
            return self::appendPasswordAndUsername($list);
        }
        foreach ($list as $index => $server) {
            $server = self::appendPasswordAndUsername($server);
            $list[$index] = $server;
        }
        return $list;
    }
}
