dcs-lua-runner/src/hooks/dcs-fiddle-server.lua

-- Note: This is a modified version of the DCS Fiddle Server to add authentication for public servers.
-- All credit goes to the original authors of DCS Fiddle: JonathanTurnock and john681611

FIDDLE = {}

-- Configs:
FIDDLE.PORT = 12080         -- keep this at 12080 if you also want to use the DCS Fiddle website.
FIDDLE.AUTH = true          -- set to true to enable basic auth, recommended for public servers.
FIDDLE.USERNAME = 'username'
FIDDLE.PASSWORD = 'password'
FIDDLE.BIND_IP = '0.0.0.0'  -- for remote access
FIDDLE.BYPASS_LOCAL = true  -- allow requests to 127.0.0.1:12080 without auth, so DCS Fiddle website can still work. (Not a very secure implementation. Use at your own risk if your 12080 port is public)
--[[
Custom JSON Serialization Module that handles DCS' usage of integers as table keys such as {[1]=1, ["name"]="Enfield11", [2]=1, [3]=1} which is not valid json

This is handled by encoding tables with mixed key types as objects so the above example would be

{
  "_1": 1,
  "_2": 1,
  "name": "Enfield11",
  "_3": 1
}

Anything that is clearly a table is still converted to a table

]]--

fiddlejson = { _version = "0.2.0" }

-------------------------------------------------------------------------------
-- Encode
-------------------------------------------------------------------------------

local encode

local escape_char_map = {
    [ "\\" ] = "\\",
    [ "\"" ] = "\"",
    [ "\b" ] = "b",
    [ "\f" ] = "f",
    [ "\n" ] = "n",
    [ "\r" ] = "r",
    [ "\t" ] = "t",
}

local escape_char_map_inv = { [ "/" ] = "/" }
for k, v in pairs(escape_char_map) do
    escape_char_map_inv[v] = k
end


local function escape_char(c)
    return "\\" .. (escape_char_map[c] or string.format("u%04x", c:byte()))
end


local function encode_nil(val)
    return "null"
end

--- Checks if a given table is an array.
-- The function verifies that all keys are numeric and sequential starting from 1.
-- @param val Table to be checked.
-- @return boolean Returns true if the table is an array, otherwise false.
local function is_table_array(val)
    -- Check if the first element of the table is not nil or the table is empty.
    -- These are two signs that the table might be an array.
    if rawget(val, 1) ~= nil or next(val) == nil then
        -- Extract all numeric keys and sort them.
        local keys = {}
        for k in pairs(val) do
            if type(k) == "number" then
                table.insert(keys, k)
            else
                -- If any key is not a number, the table is not an array.
                return false
            end
        end
        table.sort(keys)

        -- Check the sorted keys to see if they form a continuous sequence starting from 1.
        for i, k in ipairs(keys) do
            if i ~= k then
                return false
            end
        end

        -- If all keys were numeric and in sequence, then the table is an array.
        return true
    else
        -- If the first element of the table is nil and the table isn't empty,
        -- then it's not an array.
        return false
    end
end


local function encode_table(val, stack)
    local res = {}
    stack = stack or {}

    -- Circular reference?
    if stack[val] then error("circular reference") end

    stack[val] = true



    if is_table_array(val) then
        -- Treat as array -- check keys are valid and it is not sparse
        -- Encode
        for i, v in ipairs(val) do
            table.insert(res, encode(v, stack))
        end
        stack[val] = nil
        return "[" .. table.concat(res, ",") .. "]"

    else
        -- Treat as an object
        for k, v in pairs(val) do
            if type(k) ~= "string" then
                table.insert(res, encode("_"..k, stack) .. ":" .. encode(v, stack))
            else
                table.insert(res, encode(k, stack) .. ":" .. encode(v, stack))
            end
        end
        stack[val] = nil
        return "{" .. table.concat(res, ",") .. "}"
    end
end


local function encode_string(val)
    return '"' .. val:gsub('[%z\1-\31\\"]', escape_char) .. '"'
end


local function encode_number(val)
    -- Check for NaN, -inf and inf
    if val ~= val or val <= -math.huge or val >= math.huge then
        error("unexpected number value '" .. tostring(val) .. "'")
    end
    return string.format("%.14g", val)
end


local type_func_map = {
    [ "nil"     ] = encode_nil,
    [ "table"   ] = encode_table,
    [ "string"  ] = encode_string,
    [ "number"  ] = encode_number,
    [ "boolean" ] = tostring,
}


encode = function(val, stack)
    local t = type(val)
    local f = type_func_map[t]
    if f then
        return f(val, stack)
    end
    error("unexpected type '" .. t .. "'")
end


function fiddlejson.encode(val)
    return ( encode(val) )
end


-------------------------------------------------------------------------------
-- Decode
-------------------------------------------------------------------------------

local parse

local function create_set(...)
    local res = {}
    for i = 1, select("#", ...) do
        res[ select(i, ...) ] = true
    end
    return res
end

local space_chars   = create_set(" ", "\t", "\r", "\n")
local delim_chars   = create_set(" ", "\t", "\r", "\n", "]", "}", ",")
local escape_chars  = create_set("\\", "/", '"', "b", "f", "n", "r", "t", "u")
local literals      = create_set("true", "false", "null")

local literal_map = {
    [ "true"  ] = true,
    [ "false" ] = false,
    [ "null"  ] = nil,
}


local function next_char(str, idx, set, negate)
    for i = idx, #str do
        if set[str:sub(i, i)] ~= negate then
            return i
        end
    end
    return #str + 1
end


local function decode_error(str, idx, msg)
    local line_count = 1
    local col_count = 1
    for i = 1, idx - 1 do
        col_count = col_count + 1
        if str:sub(i, i) == "\n" then
            line_count = line_count + 1
            col_count = 1
        end
    end
    error( string.format("%s at line %d col %d", msg, line_count, col_count) )
end


local function codepoint_to_utf8(n)
    -- http://scripts.sil.org/cms/scripts/page.php?site_id=nrsi&id=iws-appendixa
    local f = math.floor
    if n <= 0x7f then
        return string.char(n)
    elseif n <= 0x7ff then
        return string.char(f(n / 64) + 192, n % 64 + 128)
    elseif n <= 0xffff then
        return string.char(f(n / 4096) + 224, f(n % 4096 / 64) + 128, n % 64 + 128)
    elseif n <= 0x10ffff then
        return string.char(f(n / 262144) + 240, f(n % 262144 / 4096) + 128,
                f(n % 4096 / 64) + 128, n % 64 + 128)
    end
    error( string.format("invalid unicode codepoint '%x'", n) )
end


local function parse_unicode_escape(s)
    local n1 = tonumber( s:sub(1, 4),  16 )
    local n2 = tonumber( s:sub(7, 10), 16 )
    -- Surrogate pair?
    if n2 then
        return codepoint_to_utf8((n1 - 0xd800) * 0x400 + (n2 - 0xdc00) + 0x10000)
    else
        return codepoint_to_utf8(n1)
    end
end


local function parse_string(str, i)
    local res = ""
    local j = i + 1
    local k = j

    while j <= #str do
        local x = str:byte(j)

        if x < 32 then
            decode_error(str, j, "control character in string")

        elseif x == 92 then -- `\`: Escape
            res = res .. str:sub(k, j - 1)
            j = j + 1
            local c = str:sub(j, j)
            if c == "u" then
                local hex = str:match("^[dD][89aAbB]%x%x\\u%x%x%x%x", j + 1)
                        or str:match("^%x%x%x%x", j + 1)
                        or decode_error(str, j - 1, "invalid unicode escape in string")
                res = res .. parse_unicode_escape(hex)
                j = j + #hex
            else
                if not escape_chars[c] then
                    decode_error(str, j - 1, "invalid escape char '" .. c .. "' in string")
                end
                res = res .. escape_char_map_inv[c]
            end
            k = j + 1

        elseif x == 34 then -- `"`: End of string
            res = res .. str:sub(k, j - 1)
            return res, j + 1
        end

        j = j + 1
    end

    decode_error(str, i, "expected closing quote for string")
end


local function parse_number(str, i)
    local x = next_char(str, i, delim_chars)
    local s = str:sub(i, x - 1)
    local n = tonumber(s)
    if not n then
        decode_error(str, i, "invalid number '" .. s .. "'")
    end
    return n, x
end


local function parse_literal(str, i)
    local x = next_char(str, i, delim_chars)
    local word = str:sub(i, x - 1)
    if not literals[word] then
        decode_error(str, i, "invalid literal '" .. word .. "'")
    end
    return literal_map[word], x
end


local function parse_array(str, i)
    local res = {}
    local n = 1
    i = i + 1
    while 1 do
        local x
        i = next_char(str, i, space_chars, true)
        -- Empty / end of array?
        if str:sub(i, i) == "]" then
            i = i + 1
            break
        end
        -- Read token
        x, i = parse(str, i)
        res[n] = x
        n = n + 1
        -- Next token
        i = next_char(str, i, space_chars, true)
        local chr = str:sub(i, i)
        i = i + 1
        if chr == "]" then break end
        if chr ~= "," then decode_error(str, i, "expected ']' or ','") end
    end
    return res, i
end

function set_table(res, key, val)
    -- Check if the key starts with an underscore
    if string.sub(key, 1, 1) == "_" then
        -- Remove the underscore and try to convert the rest to a number
        local numKey = tonumber(string.sub(key, 2))

        -- If the result is not nil, meaning the conversion was successful
        if numKey then
            -- Update the table with the number key instead of the string key
            res[numKey] = val
        end
    else
        -- If the key does not start with an underscore, just use it as is
        res[key] = val
    end
end

local function parse_object(str, i)
    local res = {}
    i = i + 1
    while 1 do
        local key, val
        i = next_char(str, i, space_chars, true)
        -- Empty / end of object?
        if str:sub(i, i) == "}" then
            i = i + 1
            break
        end
        -- Read key
        if str:sub(i, i) ~= '"' then
            decode_error(str, i, "expected string for key")
        end
        key, i = parse(str, i)
        -- Read ':' delimiter
        i = next_char(str, i, space_chars, true)
        if str:sub(i, i) ~= ":" then
            decode_error(str, i, "expected ':' after key")
        end
        i = next_char(str, i + 1, space_chars, true)
        -- Read value
        val, i = parse(str, i)
        -- Set
        --res[key] = val
        set_table(res, key, val)
        -- Next token
        i = next_char(str, i, space_chars, true)
        local chr = str:sub(i, i)
        i = i + 1
        if chr == "}" then break end
        if chr ~= "," then decode_error(str, i, "expected '}' or ','") end
    end
    return res, i
end


local char_func_map = {
    [ '"' ] = parse_string,
    [ "0" ] = parse_number,
    [ "1" ] = parse_number,
    [ "2" ] = parse_number,
    [ "3" ] = parse_number,
    [ "4" ] = parse_number,
    [ "5" ] = parse_number,
    [ "6" ] = parse_number,
    [ "7" ] = parse_number,
    [ "8" ] = parse_number,
    [ "9" ] = parse_number,
    [ "-" ] = parse_number,
    [ "t" ] = parse_literal,
    [ "f" ] = parse_literal,
    [ "n" ] = parse_literal,
    [ "[" ] = parse_array,
    [ "{" ] = parse_object,
}


parse = function(str, idx)
    local chr = str:sub(idx, idx)
    local f = char_func_map[chr]
    if f then
        return f(str, idx)
    end
    decode_error(str, idx, "unexpected character '" .. chr .. "'")
end


function fiddlejson.decode(str)
    if type(str) ~= "string" then
        error("expected argument of type string, got " .. type(str))
    end
    local res, idx = parse(str, next_char(str, 1, space_chars, true))
    idx = next_char(str, idx, space_chars, true)
    if idx <= #str then
        decode_error(str, idx, "trailing garbage")
    end
    return res
end

--[[
 base64 -- v1.5.3 public domain Lua base64 encoder/decoder
 no warranty implied; use at your own risk
 Needs bit32.extract function. If not present it's implemented using BitOp
 or Lua 5.3 native bit operators. For Lua 5.1 fallbacks to pure Lua
 implementation inspired by Rici Lake's post:
   http://ricilake.blogspot.co.uk/2007/10/iterating-bits-in-lua.html
 author: Ilya Kolbin (iskolbin@gmail.com)
 url: github.com/iskolbin/lbase64
 COMPATIBILITY
 Lua 5.1+, LuaJIT
 LICENSE
 See end of file for license information.
--]]

local base64 = {}

local extract = _G.bit32 and _G.bit32.extract -- Lua 5.2/Lua 5.3 in compatibility mode
if not extract then
    if _G.bit then
        -- LuaJIT
        local shl, shr, band = _G.bit.lshift, _G.bit.rshift, _G.bit.band
        extract = function(v, from, width)
            return band(shr(v, from), shl(1, width) - 1)
        end
    elseif _G._VERSION == "Lua 5.1" then
        extract = function(v, from, width)
            local w = 0
            local flag = 2 ^ from
            for i = 0, width - 1 do
                local flag2 = flag + flag
                if v % flag2 >= flag then
                    w = w + 2 ^ i
                end
                flag = flag2
            end
            return w
        end
    else
        -- Lua 5.3+
        extract = load [[return function( v, from, width )
			return ( v >> from ) & ((1 << width) - 1)
		end]]()
    end
end

function base64.makeencoder(s62, s63, spad)
    local encoder = {}
    for b64code, char in pairs { [0] = 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J',
                                 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y',
                                 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
                                 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2',
                                 '3', '4', '5', '6', '7', '8', '9', s62 or '+', s63 or '/', spad or '=' } do
        encoder[b64code] = char:byte()
    end
    return encoder
end

function base64.makedecoder(s62, s63, spad)
    local decoder = {}
    for b64code, charcode in pairs(base64.makeencoder(s62, s63, spad)) do
        decoder[charcode] = b64code
    end
    return decoder
end

local DEFAULT_ENCODER = base64.makeencoder()
local DEFAULT_DECODER = base64.makedecoder()

local char, concat = string.char, table.concat

function base64.encode(str, encoder, usecaching)
    encoder = encoder or DEFAULT_ENCODER
    local t, k, n = {}, 1, #str
    local lastn = n % 3
    local cache = {}
    for i = 1, n - lastn, 3 do
        local a, b, c = str:byte(i, i + 2)
        local v = a * 0x10000 + b * 0x100 + c
        local s
        if usecaching then
            s = cache[v]
            if not s then
                s = char(encoder[extract(v, 18, 6)], encoder[extract(v, 12, 6)], encoder[extract(v, 6, 6)], encoder[extract(v, 0, 6)])
                cache[v] = s
            end
        else
            s = char(encoder[extract(v, 18, 6)], encoder[extract(v, 12, 6)], encoder[extract(v, 6, 6)], encoder[extract(v, 0, 6)])
        end
        t[k] = s
        k = k + 1
    end
    if lastn == 2 then
        local a, b = str:byte(n - 1, n)
        local v = a * 0x10000 + b * 0x100
        t[k] = char(encoder[extract(v, 18, 6)], encoder[extract(v, 12, 6)], encoder[extract(v, 6, 6)], encoder[64])
    elseif lastn == 1 then
        local v = str:byte(n) * 0x10000
        t[k] = char(encoder[extract(v, 18, 6)], encoder[extract(v, 12, 6)], encoder[64], encoder[64])
    end
    return concat(t)
end

function base64.decode(b64, decoder, usecaching)
    decoder = decoder or DEFAULT_DECODER
    local pattern = '[^%w%+%/%=]'
    if decoder then
        local s62, s63
        for charcode, b64code in pairs(decoder) do
            if b64code == 62 then
                s62 = charcode
            elseif b64code == 63 then
                s63 = charcode
            end
        end
        pattern = ('[^%%w%%%s%%%s%%=]'):format(char(s62), char(s63))
    end
    b64 = b64:gsub(pattern, '')
    local cache = usecaching and {}
    local t, k = {}, 1
    local n = #b64
    local padding = b64:sub(-2) == '==' and 2 or b64:sub(-1) == '=' and 1 or 0
    for i = 1, padding > 0 and n - 4 or n, 4 do
        local a, b, c, d = b64:byte(i, i + 3)
        local s
        if usecaching then
            local v0 = a * 0x1000000 + b * 0x10000 + c * 0x100 + d
            s = cache[v0]
            if not s then
                local v = decoder[a] * 0x40000 + decoder[b] * 0x1000 + decoder[c] * 0x40 + decoder[d]
                s = char(extract(v, 16, 8), extract(v, 8, 8), extract(v, 0, 8))
                cache[v0] = s
            end
        else
            local v = decoder[a] * 0x40000 + decoder[b] * 0x1000 + decoder[c] * 0x40 + decoder[d]
            s = char(extract(v, 16, 8), extract(v, 8, 8), extract(v, 0, 8))
        end
        t[k] = s
        k = k + 1
    end
    if padding == 1 then
        local a, b, c = b64:byte(n - 3, n - 1)
        local v = decoder[a] * 0x40000 + decoder[b] * 0x1000 + decoder[c] * 0x40
        t[k] = char(extract(v, 16, 8), extract(v, 8, 8))
    elseif padding == 2 then
        local a, b = b64:byte(n - 3, n - 2)
        local v = decoder[a] * 0x40000 + decoder[b] * 0x1000
        t[k] = char(extract(v, 16, 8))
    end
    return concat(t)
end

--[[
------------------------------------------------------------------------------
This software is available under 2 licenses -- choose whichever you prefer.
------------------------------------------------------------------------------
ALTERNATIVE A - MIT License
Copyright (c) 2018 Ilya Kolbin
Permission is hereby granted, free of charge, to any person obtaining a copy of
this software and associated documentation files (the "Software"), to deal in
the Software without restriction, including without limitation the rights to
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
of the Software, and to permit persons to whom the Software is furnished to do
so, subject to the following conditions:
The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.
------------------------------------------------------------------------------
ALTERNATIVE B - Public Domain (www.unlicense.org)
This is free and unencumbered software released into the public domain.
Anyone is free to copy, modify, publish, use, compile, sell, or distribute this
software, either in source code form or as a compiled binary, for any purpose,
commercial or non-commercial, and by any means.
In jurisdictions that recognize copyright laws, the author or authors of this
software dedicate any and all copyright interest in the software to the public
domain. We make this dedication for the benefit of the public at large and to
the detriment of our heirs and successors. We intend this dedication to be an
overt act of relinquishment in perpetuity of all present and future rights to
this software under copyright law.
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
------------------------------------------------------------------------------
--]]

local function dumpt(t)
    if type(t) == 'table' then
        local s = '{ '
        for k, v in pairs(t) do
            if type(k) ~= 'number' then
                k = '"' .. k .. '"'
            end
            s = s .. '[' .. k .. '] = ' .. dumpt(v) .. ','
        end
        return s .. '} '
    else
        return tostring(t)
    end
end


------------------------------------------------------------------------------------------------------------------------
--- Logger
------------------------------------------------------------------------------------------------------------------------

local __debug = function(message)
    local message = '[dcs-fiddle-server] - ' .. message
    if (log and log.debug) then
        log.debug(message)
    else
        print('DEBUG - ' .. message)
    end
end

local __info = function(message)
    local message = '[dcs-fiddle-server] - ' .. message
    if (log and log.info) then
        log.info(message)
    else
        print('INFO - ' .. message)
    end
end

local __error = function(message)
    local message = '[dcs-fiddle-server] - ' .. message
    if (log and log.error) then
        log.error(message)
    else
        print('ERROR - ' .. message)
    end
end

------------------------------------------------------------------------------------------------------------------------
--- DCS Instruction Handler
------------------------------------------------------------------------------------------------------------------------

local IS_DCS = false

------------------------------------------------------------------------------------------------------------------------
--- Takes a LUA string, executes it and returns the result as a JSON string
---@param env string Environment to run the lua string within
---
local function handle_request(luastring, env)
    __info("[handle_request] - Handling request to execute string in " .. env)

    if (env ~= "default") then
        __info("[handle_request] - Executing string via dostring_in")
        local str, err = net.dostring_in(env, luastring)
        if (err) then
            __error(string.format("Error while executing string in %s\n%s", env, str))
        end
        return str
    else
        __info("[handle_request] - Loading LUA String...")
        local loaded = assert(loadstring(luastring))

        __info("[handle_request] - Executing LUA String...")
        local result = loaded()

        __info("[handle_request] - Processing result...")
        return result
    end
end

------------------------------------------------------------------------------------------------------------------------
--- Url
--- https://developer.mozilla.org/en-US/docs/Learn/Common_questions/What_is_a_URL
------------------------------------------------------------------------------------------------------------------------

------------------------------------------------------------------------------------------------------------------------
--- Parses the given url and returns a URL table
---
--- `{ parameters={sort="asc", size=20}, path="/employees" }`
---
--- @param original_url string - The Original Request URL i.e. `/employees?sort=asc&size=20`
--- @return string, table Returns the path part alongside a table of parsed parameters
---
local function parse_url(original_url)
    local resource_path, parameters = original_url:match('(.+)?(.*)')
    if (parameters) then
        local params = {}
        for parameter in string.gmatch(parameters, "[^&]+") do
            local name, value = parameter:match('(.+)=(.+)')
            params[name] = value
        end

        return resource_path, params
    end
    return original_url
end

------------------------------------------------------------------------------------------------------------------------
--- HTTP Receiver
------------------------------------------------------------------------------------------------------------------------

------------------------------------------------------------------------------------------------------------------------
--- Reads HTTP Message from the given connection
---
--- @see https://developer.mozilla.org/en-US/docs/Web/HTTP/Messages
--- @see https://lunarmodules.github.io/luasocket/tcp.html#accept
--- @param client socket.client LUA Socket Lib Client
--- @return table, number Request table containing method, original_url, protocol, path, parameters, body, headers. Optionally returns a second item representing an error_code from the match_headers failing
local function receive_http(client)
    local request = {}
    __debug("receiving start-line")
    local received, err = client:receive("*l")

    if (err) then
        __error("Failed to get start-line due to " .. err)
        return
    end

    __debug("parsing start-line")
    local method, original_url, protocol = string.match(received, "(%S+) (%S+) (%S+)")
    request.method = method
    request.original_url = original_url
    request.protocol = protocol

    __debug("parsing url")
    local path, parameters = parse_url(original_url)
    request.path = path
    request.parameters = parameters

    request.authOK = false
    if FIDDLE.AUTH then
        __debug("parsing headers for auth")
        local headers = {}
        while true do
            local line = client:receive()
            if not line or line == "" then break end
            local name, value = line:match("^([^:]+):%s*(.*)$")
            headers[name] = value
        end
        local auth_header = headers["Authorization"]
        local host_header = headers['X-Forwarded-Host'] or headers["Host"]
        if FIDDLE.BYPASS_LOCAL and (host_header == '127.0.0.1:12080' or host_header == '127.0.0.1:12081') then
            request.authOK = true
        elseif auth_header then
            local encoded_username_password = auth_header:match("^Basic%s+(.*)$")
            if encoded_username_password then
                local decoded_username_password = base64.decode(encoded_username_password)
                local username, password = decoded_username_password:match("^(.*):(.*)$")
                -- __info(decoded_username_password)
                if username == FIDDLE.USERNAME and password == FIDDLE.PASSWORD then
                    -- __info("Auth Passed")
                    request.authOK = true
                end
            end
        end
    else
        request.authOK = true
    end

    __debug("request completed")
    return request
end

------------------------------------------------------------------------------------------------------------------------
--- HTTP Sender
------------------------------------------------------------------------------------------------------------------------

local EMPTY_LINE = ""
local CRLF = "\r\n"

local status_text = {
    [100] = "Continue",
    [101] = "Switching protocols",
    [102] = "Processing",
    [103] = "Early Hints",
    [200] = "OK",
    [201] = "Created",
    [202] = "Accepted",
    [203] = "Non-Authoritative Information",
    [204] = "No Content",
    [205] = "Reset Content",
    [206] = "Partial Content",
    [207] = "Multi-Status",
    [208] = "Already Reported",
    [226] = "IM Used",
    [300] = "Multiple Choices",
    [301] = "Moved Permanently",
    [302] = "Found (Previously \"Moved Temporarily\")",
    [303] = "See Other",
    [304] = "Not Modified",
    [305] = "Use Proxy",
    [306] = "Switch Proxy",
    [307] = "Temporary Redirect",
    [308] = "Permanent Redirect",
    [400] = "Bad Request",
    [401] = "Unauthorized",
    [402] = "Payment Required",
    [403] = "Forbidden",
    [404] = "Not Found",
    [405] = "Method Not Allowed",
    [406] = "Not Acceptable",
    [407] = "Proxy Authentication Required",
    [408] = "Request Timeout",
    [409] = "Conflict",
    [410] = "Gone",
    [411] = "Length Required",
    [412] = "Precondition Failed",
    [413] = "Payload Too Large",
    [414] = "URI Too Long",
    [415] = "Unsupported Media Type",
    [416] = "Range Not Satisfiable",
    [417] = "Expectation Failed",
    [418] = "I'm a Teapot",
    [421] = "Misdirected Request",
    [422] = "Unprocessable Entity",
    [423] = "Locked",
    [424] = "Failed Dependency",
    [425] = "Too Early",
    [426] = "Upgrade Required",
    [428] = "Precondition Required",
    [429] = "Too Many Requests",
    [431] = "Request Header Fields Too Large",
    [451] = "Unavailable For Legal Reasons",
    [500] = "Internal Server Error",
    [501] = "Not Implemented",
    [502] = "Bad Gateway",
    [503] = "Service Unavailable",
    [504] = "Gateway Timeout",
    [505] = "HTTP Version Not Supported",
    [506] = "Variant Also Negotiates",
    [507] = "Insufficient Storage",
    [508] = "Loop Detected",
    [510] = "Not Extended",
    [511] = "Network Authentication Required"
}

------------------------------------------------------------------------------------------------------------------------
--- Writes HTTP Message to the given connection using the given response object
---
--- @see https://developer.mozilla.org/en-US/docs/Web/HTTP/Messages
--- @param client client @see https://lunarmodules.github.io/luasocket/tcp.html
--- @param response table response table containing 'status' and 'body'
local function send_http(client, response)
    local start_line = table.concat({ "HTTP/1.1", response.status, status_text[response.status] }, " ")

    local headers = { "Server: DCS Fiddle Server HTTP/1.1" }

    for name, value in pairs(response.headers) do
        table.insert(headers, name .. ": " .. value)
    end

    local response_string
    if (response.body) then
        response_string = table.concat({ start_line, table.concat(headers, CRLF), EMPTY_LINE, response.body }, CRLF)
    else
        response_string = table.concat({ start_line, table.concat(headers, CRLF), EMPTY_LINE, EMPTY_LINE }, CRLF)
    end

    __info("Sending HTTP Response")
    --__debug(">> " .. response_string)
    local index, err = client:send(response_string)
    if (err) then
        __error("Failed to fully send due to: " .. err)
    else
        __info("Successfully sent response")
    end
end

------------------------------------------------------------------------------------------------------------------------
--- HTTP Server
------------------------------------------------------------------------------------------------------------------------
if (not require or not package) then
    if (env and env.error) then
        env.error("DCS Fiddle failed to inject into the mission scripting environment as require or package was not found.\n\nPlease follow the installation docs to de-sanitize the mission scripting environment\nhttps://dcsfiddle.pages.dev/docs", true)
        return
    end
end

package.path = package.path .. ";.\\LuaSocket\\?.lua"
package.cpath = package.cpath .. ";.\\LuaSocket\\?.dll"

local socket = require("socket")

local clients = {}
local tcp_server

local server_config = { cors = "*" }

local client_id_seq = 1

local OK = 200

local BAD_REQUEST = 400

local INTERNAL_SERVER_ERROR = 500
local METHOD_NOT_ALLOWED = 405
local UNAUTHORIZED = 401

-----------------------------------------------------------------------------------------------------------------------
--- Gets and returns a client id incrementing the sequence
local function get_client_id()
    local id = client_id_seq
    client_id_seq = client_id_seq + 1
    return id
end

local function handle_client_connection(client)
    -- Dictionary of Headers that need to match, failure to match fails the read operation and returns the error code
    local response = { status = INTERNAL_SERVER_ERROR, headers = { ["Content-Type"] = "application/json"} }

    local request = receive_http(client)

    if (request) then
        if (request.method ~= "GET") then
            response.status = METHOD_NOT_ALLOWED
        elseif not request.authOK then
            __info("Request Unauthorized")
            response.status = UNAUTHORIZED
        else
            __info("Handling Request")
            local success, res = pcall(base64.decode, string.sub(request.path, 2))
            if (not success) then
                __error("Failed to read input due to " .. res)
                response.status = BAD_REQUEST
            else
                local env = request.parameters and request.parameters.env
                __info("Processing Command " .. res)
                local success, res = pcall(handle_request, res, env)
                if (not success) then
                    __error("Failed to handle request due to \n" .. res)
                    response.body = fiddlejson.encode({error=tostring(res)})
                    response.status = INTERNAL_SERVER_ERROR
                else
                    __info("Handled request")
                    response.body = fiddlejson.encode({result=res})
                    response.status = OK
                end
            end
        end
    end

    if (server_config.cors) then
        response.headers["Access-Control-Allow-Origin"] = server_config.cors
    end

    send_http(client, response)

    __info("Connection Completed")
    client:close()
end


local function create_server(address, port)
    tcp_server = socket.bind(address, port)
    tcp_server:settimeout(0) -- Make non blocking

    if not tcp_server then
        __error("Could not bind socket.")
    end

    local ip, port = tcp_server:getsockname()

    __info("HTTP Server running on " .. ip .. ":" .. port)

    --- Returns function which when called will perform 1 server loop
    --- Note this impl only allows 1 request to be handled at a time
    return function()
        local client = tcp_server:accept()
        if (client) then
            local success, res = handle_client_connection(client)
            if (not success) then
                __error("Failed to run client handler " .. res)
            else
                clients[id].receive_patten = res
            end
        end
    end
end

------------------------------------------------------------------------------------------------------------------------
------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------- MAIN ---------------------------------------------------------------

__info("Checking the DCS environment...")

local isMission = not DCS
local port =  FIDDLE.PORT or 12080
local bind_ip = FIDDLE.BIND_IP or '127.0.0.1'

if (isMission) then
    __info("Starting fiddle server in the mission scripting environment...")
    local loop = create_server(bind_ip, port)

    timer.scheduleFunction(function(arg, time)
        local success, err = pcall(loop)
        if not success then
            __info("loop() error: " .. tostring(err))
        end
        return timer.getTime() + .1
    end, nil, timer.getTime() + .1)

    __info("DCS Fiddle server running")
    env.info("DCS Fiddle successfully initialized.\n\nHappy Hacking!!", false)
    -- trigger.action.outText("DCS Fiddle successfully initialized.", 5)
elseif (not isMission) then
    __info("Starting fiddle server in the Hooks environment...")

    local fiddleFile = lfs.writedir() .. 'Scripts\\Hooks\\dcs-fiddle-server.lua'

    local loop = create_server(bind_ip, port+1)

    local callbacks = {}

    function callbacks.onSimulationStart()
        __info("Bootstrapping DCS Fiddle inside the mission using file " .. fiddleFile)
        net.dostring_in("mission", string.format([[a_do_script("dofile('%s')")]], fiddleFile:gsub("\\","/")))
    end

    function callbacks.onSimulationFrame()
        loop()
    end

    DCS.setUserCallbacks(callbacks)

    __info("DCS Fiddle server running")
else
    __info("Failed to start DCS fiddle, unknown environment")
end