internal/config/colorscheme.go

   1 package config
   2
   3 import (
   4         "errors"
   5         "regexp"
   6         "strconv"
   7         "strings"
   8
   9         "github.com/zyedidia/tcell"
  10 )
  11
  12 // Micro's default style
  13 var DefStyle tcell.Style = tcell.StyleDefault
  14
  15 // The current colorscheme
  16 var Colorscheme map[string]tcell.Style
  17
  18 // GetColor takes in a syntax group and returns the colorscheme's style for that group
  19 func GetColor(color string) tcell.Style {
  20         st := DefStyle
  21         if color == "" {
  22                 return st
  23         }
  24         groups := strings.Split(color, ".")
  25         if len(groups) > 1 {
  26                 curGroup := ""
  27                 for i, g := range groups {
  28                         if i != 0 {
  29                                 curGroup += "."
  30                         }
  31                         curGroup += g
  32                         if style, ok := Colorscheme[curGroup]; ok {
  33                                 st = style
  34                         }
  35                 }
  36         } else if style, ok := Colorscheme[color]; ok {
  37                 st = style
  38         } else {
  39                 st = StringToStyle(color)
  40         }
  41
  42         return st
  43 }
  44
  45 // ColorschemeExists checks if a given colorscheme exists
  46 func ColorschemeExists(colorschemeName string) bool {
  47         return FindRuntimeFile(RTColorscheme, colorschemeName) != nil
  48 }
  49
  50 // InitColorscheme picks and initializes the colorscheme when micro starts
  51 func InitColorscheme() error {
  52         Colorscheme = make(map[string]tcell.Style)
  53         DefStyle = tcell.StyleDefault
  54
  55         return LoadDefaultColorscheme()
  56 }
  57
  58 // LoadDefaultColorscheme loads the default colorscheme from $(ConfigDir)/colorschemes
  59 func LoadDefaultColorscheme() error {
  60         return LoadColorscheme(GlobalSettings["colorscheme"].(string))
  61 }
  62
  63 // LoadColorscheme loads the given colorscheme from a directory
  64 func LoadColorscheme(colorschemeName string) error {
  65         file := FindRuntimeFile(RTColorscheme, colorschemeName)
  66         if file == nil {
  67                 return errors.New(colorschemeName + " is not a valid colorscheme")
  68         }
  69         if data, err := file.Data(); err != nil {
  70                 return errors.New("Error loading colorscheme: " + err.Error())
  71         } else {
  72                 Colorscheme, err = ParseColorscheme(string(data))
  73                 if err != nil {
  74                         return err
  75                 }
  76         }
  77         return nil
  78 }
  79
  80 // ParseColorscheme parses the text definition for a colorscheme and returns the corresponding object
  81 // Colorschemes are made up of color-link statements linking a color group to a list of colors
  82 // For example, color-link keyword (blue,red) makes all keywords have a blue foreground and
  83 // red background
  84 func ParseColorscheme(text string) (map[string]tcell.Style, error) {
  85         var err error
  86         parser := regexp.MustCompile(`color-link\s+(\S*)\s+"(.*)"`)
  87
  88         lines := strings.Split(text, "\n")
  89
  90         c := make(map[string]tcell.Style)
  91
  92         for _, line := range lines {
  93                 if strings.TrimSpace(line) == "" ||
  94                         strings.TrimSpace(line)[0] == '#' {
  95                         // Ignore this line
  96                         continue
  97                 }
  98
  99                 matches := parser.FindSubmatch([]byte(line))
 100                 if len(matches) == 3 {
 101                         link := string(matches[1])
 102                         colors := string(matches[2])
 103
 104                         style := StringToStyle(colors)
 105                         c[link] = style
 106
 107                         if link == "default" {
 108                                 DefStyle = style
 109                         }
 110                 } else {
 111                         err = errors.New("Color-link statement is not valid: " + line)
 112                 }
 113         }
 114
 115         return c, err
 116 }
 117
 118 // StringToStyle returns a style from a string
 119 // The strings must be in the format "extra foregroundcolor,backgroundcolor"
 120 // The 'extra' can be bold, reverse, italic or underline
 121 func StringToStyle(str string) tcell.Style {
 122         var fg, bg string
 123         spaceSplit := strings.Split(str, " ")
 124         var split []string
 125         split = strings.Split(spaceSplit[len(spaceSplit)-1], ",")
 126         if len(split) > 1 {
 127                 fg, bg = split[0], split[1]
 128         } else {
 129                 fg = split[0]
 130         }
 131         fg = strings.TrimSpace(fg)
 132         bg = strings.TrimSpace(bg)
 133
 134         var fgColor, bgColor tcell.Color
 135         if fg == "" {
 136                 fgColor, _, _ = DefStyle.Decompose()
 137         } else {
 138                 fgColor = StringToColor(fg)
 139         }
 140         if bg == "" {
 141                 _, bgColor, _ = DefStyle.Decompose()
 142         } else {
 143                 bgColor = StringToColor(bg)
 144         }
 145
 146         style := DefStyle.Foreground(fgColor).Background(bgColor)
 147         if strings.Contains(str, "bold") {
 148                 style = style.Bold(true)
 149         }
 150         if strings.Contains(str, "italic") {
 151                 style = style.Italic(true)
 152         }
 153         if strings.Contains(str, "reverse") {
 154                 style = style.Reverse(true)
 155         }
 156         if strings.Contains(str, "underline") {
 157                 style = style.Underline(true)
 158         }
 159         return style
 160 }
 161
 162 // StringToColor returns a tcell color from a string representation of a color
 163 // We accept either bright... or light... to mean the brighter version of a color
 164 func StringToColor(str string) tcell.Color {
 165         switch str {
 166         case "black":
 167                 return tcell.ColorBlack
 168         case "red":
 169                 return tcell.ColorMaroon
 170         case "green":
 171                 return tcell.ColorGreen
 172         case "yellow":
 173                 return tcell.ColorOlive
 174         case "blue":
 175                 return tcell.ColorNavy
 176         case "magenta":
 177                 return tcell.ColorPurple
 178         case "cyan":
 179                 return tcell.ColorTeal
 180         case "white":
 181                 return tcell.ColorSilver
 182         case "brightblack", "lightblack":
 183                 return tcell.ColorGray
 184         case "brightred", "lightred":
 185                 return tcell.ColorRed
 186         case "brightgreen", "lightgreen":
 187                 return tcell.ColorLime
 188         case "brightyellow", "lightyellow":
 189                 return tcell.ColorYellow
 190         case "brightblue", "lightblue":
 191                 return tcell.ColorBlue
 192         case "brightmagenta", "lightmagenta":
 193                 return tcell.ColorFuchsia
 194         case "brightcyan", "lightcyan":
 195                 return tcell.ColorAqua
 196         case "brightwhite", "lightwhite":
 197                 return tcell.ColorWhite
 198         case "default":
 199                 return tcell.ColorDefault
 200         default:
 201                 // Check if this is a 256 color
 202                 if num, err := strconv.Atoi(str); err == nil {
 203                         return GetColor256(num)
 204                 }
 205                 // Probably a truecolor hex value
 206                 return tcell.GetColor(str)
 207         }
 208 }
 209
 210 // GetColor256 returns the tcell color for a number between 0 and 255
 211 func GetColor256(color int) tcell.Color {
 212         colors := []tcell.Color{tcell.ColorBlack, tcell.ColorMaroon, tcell.ColorGreen,
 213                 tcell.ColorOlive, tcell.ColorNavy, tcell.ColorPurple,
 214                 tcell.ColorTeal, tcell.ColorSilver, tcell.ColorGray,
 215                 tcell.ColorRed, tcell.ColorLime, tcell.ColorYellow,
 216                 tcell.ColorBlue, tcell.ColorFuchsia, tcell.ColorAqua,
 217                 tcell.ColorWhite, tcell.Color16, tcell.Color17, tcell.Color18, tcell.Color19, tcell.Color20,
 218                 tcell.Color21, tcell.Color22, tcell.Color23, tcell.Color24, tcell.Color25, tcell.Color26, tcell.Color27, tcell.Color28,
 219                 tcell.Color29, tcell.Color30, tcell.Color31, tcell.Color32, tcell.Color33, tcell.Color34, tcell.Color35, tcell.Color36,
 220                 tcell.Color37, tcell.Color38, tcell.Color39, tcell.Color40, tcell.Color41, tcell.Color42, tcell.Color43, tcell.Color44,
 221                 tcell.Color45, tcell.Color46, tcell.Color47, tcell.Color48, tcell.Color49, tcell.Color50, tcell.Color51, tcell.Color52,
 222                 tcell.Color53, tcell.Color54, tcell.Color55, tcell.Color56, tcell.Color57, tcell.Color58, tcell.Color59, tcell.Color60,
 223                 tcell.Color61, tcell.Color62, tcell.Color63, tcell.Color64, tcell.Color65, tcell.Color66, tcell.Color67, tcell.Color68,
 224                 tcell.Color69, tcell.Color70, tcell.Color71, tcell.Color72, tcell.Color73, tcell.Color74, tcell.Color75, tcell.Color76,
 225                 tcell.Color77, tcell.Color78, tcell.Color79, tcell.Color80, tcell.Color81, tcell.Color82, tcell.Color83, tcell.Color84,
 226                 tcell.Color85, tcell.Color86, tcell.Color87, tcell.Color88, tcell.Color89, tcell.Color90, tcell.Color91, tcell.Color92,
 227                 tcell.Color93, tcell.Color94, tcell.Color95, tcell.Color96, tcell.Color97, tcell.Color98, tcell.Color99, tcell.Color100,
 228                 tcell.Color101, tcell.Color102, tcell.Color103, tcell.Color104, tcell.Color105, tcell.Color106, tcell.Color107, tcell.Color108,
 229                 tcell.Color109, tcell.Color110, tcell.Color111, tcell.Color112, tcell.Color113, tcell.Color114, tcell.Color115, tcell.Color116,
 230                 tcell.Color117, tcell.Color118, tcell.Color119, tcell.Color120, tcell.Color121, tcell.Color122, tcell.Color123, tcell.Color124,
 231                 tcell.Color125, tcell.Color126, tcell.Color127, tcell.Color128, tcell.Color129, tcell.Color130, tcell.Color131, tcell.Color132,
 232                 tcell.Color133, tcell.Color134, tcell.Color135, tcell.Color136, tcell.Color137, tcell.Color138, tcell.Color139, tcell.Color140,
 233                 tcell.Color141, tcell.Color142, tcell.Color143, tcell.Color144, tcell.Color145, tcell.Color146, tcell.Color147, tcell.Color148,
 234                 tcell.Color149, tcell.Color150, tcell.Color151, tcell.Color152, tcell.Color153, tcell.Color154, tcell.Color155, tcell.Color156,
 235                 tcell.Color157, tcell.Color158, tcell.Color159, tcell.Color160, tcell.Color161, tcell.Color162, tcell.Color163, tcell.Color164,
 236                 tcell.Color165, tcell.Color166, tcell.Color167, tcell.Color168, tcell.Color169, tcell.Color170, tcell.Color171, tcell.Color172,
 237                 tcell.Color173, tcell.Color174, tcell.Color175, tcell.Color176, tcell.Color177, tcell.Color178, tcell.Color179, tcell.Color180,
 238                 tcell.Color181, tcell.Color182, tcell.Color183, tcell.Color184, tcell.Color185, tcell.Color186, tcell.Color187, tcell.Color188,
 239                 tcell.Color189, tcell.Color190, tcell.Color191, tcell.Color192, tcell.Color193, tcell.Color194, tcell.Color195, tcell.Color196,
 240                 tcell.Color197, tcell.Color198, tcell.Color199, tcell.Color200, tcell.Color201, tcell.Color202, tcell.Color203, tcell.Color204,
 241                 tcell.Color205, tcell.Color206, tcell.Color207, tcell.Color208, tcell.Color209, tcell.Color210, tcell.Color211, tcell.Color212,
 242                 tcell.Color213, tcell.Color214, tcell.Color215, tcell.Color216, tcell.Color217, tcell.Color218, tcell.Color219, tcell.Color220,
 243                 tcell.Color221, tcell.Color222, tcell.Color223, tcell.Color224, tcell.Color225, tcell.Color226, tcell.Color227, tcell.Color228,
 244                 tcell.Color229, tcell.Color230, tcell.Color231, tcell.Color232, tcell.Color233, tcell.Color234, tcell.Color235, tcell.Color236,
 245                 tcell.Color237, tcell.Color238, tcell.Color239, tcell.Color240, tcell.Color241, tcell.Color242, tcell.Color243, tcell.Color244,
 246                 tcell.Color245, tcell.Color246, tcell.Color247, tcell.Color248, tcell.Color249, tcell.Color250, tcell.Color251, tcell.Color252,
 247                 tcell.Color253, tcell.Color254, tcell.Color255,
 248         }
 249
 250         if color >= 0 && color < len(colors) {
 251                 return colors[color]
 252         }
 253
 254         return tcell.ColorDefault
 255 }