local M = {}

function M.is_import_string(string)
    return string:sub(1, 6) == 'import' or string:sub(1, 5) == 'const'
end

function M.normalize_string(raw_string)
    local string = raw_string

    -- pad with semicolon
    if string:sub(-1, -1) ~= ';' then
        string = string .. ';'
    end

    -- stylua: ignore
    string = string
        :match('.-;') -- extract first statement
        :gsub(' as %S+', '') -- remove aliases
        :gsub("'", '"') -- swap single for double quotes
        :gsub('%s+', '') -- remove whitespace

    return string
end

function M.parse_data(chunk)
    local ok, json = pcall(vim.fn.json_decode, chunk)

    if not ok or not json then
        return
    end

    return json.data
end

function M.is_ok(data)
    return data and data.size
end

return M