gBoards Common (#8921)

Co-Authored-By: Drashna Jaelre <drashna@live.com>

keyboards/gboards/dicts/aset/_generator/main.go

@@ -0,0 +1,298 @@
+package main
+
+import (
+	"io/ioutil"
+	"fmt"
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"strings"
+	"sort"
+	"hash/crc64"
+	//"encoding/base64"
+)
+
+func main() {
+	// Show Usage
+	if len(os.Args) < 3 {
+		fmt.Println("Usage: ./keymap-gen src-dir out-dir")
+		fmt.Println("Outputs c files in out-dir")
+		fmt.Println("Make sure the dirs exist.")
+		return
+	}
+
+	files, err := filepath.Glob(os.Args[1] + "/*.json")
+	if err != nil {
+		fmt.Printf("Could not open src-dir: %v\n", err)
+		return
+	}
+
+	for _, fname := range(files) {
+		fmt.Println("Processing: ", fname)
+		// Read the source
+		data, err := ioutil.ReadFile(fname)
+		if err != nil {
+			panic(err)
+		}
+
+		// Unbundle Data
+		var FullDict map[string]Entry
+		json.Unmarshal(data, &FullDict)
+
+		// Loop over entries and store
+		var output []string
+		for i,v := range FullDict {
+			if i == "0" {
+				continue
+			}
+			
+			// Special handling for numbermap
+			var entry string
+			if strings.Contains(fname, "num") {
+				entry = v.toKeymap("NUM|")
+			} else {
+				entry = v.toKeymap("")
+			}
+
+			if entry != "" {
+				output = append(output, entry)
+			}
+		}
+
+		// Sort by length, then amount of whitespace lol
+		sort.Slice(output, func (i,j int) bool {
+			var maxLen int
+			if len(output[i]) > len(output[j]) { 
+				maxLen = len(output[i])
+			} else {
+				maxLen = len(output[j])
+			}
+
+			return maxLen-strings.Count(output[i][:40], " ") < maxLen-strings.Count(output[j][:40], " ")
+		})
+
+		// Whack a disclaimer
+		output = append([]string{"// This file is automatically generated. Do not edit it!\n\n"}, output...)
+		
+		// Write all data out
+		outName := filepath.Base(fname)
+		outName = os.Args[2] + outName[:len(outName)-5]+".def"
+		fmt.Println("Saving: ", outName)
+		ioutil.WriteFile(outName, []byte(strings.Join(output, "")), 0755)
+	}
+}
+
+func (e Entry) toKeymap(prefix string) (string) {
+	// storage for parts
+	var command, chord, arg string
+	wordInfo := parseWords(e)
+
+	// Handle prefix
+	if prefix != "" {
+		chord = prefix
+	}
+
+	// Format the chord
+	keys := []string{"AA", "AS", "AE", "AT", "AN", "AI", "AO", "AP"}
+	for i, v := range e.Input { 
+		chord += keys[v-1]
+
+		if i != len(e.Input)-1 {
+			chord += "|"
+		}
+	}
+
+	// Handle specials/base first
+	var ok bool
+	var v  []string
+
+	if e.Special != "" {
+		v, ok = QMKLookup[e.Special] 
+	}
+	if !ok && e.Base != "" {
+		v, ok = QMKLookup[e.Base] 
+	}
+
+	if ok {
+		// Determine way to send key
+		if len(v) == 1 {
+			command = "PRES("
+		} else {
+			command = "KEYS("
+		}
+	}
+
+	if ok {
+		if len(v) > 1 {
+			arg += "{"
+		}
+
+		// String together args
+		for ii, vv := range(v) {
+			arg += vv
+			if ii == len(v)-1 && len(v) > 1 {
+				arg += ", COMBO_END}"
+			} else if ii != len(v)-1 {
+				arg += ", "
+			}
+		}
+
+		
+		hash := crc64.Checksum([]byte(fmt.Sprintf("%v%v", arg, chord)), crc64.MakeTable(crc64.ECMA))
+		hashStr := fmt.Sprintf("cmb_%x", hash)
+		wordSpacer := strings.Repeat(" ", 42-len(arg))
+		if command == "KEYS(" {
+			arg = fmt.Sprintf("%v, %v %v", hashStr, wordSpacer, arg)
+		} else {
+			arg = fmt.Sprintf("%65v", arg)
+		}
+			
+		goto Found
+	}
+
+	// Parse out word info
+	if wordInfo.LRank == 0 && wordInfo.RRank == 0 {
+		goto Blank
+	}
+
+	if wordInfo.LRank != 0 || wordInfo.RRank != 0 {
+		if wordInfo.LRank != 0 && wordInfo.RRank != 0 {
+			// Just blank the structure and recall
+			left, right := e, e
+			left.Trw = nil
+			right.Tlw = nil
+			return fmt.Sprintf("%v%v", left.toKeymap("LFT|"), right.toKeymap("RGT|"))
+		}
+
+		var word string
+		if wordInfo.LRank > wordInfo.RRank {
+			word = wordInfo.LWord
+		} else {
+			word = wordInfo.RWord
+		}
+
+		// Add in thumb
+		chord = "AR|" + chord
+
+
+		// generate function name
+		hash := crc64.Checksum([]byte(word), crc64.MakeTable(crc64.ECMA))
+		hashStr := fmt.Sprintf("str_%016X", hash)
+		command = "SUBS("
+		wordSpacer := strings.Repeat(" ", 40-len(word))
+		arg = fmt.Sprintf("%v, %v \"%v \"", hashStr, wordSpacer, word)
+		goto Found
+	}
+
+	panic(e.String())
+
+	Found:
+	chord += ","
+	return fmt.Sprintf("%v%-35v%v)\n", command, chord, arg)
+
+	Blank:
+	return ""
+}
+
+type Entry struct {
+	Input 	[]int
+	Base		string
+	Tlw			[]interface{}
+	Trw			[]interface{}
+	Special string
+}
+type Word struct {
+	LWord 		string 
+	LRank			float64
+	RWord 		string 
+	RRank			float64
+}
+
+func parseWords(e Entry) (ret Word) {
+	if len(e.Tlw) > 0 { 
+		ret.LWord = e.Tlw[0].(string)
+		ret.LRank=  e.Tlw[1].(float64)
+	}
+	if len(e.Trw) > 0 { 
+		ret.RWord = e.Trw[0].(string)
+		ret.RRank=  e.Trw[1].(float64)
+	}
+	return ret
+}
+func (e Entry) String() (ret string) {
+	ret = fmt.Sprintln("Input: ", e.Input)
+	ret += fmt.Sprintln("Base: ",  e.Base)
+	ret += fmt.Sprintln("Left: ", e.Tlw)
+	ret += fmt.Sprintln("Right: ", e.Trw)
+	ret += fmt.Sprintln("Special: ", e.Special)
+
+	return ret
+}
+
+var QMKLookup = map[string][]string {
+"!":[]string{"KC_LSFT", "KC_1"},
+"'":[]string{"KC_QUOT"},
+"(":[]string{"KC_LSFT", "KC_9"},
+")":[]string{"KC_LSFT", "KC_0"},
+",":[]string{"KC_COMM"},
+"-":[]string{"KC_MINS"},
+".":[]string{"KC_DOT"},
+";":[]string{"KC_SCLN"},
+"?":[]string{"KC_QUOT"},
+"a":[]string{"KC_A"},
+"b":[]string{"KC_B"},
+"c":[]string{"KC_C"},
+"d":[]string{"KC_D"},
+"e":[]string{"KC_E"},
+"f":[]string{"KC_F"},
+"g":[]string{"KC_G"},
+"h":[]string{"KC_H"},
+"i":[]string{"KC_I"},
+"j":[]string{"KC_J"},
+"k":[]string{"KC_K"},
+"l":[]string{"KC_L"},
+"m":[]string{"KC_M"},
+"n":[]string{"KC_N"},
+"o":[]string{"KC_O"},
+"p":[]string{"KC_P"},
+"q":[]string{"KC_Q"},
+"r":[]string{"KC_R"},
+"s":[]string{"KC_S"},
+"t":[]string{"KC_T"},
+"u":[]string{"KC_U"},
+"v":[]string{"KC_V"},
+"w":[]string{"KC_W"},
+"x":[]string{"KC_X"},
+"y":[]string{"KC_Y"},
+"z":[]string{"KC_Z"},
+
+//specials
+"bksp":[]string{"KC_BSPC"},
+"enter":[]string{"KC_ENT"},
+//"numsym":[]string{"NUM)"}, //TODO: Sticky
+//"LETTERS":[]string{"KC_SPC"},
+
+//symbols
+"[":[]string{"KC_LBRC"},
+"]":[]string{"KC_RBRC"},
+" ":[]string{"KC_SPC"},
+"1":[]string{"KC_1"},
+"2":[]string{"KC_2"},
+"3":[]string{"KC_3"},
+"4":[]string{"KC_4"},
+"5":[]string{"KC_5"},
+"6":[]string{"KC_6"},
+"7":[]string{"KC_7"},
+"8":[]string{"KC_8"},
+"9":[]string{"KC_9"},
+"0":[]string{"KC_0"},
+"=":[]string{"KC_EQL"},
+"Fn":[]string{"KC_NO"},
+"SPACE":[]string{"KC_SPC"},
+"Home":[]string{"KC_HOME"},
+"End":[]string{"KC_END"},
+"    ":[]string{"KC_TAB"},
+"   ":[]string{"KC_TAB"},
+"\t":[]string{"KC_TAB"},
+"`":[]string{"KC_GRV"},
+}