Add parseConfig func
This commit is contained in:
parent
da1d5980e2
commit
3b888d4fc7
|
@ -0,0 +1,110 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
|
||||
"tildegit.org/andinus/lynx"
|
||||
)
|
||||
|
||||
type Pavo struct {
|
||||
Config []Config `json:"config"`
|
||||
}
|
||||
|
||||
type Unveil struct {
|
||||
Permissions string `json:"permissions"`
|
||||
Path string `json:"path"`
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
Command string `json:"command"`
|
||||
Unveil []Unveil `json:"unveil"`
|
||||
CommandsToUnveil []string `json:"commandstounveil"`
|
||||
Execpromises string `json:"execpromises"`
|
||||
}
|
||||
|
||||
func parseConfig() {
|
||||
// Read the config file
|
||||
data, err := ioutil.ReadFile(configFile())
|
||||
if err != nil {
|
||||
fmt.Printf("%s :: %s",
|
||||
"Failed to read the config file",
|
||||
err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
var Pavo Pavo
|
||||
|
||||
// Unmarshal data to struct.
|
||||
err = json.Unmarshal(data, &Pavo)
|
||||
if err != nil {
|
||||
fmt.Printf("%s :: %s\n",
|
||||
"Failed to unmarshal the config file",
|
||||
err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Get command & flags from flag.Args().
|
||||
flag.Parse()
|
||||
flags = flag.Args()[1:]
|
||||
|
||||
for k, v := range Pavo.Config {
|
||||
// If we find the command's config then break.
|
||||
if v.Command == flag.Args()[0] {
|
||||
cmd = v
|
||||
break
|
||||
}
|
||||
|
||||
// If we don't find the config in file then return an
|
||||
// error.
|
||||
if k == len(Pavo.Config)-1 {
|
||||
fmt.Printf("%s %s",
|
||||
"Failed to find config for",
|
||||
flag.Args()[0])
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
// Commands to unveil.
|
||||
commands := []string{cmd.Command}
|
||||
|
||||
// Get commands to unveil from config.
|
||||
for _, v := range cmd.CommandsToUnveil {
|
||||
commands = append(commands, v)
|
||||
}
|
||||
|
||||
// Unveil all values of $PATH/command.
|
||||
err = lynx.UnveilCommands(commands)
|
||||
if err != nil {
|
||||
fmt.Printf("%s :: %s",
|
||||
"Failed to unveil command",
|
||||
err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Set execpromises from config.
|
||||
err = lynx.PledgeExecpromises(cmd.Execpromises)
|
||||
if err != nil {
|
||||
fmt.Printf("%s :: %s",
|
||||
"PledgeExecpromises failed",
|
||||
err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// Unveil all paths from config.
|
||||
paths := make(map[string]string)
|
||||
for _, v := range cmd.Unveil {
|
||||
paths[v.Path] = v.Permissions
|
||||
}
|
||||
|
||||
err = lynx.UnveilPaths(paths)
|
||||
if err != nil {
|
||||
fmt.Printf("%s :: %s",
|
||||
"Failed to unveil paths",
|
||||
err.Error())
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue