133 lines
3.2 KiB
Go
133 lines
3.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/quexten/goldwarden/ipc/messages"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var vaultCmd = &cobra.Command{
|
|
Use: "vault",
|
|
Short: "Manage the vault",
|
|
Long: `Manage the vault.`,
|
|
}
|
|
|
|
var unlockCmd = &cobra.Command{
|
|
Use: "unlock",
|
|
Short: "Unlocks the vault",
|
|
Long: `Unlocks the vault. You will be prompted for your pin. The pin is empty by default.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
request := messages.UnlockVaultRequest{}
|
|
|
|
result, err := commandClient.SendToAgent(request)
|
|
if err != nil {
|
|
handleSendToAgentError(err)
|
|
return
|
|
}
|
|
|
|
switch result.(type) {
|
|
case messages.ActionResponse:
|
|
if result.(messages.ActionResponse).Success {
|
|
println("Unlocked")
|
|
} else {
|
|
println("Not unlocked: " + result.(messages.ActionResponse).Message)
|
|
}
|
|
default:
|
|
println("Wrong response type")
|
|
}
|
|
},
|
|
}
|
|
|
|
var lockCmd = &cobra.Command{
|
|
Use: "lock",
|
|
Short: "Locks the vault",
|
|
Long: `Locks the vault.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
request := messages.LockVaultRequest{}
|
|
|
|
result, err := commandClient.SendToAgent(request)
|
|
if err != nil {
|
|
handleSendToAgentError(err)
|
|
return
|
|
}
|
|
|
|
switch result.(type) {
|
|
case messages.ActionResponse:
|
|
if result.(messages.ActionResponse).Success {
|
|
println("Locked")
|
|
} else {
|
|
println("Not locked: " + result.(messages.ActionResponse).Message)
|
|
}
|
|
default:
|
|
println("Wrong response type")
|
|
}
|
|
},
|
|
}
|
|
|
|
var purgeCmd = &cobra.Command{
|
|
Use: "purge",
|
|
Short: "Wipes the vault",
|
|
Long: `Wipes the vault and encryption keys from ram and config. Does not delete any entries on the server side.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
request := messages.WipeVaultRequest{}
|
|
|
|
result, err := commandClient.SendToAgent(request)
|
|
if err != nil {
|
|
handleSendToAgentError(err)
|
|
return
|
|
}
|
|
|
|
switch result.(type) {
|
|
case messages.ActionResponse:
|
|
if result.(messages.ActionResponse).Success {
|
|
println("Purged")
|
|
} else {
|
|
println("Not purged: " + result.(messages.ActionResponse).Message)
|
|
}
|
|
default:
|
|
println("Wrong response type")
|
|
}
|
|
},
|
|
}
|
|
|
|
var statusCmd = &cobra.Command{
|
|
Use: "status",
|
|
Short: "Shows the vault status",
|
|
Long: `Shows the vault status.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
request := messages.VaultStatusRequest{}
|
|
|
|
result, err := commandClient.SendToAgent(request)
|
|
if err != nil {
|
|
handleSendToAgentError(err)
|
|
return
|
|
}
|
|
|
|
switch result.(type) {
|
|
case messages.VaultStatusResponse:
|
|
status := result.(messages.VaultStatusResponse)
|
|
fmt.Println("{")
|
|
fmt.Println(" \"locked\":", status.Locked, ",")
|
|
fmt.Println(" \"loginEntries\":", status.NumberOfLogins, ",")
|
|
fmt.Println(" \"noteEntries\":", status.NumberOfNotes, ",")
|
|
fmt.Println(" \"lastSynced\": \"" + time.Unix(status.LastSynced, 0).String() + "\",")
|
|
fmt.Println(" \"websocketConnected\":", status.WebsockedConnected, ",")
|
|
fmt.Println(" \"pinSet\":", status.PinSet, ",")
|
|
fmt.Println(" \"loggedIn\":", status.LoggedIn)
|
|
fmt.Println("}")
|
|
default:
|
|
println("Wrong response type")
|
|
}
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(vaultCmd)
|
|
vaultCmd.AddCommand(unlockCmd)
|
|
vaultCmd.AddCommand(lockCmd)
|
|
vaultCmd.AddCommand(purgeCmd)
|
|
vaultCmd.AddCommand(statusCmd)
|
|
}
|