179 lines
3.2 KiB
Go
179 lines
3.2 KiB
Go
package trafficprofiles
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
func CanonicalizeAppKey(appKey string, command string) string {
|
|
key := strings.TrimSpace(appKey)
|
|
cmd := strings.TrimSpace(command)
|
|
|
|
fields := SplitCommandTokens(cmd)
|
|
if len(fields) == 0 && key != "" {
|
|
fields = []string{key}
|
|
}
|
|
|
|
primary := key
|
|
if len(fields) > 0 {
|
|
primary = fields[0]
|
|
}
|
|
primary = stripOuterQuotes(strings.TrimSpace(primary))
|
|
if primary == "" {
|
|
return ""
|
|
}
|
|
|
|
base := strings.ToLower(filepath.Base(primary))
|
|
clean := make([]string, 0, len(fields))
|
|
for _, f := range fields {
|
|
f = stripOuterQuotes(strings.TrimSpace(f))
|
|
if f == "" {
|
|
continue
|
|
}
|
|
clean = append(clean, f)
|
|
}
|
|
|
|
switch base {
|
|
case "flatpak":
|
|
if id := extractRunTarget(clean); id != "" {
|
|
return "flatpak:" + strings.ToLower(strings.TrimSpace(id))
|
|
}
|
|
return "flatpak"
|
|
case "snap":
|
|
if name := extractRunTarget(clean); name != "" {
|
|
return "snap:" + strings.ToLower(strings.TrimSpace(name))
|
|
}
|
|
return "snap"
|
|
case "gtk-launch":
|
|
if len(clean) >= 2 {
|
|
id := strings.TrimSpace(clean[1])
|
|
if id != "" && !strings.HasPrefix(id, "-") {
|
|
return "desktop:" + strings.ToLower(id)
|
|
}
|
|
}
|
|
case "env":
|
|
for i := 1; i < len(clean); i++ {
|
|
tok := strings.TrimSpace(clean[i])
|
|
if tok == "" {
|
|
continue
|
|
}
|
|
if strings.HasPrefix(tok, "-") {
|
|
continue
|
|
}
|
|
if strings.Contains(tok, "=") {
|
|
continue
|
|
}
|
|
return CanonicalizeAppKey(tok, strings.Join(clean[i:], " "))
|
|
}
|
|
return "env"
|
|
}
|
|
|
|
if strings.Contains(primary, "/") {
|
|
b := filepath.Base(primary)
|
|
if b != "" && b != "." && b != "/" {
|
|
return strings.ToLower(strings.TrimSpace(b))
|
|
}
|
|
}
|
|
|
|
return strings.ToLower(strings.TrimSpace(primary))
|
|
}
|
|
|
|
func stripOuterQuotes(s string) string {
|
|
in := strings.TrimSpace(s)
|
|
if len(in) >= 2 {
|
|
if (in[0] == '"' && in[len(in)-1] == '"') || (in[0] == '\'' && in[len(in)-1] == '\'') {
|
|
return strings.TrimSpace(in[1 : len(in)-1])
|
|
}
|
|
}
|
|
return in
|
|
}
|
|
|
|
func extractRunTarget(fields []string) string {
|
|
if len(fields) == 0 {
|
|
return ""
|
|
}
|
|
idx := -1
|
|
for i := 0; i < len(fields); i++ {
|
|
if strings.TrimSpace(fields[i]) == "run" {
|
|
idx = i
|
|
break
|
|
}
|
|
}
|
|
if idx < 0 {
|
|
return ""
|
|
}
|
|
for j := idx + 1; j < len(fields); j++ {
|
|
tok := strings.TrimSpace(fields[j])
|
|
if tok == "" {
|
|
continue
|
|
}
|
|
if tok == "--" {
|
|
continue
|
|
}
|
|
if strings.HasPrefix(tok, "-") {
|
|
continue
|
|
}
|
|
return tok
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func SplitCommandTokens(raw string) []string {
|
|
s := strings.TrimSpace(raw)
|
|
if s == "" {
|
|
return nil
|
|
}
|
|
|
|
out := make([]string, 0, 8)
|
|
var cur strings.Builder
|
|
inSingle := false
|
|
inDouble := false
|
|
escaped := false
|
|
|
|
flush := func() {
|
|
if cur.Len() == 0 {
|
|
return
|
|
}
|
|
out = append(out, cur.String())
|
|
cur.Reset()
|
|
}
|
|
|
|
for _, r := range s {
|
|
if escaped {
|
|
cur.WriteRune(r)
|
|
escaped = false
|
|
continue
|
|
}
|
|
switch r {
|
|
case '\\':
|
|
if inSingle {
|
|
cur.WriteRune(r)
|
|
} else {
|
|
escaped = true
|
|
}
|
|
case '\'':
|
|
if inDouble {
|
|
cur.WriteRune(r)
|
|
} else {
|
|
inSingle = !inSingle
|
|
}
|
|
case '"':
|
|
if inSingle {
|
|
cur.WriteRune(r)
|
|
} else {
|
|
inDouble = !inDouble
|
|
}
|
|
case ' ', '\t', '\n', '\r':
|
|
if inSingle || inDouble {
|
|
cur.WriteRune(r)
|
|
} else {
|
|
flush()
|
|
}
|
|
default:
|
|
cur.WriteRune(r)
|
|
}
|
|
}
|
|
flush()
|
|
return out
|
|
}
|