Merge pull request #1379 from FAForever/add-color-support

add color support

1 files changed, 77 insertions, 0 deletions

diff --git a/script/core/color.lua b/script/core/color.lua
new file mode 100644
index 00000000..f5c792a3
--- /dev/null
+++ b/script/core/color.lua
@@ -0,0 +1,77 @@
+local files = require "files"
+local guide = require "parser.guide"
+
+local colorPattern = string.rep('%x', 8)
+---@param source parser.object
+---@return boolean
+local function isColor(source)
+    ---@type string
+    local text = source[1]
+    if text:len() ~= 8 then
+        return false
+    end
+    return text:match(colorPattern)
+end
+
+
+---@param colorText string
+---@return Color
+local function textToColor(colorText)
+    return {
+        alpha = tonumber(colorText:sub(1, 2), 16) / 255,
+        red = tonumber(colorText:sub(3, 4), 16) / 255,
+        green = tonumber(colorText:sub(5, 6), 16) / 255,
+        blue = tonumber(colorText:sub(7, 8), 16) / 255,
+    }
+end
+
+
+---@param color Color
+---@return string
+local function colorToText(color)
+    return (''
+        .. string.format('%02x', math.tointeger(color.alpha * 255))
+        .. string.format('%02x', math.tointeger(color.red * 255))
+        .. string.format('%02x', math.tointeger(color.green * 255))
+        .. string.format('%02x', math.tointeger(color.blue * 255))):upper()
+end
+
+---@class Color
+---@field red number
+---@field green number
+---@field blue number
+---@field alpha number
+
+---@class ColorValue
+---@field color Color
+---@field start integer
+---@field finish integer
+
+---@async
+function colors(uri)
+    local state = files.getState(uri)
+    local text  = files.getText(uri)
+    if not state or not text then
+        return nil
+    end
+    ---@type ColorValue[]
+    local colors = {}
+
+    guide.eachSource(state.ast, function (source) ---@async
+        if source.type == 'string' and isColor(source) then
+            ---@type string
+            local colorText = source[1]
+            
+            colors[#colors+1] = {
+                start = source.start + 1,
+                finish = source.finish - 1,
+                color = textToColor(colorText)
+            }
+        end
+    end)
+    return colors
+end
+return {
+    colors = colors,
+    colorToText = colorToText
+}