Gammon Forum

"\u001b[0;1;33mYou tell Drunken Master Kalas Malarious, Kitten Avenger, \"// Test string.\"\u001b[0;37m"

test = "\027[0;1;33mYou tell foo, \"// Test string.\"\027[0;37mbar"

for sequence, text in string.gmatch (test, "\027%[([0-9;]+)m([^\027]+)") do
    
  for ansi in string.gmatch (sequence, "%d+") do
    print ("ANSI code: ", ansi)
  end -- for

  print ("Text: ", text)

end  -- for

ANSI code:  0
ANSI code:  1
ANSI code:  33
Text:  You tell foo, "// Test string."
ANSI code:  0
ANSI code:  37
Text:  bar

// ANSI Colour Codes

#define ANSI_RESET             0 
#define ANSI_BOLD              1
#define ANSI_BLINK             3 
#define ANSI_UNDERLINE         4 
#define ANSI_SLOW_BLINK        5
#define ANSI_FAST_BLINK        6 
#define ANSI_INVERSE           7 

#define ANSI_CANCEL_BOLD      22 
#define ANSI_CANCEL_BLINK     23 
#define ANSI_CANCEL_UNDERLINE 24 
#define ANSI_CANCEL_SLOW_BLINK  25
#define ANSI_CANCEL_INVERSE   27 

#define ANSI_TEXT_BLACK       30 
#define ANSI_TEXT_RED         31 
#define ANSI_TEXT_GREEN       32 
#define ANSI_TEXT_YELLOW      33 
#define ANSI_TEXT_BLUE        34 
#define ANSI_TEXT_MAGENTA     35 
#define ANSI_TEXT_CYAN        36 
#define ANSI_TEXT_WHITE       37 
#define ANSI_TEXT_256_COLOUR  38
         
#define ANSI_SET_FOREGROUND_DEFAULT 39
                     
#define ANSI_BACK_BLACK       40 
#define ANSI_BACK_RED         41 
#define ANSI_BACK_GREEN       42 
#define ANSI_BACK_YELLOW      43 
#define ANSI_BACK_BLUE        44 
#define ANSI_BACK_MAGENTA     45 
#define ANSI_BACK_CYAN        46 
#define ANSI_BACK_WHITE       47 
#define ANSI_BACK_256_COLOUR  48

#define ANSI_SET_BACKGROUND_DEFAULT 49

-- Aardwolf bold colors
local ansi_digit_to_bold_atcode = {
   [30] = "@D",
   [31] = "@R",
   [32] = "@G",
   [33] = "@Y",
   [34] = "@B",
   [35] = "@M",
   [36] = "@C",
   [37] = "@W"
}

-- Tries to convert ANSI sequences to Aardwolf color codes
function AnsiToColours(ansi, default_foreground_code)
   if not default_foreground_code then
      default_foreground_code = "@w"
   end

   local ansi_capture = "\027%[([%d;]+)m"

   -- this stuff goes outside because ANSI is a state machine (lolsigh)
   local bold = false
   local color = ""
   local xstage = 0

   ansi = ansi:gsub("@","@@"):gsub(ansi_capture, function(a)
      for c in a:gmatch("%d+") do
         local nc = tonumber(c)
         if nc == 38 then
            xstage = 1
         elseif nc == 5 and xstage == 1 then
            xstage = 2
         elseif xstage == 2 then -- xterm 256 color
            if bold and ansi_digit_to_bold_atcode[nc+30] then
               color = ansi_digit_to_bold_atcode[nc+30]
            else
               color = string.format("@x%03d", nc)
            end
            xstage = 0
         elseif nc == 1 then
            bold = true
            xstage = 0
         elseif nc == 0 then
            bold = false
            -- not actually sure if we should set color here or not
            color = default_foreground_code
         elseif nc <= 37 and nc >= 30 then -- regular color
            if bold and ansi_digit_to_bold_atcode[nc] then
               color = ansi_digit_to_bold_atcode[nc]
            else
               color = string.format("@x%03d", nc-30)
            end
            xstage = 0
         end
      end
      return color
   end)
   
   return ansi
end

-- Converts text with Aardwolf normal and xterm 256 color codes into MUSHclient style runs
function ColorsToStyles (Text, default_foreground_code, default_background_code)
   if not default_foreground_code then
      default_foreground_code = "@w"
   end
   default_foreground = atcode_to_color_value[default_foreground_code]
  
   if not default_background_code then
      default_background_code = "@k"
   end
   default_background = atcode_to_color_value[default_background_code]
    
   if Text:match ("@") then
      astyles = {}

      -- make sure we start with a color
      if Text:sub(1, 1) ~= "@" then
         Text = default_foreground_code .. Text
      end -- if

      Text = Text:gsub ("@%-", "~") -- fix tildes
      Text = Text:gsub ("@@", "\0") -- change @@ to 0x00
      Text = Text:gsub ("@x([^%d])","%1") -- strip invalid xterm codes (non-number)
      Text = Text:gsub ("@x[3-9]%d%d","") -- strip invalid xterm codes (300+)
      Text = Text:gsub ("@x2[6-9]%d","") -- strip invalid xterm codes (260+)
      Text = Text:gsub ("@x25[6-9]","") -- strip invalid xterm codes (256+)
      Text = Text:gsub ("@[^xrgybmcwDRGYBMCWd]", "")  -- strip hidden garbage

      for color, text in Text:gmatch ("@(%a)([^@]+)") do
         text = text:gsub ("%z", "@") -- put any @ characters back

         if color == "x" then -- xterm 256 colors
            code,text = text:match("(%d%d?%d?)(.*)")
            color = color..code
         end

         if #text > 0 then
            table.insert (astyles, { text = text, 
               bold = (color == color:upper()),
               length = #text, 
               textcolour = atcode_to_color_value[color] or default_foreground,
               backcolour = default_background })
         end -- if some text
      end -- for each color run.

      return astyles
   end -- if any color codes at all

   -- No color codes, create a single style.
   return { { text = Text, 
      length = #Text, 
      textcolour = default_foreground,
      backcolour = default_background } }
end  -- function ColorsToStyles