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, or underline
 121 func StringToStyle(str string) tcell.Style {
 122         var fg, bg string
 123         spaceSplit := strings.Split(str, " ")
 124         var split []string
 125         if len(spaceSplit) > 1 {
 126                 split = strings.Split(spaceSplit[1], ",")
 127         } else {
 128                 split = strings.Split(str, ",")
 129         }
 130         if len(split) > 1 {
 131                 fg, bg = split[0], split[1]
 132         } else {
 133                 fg = split[0]
 134         }
 135         fg = strings.TrimSpace(fg)
 136         bg = strings.TrimSpace(bg)
 137
 138         var fgColor, bgColor tcell.Color
 139         if fg == "" {
 140                 fgColor, _, _ = DefStyle.Decompose()
 141         } else {
 142                 fgColor = StringToColor(fg)
 143         }
 144         if bg == "" {
 145                 _, bgColor, _ = DefStyle.Decompose()
 146         } else {
 147                 bgColor = StringToColor(bg)
 148         }
 149
 150         style := DefStyle.Foreground(fgColor).Background(bgColor)
 151         if strings.Contains(str, "bold") {
 152                 style = style.Bold(true)
 153         }
 154         if strings.Contains(str, "reverse") {
 155                 style = style.Reverse(true)
 156         }
 157         if strings.Contains(str, "underline") {
 158                 style = style.Underline(true)
 159         }
 160         return style
 161 }
 162
 163 // StringToColor returns a tcell color from a string representation of a color
 164 // We accept either bright... or light... to mean the brighter version of a color
 165 func StringToColor(str string) tcell.Color {
 166         switch str {
 167         case "black":
 168                 return tcell.ColorBlack
 169         case "red":
 170                 return tcell.ColorMaroon
 171         case "green":
 172                 return tcell.ColorGreen
 173         case "yellow":
 174                 return tcell.ColorOlive
 175         case "blue":
 176                 return tcell.ColorNavy
 177         case "magenta":
 178                 return tcell.ColorPurple
 179         case "cyan":
 180                 return tcell.ColorTeal
 181         case "white":
 182                 return tcell.ColorSilver
 183         case "brightblack", "lightblack":
 184                 return tcell.ColorGray
 185         case "brightred", "lightred":
 186                 return tcell.ColorRed
 187         case "brightgreen", "lightgreen":
 188                 return tcell.ColorLime
 189         case "brightyellow", "lightyellow":
 190                 return tcell.ColorYellow
 191         case "brightblue", "lightblue":
 192                 return tcell.ColorBlue
 193         case "brightmagenta", "lightmagenta":
 194                 return tcell.ColorFuchsia
 195         case "brightcyan", "lightcyan":
 196                 return tcell.ColorAqua
 197         case "brightwhite", "lightwhite":
 198                 return tcell.ColorWhite
 199         case "default":
 200                 return tcell.ColorDefault
 201         default:
 202                 // Check if this is a 256 color
 203                 if num, err := strconv.Atoi(str); err == nil {
 204                         return GetColor256(num)
 205                 }
 206                 // Probably a truecolor hex value
 207                 return tcell.GetColor(str)
 208         }
 209 }
 210
 211 // GetColor256 returns the tcell color for a number between 0 and 255
 212 func GetColor256(color int) tcell.Color {
 213         colors := []tcell.Color{tcell.ColorBlack, tcell.ColorMaroon, tcell.ColorGreen,
 214                 tcell.ColorOlive, tcell.ColorNavy, tcell.ColorPurple,
 215                 tcell.ColorTeal, tcell.ColorSilver, tcell.ColorGray,
 216                 tcell.ColorRed, tcell.ColorLime, tcell.ColorYellow,
 217                 tcell.ColorBlue, tcell.ColorFuchsia, tcell.ColorAqua,
 218                 tcell.ColorWhite, tcell.Color16, tcell.Color17, tcell.Color18, tcell.Color19, tcell.Color20,
 219                 tcell.Color21, tcell.Color22, tcell.Color23, tcell.Color24, tcell.Color25, tcell.Color26, tcell.Color27, tcell.Color28,
 220                 tcell.Color29, tcell.Color30, tcell.Color31, tcell.Color32, tcell.Color33, tcell.Color34, tcell.Color35, tcell.Color36,
 221                 tcell.Color37, tcell.Color38, tcell.Color39, tcell.Color40, tcell.Color41, tcell.Color42, tcell.Color43, tcell.Color44,
 222                 tcell.Color45, tcell.Color46, tcell.Color47, tcell.Color48, tcell.Color49, tcell.Color50, tcell.Color51, tcell.Color52,
 223                 tcell.Color53, tcell.Color54, tcell.Color55, tcell.Color56, tcell.Color57, tcell.Color58, tcell.Color59, tcell.Color60,
 224                 tcell.Color61, tcell.Color62, tcell.Color63, tcell.Color64, tcell.Color65, tcell.Color66, tcell.Color67, tcell.Color68,
 225                 tcell.Color69, tcell.Color70, tcell.Color71, tcell.Color72, tcell.Color73, tcell.Color74, tcell.Color75, tcell.Color76,
 226                 tcell.Color77, tcell.Color78, tcell.Color79, tcell.Color80, tcell.Color81, tcell.Color82, tcell.Color83, tcell.Color84,
 227                 tcell.Color85, tcell.Color86, tcell.Color87, tcell.Color88, tcell.Color89, tcell.Color90, tcell.Color91, tcell.Color92,
 228                 tcell.Color93, tcell.Color94, tcell.Color95, tcell.Color96, tcell.Color97, tcell.Color98, tcell.Color99, tcell.Color100,
 229                 tcell.Color101, tcell.Color102, tcell.Color103, tcell.Color104, tcell.Color105, tcell.Color106, tcell.Color107, tcell.Color108,
 230                 tcell.Color109, tcell.Color110, tcell.Color111, tcell.Color112, tcell.Color113, tcell.Color114, tcell.Color115, tcell.Color116,
 231                 tcell.Color117, tcell.Color118, tcell.Color119, tcell.Color120, tcell.Color121, tcell.Color122, tcell.Color123, tcell.Color124,
 232                 tcell.Color125, tcell.Color126, tcell.Color127, tcell.Color128, tcell.Color129, tcell.Color130, tcell.Color131, tcell.Color132,
 233                 tcell.Color133, tcell.Color134, tcell.Color135, tcell.Color136, tcell.Color137, tcell.Color138, tcell.Color139, tcell.Color140,
 234                 tcell.Color141, tcell.Color142, tcell.Color143, tcell.Color144, tcell.Color145, tcell.Color146, tcell.Color147, tcell.Color148,
 235                 tcell.Color149, tcell.Color150, tcell.Color151, tcell.Color152, tcell.Color153, tcell.Color154, tcell.Color155, tcell.Color156,
 236                 tcell.Color157, tcell.Color158, tcell.Color159, tcell.Color160, tcell.Color161, tcell.Color162, tcell.Color163, tcell.Color164,
 237                 tcell.Color165, tcell.Color166, tcell.Color167, tcell.Color168, tcell.Color169, tcell.Color170, tcell.Color171, tcell.Color172,
 238                 tcell.Color173, tcell.Color174, tcell.Color175, tcell.Color176, tcell.Color177, tcell.Color178, tcell.Color179, tcell.Color180,
 239                 tcell.Color181, tcell.Color182, tcell.Color183, tcell.Color184, tcell.Color185, tcell.Color186, tcell.Color187, tcell.Color188,
 240                 tcell.Color189, tcell.Color190, tcell.Color191, tcell.Color192, tcell.Color193, tcell.Color194, tcell.Color195, tcell.Color196,
 241                 tcell.Color197, tcell.Color198, tcell.Color199, tcell.Color200, tcell.Color201, tcell.Color202, tcell.Color203, tcell.Color204,
 242                 tcell.Color205, tcell.Color206, tcell.Color207, tcell.Color208, tcell.Color209, tcell.Color210, tcell.Color211, tcell.Color212,
 243                 tcell.Color213, tcell.Color214, tcell.Color215, tcell.Color216, tcell.Color217, tcell.Color218, tcell.Color219, tcell.Color220,
 244                 tcell.Color221, tcell.Color222, tcell.Color223, tcell.Color224, tcell.Color225, tcell.Color226, tcell.Color227, tcell.Color228,
 245                 tcell.Color229, tcell.Color230, tcell.Color231, tcell.Color232, tcell.Color233, tcell.Color234, tcell.Color235, tcell.Color236,
 246                 tcell.Color237, tcell.Color238, tcell.Color239, tcell.Color240, tcell.Color241, tcell.Color242, tcell.Color243, tcell.Color244,
 247                 tcell.Color245, tcell.Color246, tcell.Color247, tcell.Color248, tcell.Color249, tcell.Color250, tcell.Color251, tcell.Color252,
 248                 tcell.Color253, tcell.Color254, tcell.Color255,
 249         }
 250
 251         if color >= 0 && color < len(colors) {
 252                 return colors[color]
 253         }
 254
 255         return tcell.ColorDefault
 256 }