Add parseConfig func
This commit is contained in:
parent
da1d5980e2
commit
3b888d4fc7
110
parseconfig.go
Normal file
110
parseconfig.go
Normal file
|
@ -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
Block a user