66 lines
1.5 KiB
Go
66 lines
1.5 KiB
Go
|
package cmd
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/quexten/goldwarden/client"
|
||
|
|
"github.com/quexten/goldwarden/ipc"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
)
|
||
|
|
|
||
|
|
var pinCmd = &cobra.Command{
|
||
|
|
Use: "pin",
|
||
|
|
Short: "Manage the vault pin",
|
||
|
|
Long: `Manage the vault pin. The pin is used to unlock the vault.`,
|
||
|
|
}
|
||
|
|
|
||
|
|
var setPinCmd = &cobra.Command{
|
||
|
|
Use: "set",
|
||
|
|
Short: "Set a new pin",
|
||
|
|
Long: `Set a new pin. The pin is used to unlock the vault.`,
|
||
|
|
Run: func(cmd *cobra.Command, args []string) {
|
||
|
|
result, err := client.SendToAgent(ipc.UpdateVaultPINRequest{})
|
||
|
|
if err != nil {
|
||
|
|
println("Error: " + err.Error())
|
||
|
|
println("Is the daemon running?")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
switch result.(type) {
|
||
|
|
case ipc.ActionResponse:
|
||
|
|
if result.(ipc.ActionResponse).Success {
|
||
|
|
println("Pin updated")
|
||
|
|
} else {
|
||
|
|
println("Pin updating failed: " + result.(ipc.ActionResponse).Message)
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
println("Wrong response type")
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
var pinStatusCmd = &cobra.Command{
|
||
|
|
Use: "status",
|
||
|
|
Short: "Check if a pin is set",
|
||
|
|
Long: `Check if a pin is set. The pin is used to unlock the vault.`,
|
||
|
|
Run: func(cmd *cobra.Command, args []string) {
|
||
|
|
result, err := client.SendToAgent(ipc.GetVaultPINRequest{})
|
||
|
|
if err != nil {
|
||
|
|
println("Error: " + err.Error())
|
||
|
|
println("Is the daemon running?")
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
switch result.(type) {
|
||
|
|
case ipc.ActionResponse:
|
||
|
|
println("Pin status: " + result.(ipc.ActionResponse).Message)
|
||
|
|
default:
|
||
|
|
println("Wrong response type")
|
||
|
|
}
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
vaultCmd.AddCommand(pinCmd)
|
||
|
|
pinCmd.AddCommand(setPinCmd)
|
||
|
|
pinCmd.AddCommand(pinStatusCmd)
|
||
|
|
}
|