goldwarden/cmd/session.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)
}