<?php


namespace ZM\Store;


use Exception;
use Swoole\Table;
use ZM\Annotation\Swoole\OnSave;
use ZM\Console\Console;
use ZM\Event\EventDispatcher;
use ZM\Exception\ZMException;
use ZM\Framework;
use ZM\Utils\ProcessManager;

class LightCache
{
    /** @var Table|null */
    private static $kv_table = null;

    private static $config = [];

    public static $last_error = '';

    /**
     * @param $config
     * @return bool|mixed
     * @throws Exception
     * @noinspection PhpMissingReturnTypeInspection
     */
    public static function init($config) {
        self::$config = $config;
        self::$kv_table = new Table($config["size"], $config["hash_conflict_proportion"]);
        self::$kv_table->column("value", Table::TYPE_STRING, $config["max_strlen"]);
        self::$kv_table->column("expire", Table::TYPE_INT);
        self::$kv_table->column("data_type", Table::TYPE_STRING, 8);
        $result = self::$kv_table->create();
        // 加载内容
        if ($result === true && isset($config["persistence_path"])) {
            if (file_exists($config["persistence_path"])) {
                $r = json_decode(file_get_contents($config["persistence_path"]), true);
                if ($r === null) $r = [];
                foreach ($r as $k => $v) {
                    $write = self::set($k, $v);
                    Console::verbose("Writing LightCache: " . $k);
                    if ($write === false) {
                        self::$last_error = '可能是由于 Hash 冲突过多导致动态空间无法分配内存';
                        return false;
                    }
                }
            }
        }
        if ($result === false) {
            self::$last_error = '系统内存不足，申请失败';
        } else {
            $obj = Framework::loadFrameworkState();
            foreach(($obj["expiring_light_cache"] ?? []) as $k => $v) {
                $value = $v["value"];
                if (is_array($value)) {
                    $value = json_encode($value, JSON_UNESCAPED_UNICODE);
                    if (strlen($value) >= self::$config["max_strlen"]) return false;
                    $data_type = "json";
                } elseif (is_string($value)) {
                    $data_type = "";
                } elseif (is_int($value)) {
                    $data_type = "int";
                } elseif (is_bool($value)) {
                    $data_type = "bool";
                    $value = json_encode($value);
                } else {
                    return false;
                }
                $result = self::$kv_table->set($k, [
                    "value" => $value,
                    "expire" => $v["expire"],
                    "data_type" => $data_type
                ]);
                if ($result === false) return false;
            }
        }
        return $result;
    }

    /**
     * @param string $key
     * @return null|mixed
     * @throws ZMException
     */
    public static function get(string $key) {
        if (self::$kv_table === null) throw new ZMException("not initialized LightCache");
        self::checkExpire($key);
        $r = self::$kv_table->get($key);
        return $r === false ? null : self::parseGet($r);
    }

    /**
     * @param string $key
     * @return mixed|null
     * @throws ZMException
     */
    public static function getExpire(string $key) {
        if (self::$kv_table === null) throw new ZMException("not initialized LightCache");
        self::checkExpire($key);
        $r = self::$kv_table->get($key, "expire");
        return $r === false ? null : $r - time();
    }

    /**
     * @param string $key
     * @return mixed|null
     * @throws ZMException
     * @since 2.4.3
     */
    public static function getExpireTS(string $key) {
        if (self::$kv_table === null) throw new ZMException("not initialized LightCache");
        self::checkExpire($key);
        $r = self::$kv_table->get($key, "expire");
        return $r === false ? null : $r;
    }

    /**
     * @param string $key
     * @param string|array|int $value
     * @param int $expire
     * @return mixed
     * @throws ZMException
     * @noinspection PhpMissingReturnTypeInspection
     */
    public static function set(string $key, $value, int $expire = -1) {
        if (self::$kv_table === null) throw new ZMException("not initialized LightCache");
        if (is_array($value)) {
            $value = json_encode($value, JSON_UNESCAPED_UNICODE);
            if (strlen($value) >= self::$config["max_strlen"]) return false;
            $data_type = "json";
        } elseif (is_string($value)) {
            $data_type = "";
        } elseif (is_int($value)) {
            $data_type = "int";
        } elseif (is_bool($value)) {
            $data_type = "bool";
            $value = json_encode($value);
        } else {
            throw new ZMException("Only can set string, array and int");
        }
        try {
            return self::$kv_table->set($key, [
                "value" => $value,
                "expire" => $expire >= 0 ? $expire + time() : $expire,
                "data_type" => $data_type
            ]);
        } catch (Exception $e) {
            return false;
        }
    }

    /**
     * @param string $key
     * @param $value
     * @return bool|mixed
     * @throws ZMException
     * @noinspection PhpMissingReturnTypeInspection
     */
    public static function update(string $key, $value) {
        if (self::$kv_table === null) throw new ZMException("not initialized LightCache.");
        if (is_array($value)) {
            $value = json_encode($value, JSON_UNESCAPED_UNICODE);
            if (strlen($value) >= self::$config["max_strlen"]) return false;
            $data_type = "json";
        } elseif (is_string($value)) {
            $data_type = "";
        } elseif (is_int($value)) {
            $data_type = "int";
        } else {
            throw new ZMException("Only can set string, array and int");
        }
        try {
            if (self::$kv_table->get($key) === false) return false;
            return self::$kv_table->set($key, [
                "value" => $value,
                "data_type" => $data_type
            ]);
        } catch (Exception $e) {
            return false;
        }
    }

    public static function getMemoryUsage() {
        return self::$kv_table->getMemorySize();
    }

    /**
     * @param string $key
     * @return bool
     * @throws Exception
     */
    public static function isset(string $key): bool {
        return self::get($key) !== null;
    }

    public static function unset(string $key) {
        return self::$kv_table->del($key);
    }

    public static function getAll(): array {
        $r = [];
        $del = [];
        foreach (self::$kv_table as $k => $v) {
            if ($v["expire"] <= time() && $v["expire"] >= 0) {
                $del[] = $k;
                continue;
            }
            $r[$k] = self::parseGet($v);
        }
        foreach ($del as $v) {
            self::unset($v);
        }
        return $r;
    }

    public static function addPersistence($key) {
        if (file_exists(self::$config["persistence_path"])) {
            $r = json_decode(file_get_contents(self::$config["persistence_path"]), true);
            if ($r === null) $r = [];
            if (!isset($r[$key])) $r[$key] = null;
            file_put_contents(self::$config["persistence_path"], json_encode($r, 64 | 128 | 256));
            return true;
        } else {
            return false;
        }
    }

    public static function removePersistence($key) {
        if (file_exists(self::$config["persistence_path"])) {
            $r = json_decode(file_get_contents(self::$config["persistence_path"]), true);
            if ($r === null) $r = [];
            if (isset($r[$key])) unset($r[$key]);
            file_put_contents(self::$config["persistence_path"], json_encode($r, 64 | 128 | 256));
            return true;
        } else {
            return false;
        }
    }

    /**
     * 这个只能在唯一一个工作进程中执行
     * @throws Exception
     */
    public static function savePersistence() {
        if (server()->worker_id !== MAIN_WORKER) {
            ProcessManager::sendActionToWorker(MAIN_WORKER, "save_persistence", []);
            return;
        }
        $dispatcher = new EventDispatcher(OnSave::class);
        $dispatcher->dispatchEvents();

        if (self::$kv_table === null) return;

        if (!empty(self::$config["persistence_path"])) {
            if (file_exists(self::$config["persistence_path"])) {
                $r = json_decode(file_get_contents(self::$config["persistence_path"]), true);
            } else {
                $r = [];
            }
            if ($r === null) $r = [];
            foreach ($r as $k => $v) {
                Console::verbose("Saving " . $k);
                $r[$k] = self::get($k);
            }
            file_put_contents(self::$config["persistence_path"], json_encode($r, 64 | 128 | 256));
        }

        $obj = Framework::loadFrameworkState();
        $obj["expiring_light_cache"] = [];
        $del = [];
        foreach (self::$kv_table as $k => $v) {
            if ($v["expire"] <= time() && $v["expire"] >= 0) {
                $del[] = $k;
                continue;
            } elseif ($v["expire"] > time()) {
                $obj["expiring_light_cache"][$k] = [
                    "expire" => $v["expire"],
                    "value" => self::parseGet($v)
                ];
            }
        }
        foreach ($del as $v) {
            self::unset($v);
        }
        Framework::saveFrameworkState($obj);
        Console::verbose("Saved.");
    }

    private static function checkExpire($key) {
        if (($expire = self::$kv_table->get($key, "expire")) >= 0) {
            if ($expire <= time()) {
                self::$kv_table->del($key);
            }
        }
    }

    private static function parseGet($r) {
        switch ($r["data_type"]) {
            case "json":
            case "int":
            case "bool":
                return json_decode($r["value"], true);
            case "":
            default:
                return $r["value"];
        }
    }
}