local json   = require 'json'
local parser = require 'parser'

local ErrorCodes = {
	-- Defined by JSON RPC
	ParseError           = -32700,
	InvalidRequest       = -32600,
	MethodNotFound       = -32601,
	InvalidParams        = -32602,
	InternalError        = -32603,
	serverErrorStart     = -32099,
	serverErrorEnd       = -32000,
	ServerNotInitialized = -32002,
	UnknownErrorCode     = -32001,

	-- Defined by the protocol.
	RequestCancelled     = -32800,
}

local mt = {}
mt.__index = mt

mt._input = nil
mt._output = nil
mt._method = nil
mt._file = nil

function mt:_callback(method, params)
    local f = self._method
    if f then
        return f(method, params)
    end
    return nil, '没有注册method'
end

function mt:_send(data)
    local f = self._output
    if not f then
        return
    end
    data.jsonrpc = '2.0'
    local content = json.encode(data)
    local buf = ('Content-Length: %d\r\n\r\n%s'):format(#content, content)
    f(buf)
end

function mt:_readProtoHead()
    local header = self:read 'l'
    if not header then
        return
    end
    if header:sub(1, #'Content-Length') == 'Content-Length' then
        return header
    elseif header:sub(1, #'Content-Type') == 'Content-Type' then
    else
        log.error('错误的协议头：', header)
    end
end

function mt:_readProtoContent(header)
    local len = tonumber(header:match('%d+'))
    if not len then
        log.error('错误的协议头：', header)
        return
    end
    local buf = self:read(len+2)
    if not buf then
        return
    end
    local suc, res = pcall(json.decode, buf)
    if not suc then
        log.error('错误的协议：', buf)
        return
    end
    local id     = res.id
    local method = res.method
    local params = res.params
    local response, err = self:_callback(method, params)
    if id then
        if response then
            self:_send {
                id = id,
                result = response,
            }
        else
            self:_send {
                id = id,
                error = {
                    code = ErrorCodes.UnknownErrorCode,
                    message = err or ('没有回应：' .. method),
                },
            }
        end
    end
    if response == nil then
        log.error(err or ('没有回应：' .. method))
    end
    return true
end

function mt:_buildTextCache()
    if not next(self._need_compile) then
        return
    end
    local list = {}
    for uri in pairs(self._need_compile) do
        list[#list+1] = uri
    end

    local size = 0
    local clock = os.clock()
    for _, uri in ipairs(list) do
        local obj = self:compileText(uri)
        if obj then
            size = size + #obj.text
        end
    end
    local passed = os.clock() - clock
    log.debug(('\n\z
    语法树缓存完成\n\z
    耗时：[%.3f]秒\n\z
    数量：[%d]\n\z
    总大小：[%.3f]kb\n\z
    速度：[%.3f]kb/s\n\z
    内存：[%.3f]kb'):format(
        passed,
        #list,
        size / 1000,
        size / passed / 1000,
        collectgarbage 'count'
    ))
end

function mt:setInput(input)
    self._input = input
end

function mt:setOutput(output)
    self._output = output
end

function mt:read(mode)
    if not self._input then
        return nil
    end
    return self._input(mode)
end

function mt:saveText(uri, version, text)
    local obj = self._file[uri]
    if obj then
        if obj.version >= version then
            return
        end
        obj.version = version
        obj.text = text
        self._need_compile[uri] = true
    else
        self._file[uri] = {
            version = version,
            text = text,
        }
        self._need_compile[uri] = true
    end
end

function mt:loadText(uri)
    local obj = self._file[uri]
    if not obj then
        return nil
    end
    self:compileText(uri)
    return obj.ast, obj.lines
end

function mt:compileText(uri)
    local obj = self._file[uri]
    if not obj then
        return nil
    end
    if not self._need_compile[uri] then
        return nil
    end
    self._need_compile[uri] = nil
    local ast, err = parser:ast(obj.text)
    if not ast then
        return nil
    end
    obj.ast   = ast
    obj.lines = parser:lines(obj.text)
    return obj
end

function mt:removeText(uri)
    self._file[uri] = nil
    self._need_compile[uri] = nil
end

function mt:setMethod(method)
    self._method = method
end

function mt:runStep()
    if not self._header then
        -- 如果没有协议头，则尝试读一条协议头
        self._header = self:_readProtoHead()
    end
    if self._header then
        -- 如果有协议头，则读取协议内容
        local suc = self:_readProtoContent(self._header)
        if suc then
            -- 协议内容读取成功后重置
            self._header = nil
            self._idle_clock = os.clock()
        end
        return
    end
    if os.clock() - self._idle_clock >= 1 then
        self:_buildTextCache()
    end
end

function mt:stop()
    self._input = nil
    self._output = nil
end

return function ()
    return setmetatable({
        _file = {},
        _need_compile = {},
        _header = nil,
        _idle_clock = os.clock(),
    }, mt)
end