91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/quexten/goldwarden/ipc/messages"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// sessionCmd represents the run command
|
|
var sessionCmd = &cobra.Command{
|
|
Use: "session",
|
|
Hidden: true,
|
|
Short: "Starts a new session",
|
|
Long: `Starts a new session.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
for {
|
|
reader := bufio.NewReader(os.Stdin)
|
|
text, _ := reader.ReadString('\n')
|
|
text = strings.TrimSuffix(text, "\n")
|
|
args := strings.Split(text, " ")
|
|
rootCmd.SetArgs(args)
|
|
rootCmd.Execute()
|
|
}
|
|
},
|
|
}
|
|
|
|
var pinentry = &cobra.Command{
|
|
Use: "pinentry",
|
|
Hidden: true,
|
|
Short: "Registers as a pinentry program",
|
|
Long: `Registers as a pinentry program.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
conn, err := commandClient.Connect()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer conn.Close()
|
|
_, err = conn.SendCommand(messages.PinentryRegistrationRequest{})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
for {
|
|
response := conn.ReadMessage()
|
|
switch response.(type) {
|
|
case messages.PinentryPinRequest:
|
|
fmt.Println("pin-request" + "," + response.(messages.PinentryPinRequest).Message)
|
|
case messages.PinentryApprovalRequest:
|
|
fmt.Println("approval-request" + "," + response.(messages.PinentryApprovalRequest).Message)
|
|
}
|
|
|
|
// read line
|
|
reader := bufio.NewReader(os.Stdin)
|
|
text, _ := reader.ReadString('\n')
|
|
text = strings.TrimSuffix(text, "\n")
|
|
|
|
switch response.(type) {
|
|
case messages.PinentryPinRequest:
|
|
err = conn.WriteMessage(messages.PinentryPinResponse{Pin: text})
|
|
case messages.PinentryApprovalRequest:
|
|
err = conn.WriteMessage(messages.PinentryApprovalResponse{Approved: text == "true"})
|
|
}
|
|
}
|
|
},
|
|
}
|
|
|
|
var authenticateSession = &cobra.Command{
|
|
Use: "authenticate-session",
|
|
Hidden: true,
|
|
Short: "Authenticates a session",
|
|
Long: `Authenticates a session.`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
token := args[0]
|
|
response, err := commandClient.SendToAgent(messages.SessionAuthRequest{Token: token})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Println(response.(messages.SessionAuthResponse).Verified)
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(sessionCmd)
|
|
rootCmd.AddCommand(pinentry)
|
|
rootCmd.AddCommand(authenticateSession)
|
|
}
|