Move language map to json file in order to not trigger microsoft antivirus

This commit is contained in:
Lukasz
2024-03-19 18:00:05 +01:00
parent fc48cca110
commit 1546e3f145
5 changed files with 153 additions and 114 deletions

View File

@ -1,8 +1,12 @@
package resources
import (
"encoding/json"
"flag"
"fmt"
"io"
"os"
"strconv"
"strings"
)
@ -10,10 +14,46 @@ import (
type Arguments struct {
Language string
ModelPath string
Port int
}
type ParsedArguments struct {
Language int32
ModelPath string
Port int
}
type LanguageMap map[string]string
func processLanguageAndCode(args *Arguments) (int32, error) {
// Read the language map from JSON file
jsonFile, err := os.Open("languageMap.json")
if err != nil {
return 0x6E65, fmt.Errorf("error opening language map: %w", err) // Wrap error for context
}
defer jsonFile.Close()
byteData, err := io.ReadAll(jsonFile)
if err != nil {
return 0x6E65, fmt.Errorf("error reading language map: %w", err)
}
var languageMap LanguageMap
err = json.Unmarshal(byteData, &languageMap)
if err != nil {
return 0x6E65, fmt.Errorf("error parsing language map: %w", err)
}
hexCode, ok := languageMap[strings.ToLower(args.Language)]
if !ok {
return 0x6E65, fmt.Errorf("unsupported language: %s", args.Language)
}
languageCode, err := strconv.ParseInt(hexCode, 0, 32)
if err != nil {
return 0x6E65, fmt.Errorf("error converting hex code: %w", err)
}
return int32(languageCode), nil
}
// ParseFlags parses command line arguments and returns an Arguments struct
@ -24,6 +64,8 @@ func ParseFlags() (*ParsedArguments, error) {
flag.StringVar(&args.Language, "language", "", "Language to be processed") // Optional: Redundant to demonstrate
flag.StringVar(&args.ModelPath, "m", "", "Path to the model file (required)")
flag.StringVar(&args.ModelPath, "modelPath", "", "Path to the model file (required)") // Optional: Redundant
flag.IntVar(&args.Port, "p", 3031, "Port to start the server on")
flag.IntVar(&args.Port, "port", 3031, "Port to start the server on") // Optional: Redundant
flag.Usage = func() {
fmt.Println("Usage: your_program [OPTIONS]")
@ -36,23 +78,19 @@ func ParseFlags() (*ParsedArguments, error) {
args.Language = strings.ToLower(args.Language)
var pickedCode int32
// Validate against LanguageMap and get associated code
if code, exists := LanguageMap[args.Language]; exists {
fmt.Println("Language code:", code) // Use the code as needed
pickedCode = code
} else {
fmt.Println("unsupported language: ", args.Language, " Defaulting to english")
pickedCode = 0x6E65 // Default to english
}
// Check for required flags
if args.ModelPath == "" {
return nil, fmt.Errorf("modelPath argument is required")
}
languageCode, err := processLanguageAndCode(args)
if err != nil {
fmt.Println("Error setting language, defaulting to English:", err)
// Use default language code directly as the result here
}
return &ParsedArguments{
Language: pickedCode,
Language: languageCode,
ModelPath: args.ModelPath,
Port: args.Port,
}, nil
}

View File

@ -1,86 +0,0 @@
package resources
var LanguageMap = map[string]int32{
"af": 0x6661, // Afrikaans
"sq": 0x7173, // Albanian
"am": 0x6D61, // Amharic
"ar": 0x7261, // Arabic
"hy": 0x7968, // Armenian
"as": 0x7361, // Assamese
"az": 0x7A61, // Azerbaijani
"ba": 0x6162, // Bashkir
"eu": 0x7565, // Basque
"be": 0x6562, // Belarusian
"bn": 0x6E62, // Bengali
"bs": 0x7362, // Bosnian
"br": 0x7262, // Breton
"bg": 0x6762, // Bulgarian
"ca": 0x6163, // Catalan
"zh": 0x687A, // Chinese
"hr": 0x7268, // Croatian
"cs": 0x7363, // Czech
"da": 0x6164, // Danish
"nl": 0x6C6E, // Dutch
"en": 0x6E65, // English
"et": 0x7465, // Estonian
"fo": 0x6F66, // Faroese
"fi": 0x6966, // Finnish
"fr": 0x7266, // French
"gl": 0x6C67, // Galician
"ka": 0x616B, // Georgian
"de": 0x7265, // German
"el": 0x6C61, // Greek
"gu": 0x7567, // Gujarati
"he": 0x6568, // Hebrew
"hi": 0x6968, // Hindi
"hu": 0x7568, // Hungarian
"is": 0x7369, // Icelandic
"id": 0x6469, // Indonesian
"it": 0x7469, // Italian
"ja": 0x616A, // Japanese
"kn": 0x6E6B, // Kannada
"kk": 0x6B6B, // Kazakh
"km": 0x6D6B, // Khmer
"ko": 0x6F6B, // Korean
"ky": 0x796B, // Kyrgyz
"lo": 0x6F6C, // Lao
"lv": 0x766C, // Latvian
"lt": 0x746C, // Lithuanian
"mk": 0x6B6D, // Macedonian
"ms": 0x736D, // Malay
"ml": 0x6C6D, // Malayalam
"mr": 0x726D, // Marathi
"mn": 0x6E6D, // Mongolian
"ne": 0x6570, // Nepali
"no": 0x6F6E, // Norwegian
"or": 0x726F, // Oriya
"ps": 0x7368, // Pashto
"fa": 0x6172, // Persian
"pl": 0x6C70, // Polish
"pt": 0x7470, // Portuguese
"pa": 0x6170, // Punjabi
"ro": 0x6F72, // Romanian
"ru": 0x7572, // Russian
"sa": 0x6173, // Sanskrit
"sr": 0x7273, // Serbian
"sd": 0x6473, // Sindhi
"si": 0x6973, // Sinhalese
"sk": 0x6B73, // Slovak
"sl": 0x6C73, // Slovenian
"es": 0x6573, // Spanish
"sw": 0x7773, // Swahili
"sv": 0x6576, // Swedish
"tg": 0x6769, // Tajik
"ta": 0x6174, // Tamil
"te": 0x6574, // Telugu
"th": 0x6874, // Thai
"tr": 0x7274, // Turkish
"uk": 0x6B75, // Ukrainian
"ur": 0x7275, // Urdu
"uz": 0x7A75, // Uzbek
"vi": 0x6976, // Vietnamese
"cy": 0x7963, // Welsh
"xh": 0x6877, // Xhosa
"yi": 0x6979, // Yiddish
"yo": 0x6F79, // Yoruba
}