package highlight

import (
	"strings"
	"timshome.page/gilo/terminal"
)

// ----------------------------------------------------------------------------
// !Syntax Highlighting Constants
// ----------------------------------------------------------------------------
const (
	Normal = iota
	Comment
	MLComment
	Keyword1
	Keyword2
	String
	Number
	Match
)

const (
	DoNumbers = 1 << 0
	DoStrings = 1 << 1
)

// ------------------------------------------------------------------
// Syntax Type -> Color Mapping
// ------------------------------------------------------------------

var syntaxColorMap = map[int]string{
	Comment:   terminal.FGCyan,
	MLComment: terminal.FGBrightCyan,
	Keyword1:  terminal.FGYellow,
	Keyword2:  terminal.FGGreen,
	String:    terminal.FGBrightMagenta,
	Number:    terminal.FGRed,
	Match:     terminal.FGBlue,
	Normal:    terminal.DefaultFGColor,
}

// SyntaxToColor Take a highlighting type and map it to
// an ANSI color escape code for display
func SyntaxToColor(hl int) string {
	color := syntaxColorMap[hl]
	if len(color) == 0 {
		color = terminal.DefaultFGColor
	}

	return color
}

// ------------------------------------------------------------------
// File Type -> Syntax Mapping
// ------------------------------------------------------------------

type Syntax struct {
	FileType              string
	FileMatch             []string
	LineCommentStart      string
	MultilineCommentStart string
	MultilineCommentEnd   string
	Keywords1             []string
	Keywords2             []string
	Flags                 int
}

// HLDB - The "database" of syntax types
var HLDB = []*Syntax{{
	"c",
	[]string{".c", ".h", ".cpp"},
	"//",
	"/*",
	"*/",
	[]string{
		"switch", "if", "while", "for", "break", "continue", "return",
		"else", "struct", "union", "typedef", "static", "enum",
		"class", "case",
	},
	[]string{"int",
		"long", "double", "float", "char", "unsigned", "signed", "void",
		"#define", "#endif", "#error", "#if", "#ifdef", "#ifndef", "#include",
		"#undef",
	},
	DoNumbers | DoStrings,
}, {
	"go",
	[]string{".go", "go.mod"},
	"//",
	"/*",
	"*/",
	[]string{
		"break", "case", "chan", "const", "continue", "default", "defer",
		"else", "fallthrough", "for", "func", "go", "goto", "if",
		"import", "interface", "map", "package", "range", "return", "select",
		"struct", "switch", "type", "var", "iota",
	},
	[]string{
		"bool", "uint8", "uint16", "uint32", "uint64", "int8", "int16",
		"int32", "int64", "float32", "float64", "complex64", "complex128",
		"byte", "rune", "uint", "int", "uintptr", "string",
		"struct", "type", "map",
	},
	DoNumbers | DoStrings,
}, {
	"makefile",
	[]string{"Makefile", "makefile", "justfile"},
	"#",
	"",
	"",
	[]string{},
	[]string{},
	0,
}}

func GetSyntaxByFilename(filename string) *Syntax {
	if filename == "" {
		return nil
	}

	var ext string = ""
	extInd := strings.LastIndex(filename, ".")
	if extInd > -1 {
		ext = filename[extInd:len(filename)]
	}

	for i := 0; i < len(HLDB); i++ {
		s := HLDB[i]
		for j := range s.FileMatch {
			if s.FileMatch[j] == ext || strings.Contains(filename, s.FileMatch[j]) {
				return s
			}
		}
	}

	return nil
}