passgo.go

package main

import (
	"fmt"
	"runtime/debug"
	"strconv"

	"github.com/ejcx/passgo/v2/edit"
	"github.com/ejcx/passgo/v2/generate"
	"github.com/ejcx/passgo/v2/initialize"
	"github.com/ejcx/passgo/v2/insert"
	"github.com/ejcx/passgo/v2/pio"
	"github.com/ejcx/passgo/v2/show"
	"github.com/spf13/cobra"
)

var (
	copyPass bool
	RootCmd  = &cobra.Command{
		Use:   "passgo",
		Short: "Print the contents of the vault.",
		Long: `Print the contents of the vault. If you have
not yet initialized your vault, it is necessary to run
the init subcommand in order to create your passgo
directory, and initialize your cryptographic keys.`,
		Run: func(cmd *cobra.Command, args []string) {
			if exists, _ := pio.PassFileDirExists(); exists {
				show.ListAll()
			} else {
				cmd.Help()
			}
		},
	}
	versionCmd = &cobra.Command{
		Use:   "version",
		Short: "Print the version of your passgo binary.",
		Run: func(cmd *cobra.Command, args []string) {
			info, ok := debug.ReadBuildInfo()
			if !ok {
				fmt.Println("(unknown)")
				return
			}
			fmt.Println(info.Main.Version)
		},
	}
	initCmd = &cobra.Command{
		Use:   "init",
		Short: "Initialize your passgo vault",
		Long:  "Initialize the .passgo directory, and generate your secret keys",
		Run: func(cmd *cobra.Command, args []string) {
			initialize.Init()
		},
	}
	insertCmd = &cobra.Command{
		Use:     "insert",
		Short:   "Insert a file or password in to your vault",
		Example: "passgo insert money/bank.com",
		Args:    cobra.RangeArgs(1, 2),
		Long: `Add a site to your password store. This site can optionally be a part
of a group by prepending a group name and slash to the site name.
Will prompt for confirmation when a site path is not unique.`,
		Run: func(cmd *cobra.Command, args []string) {
			if len(args) == 2 {
				path := args[0]
				filename := args[1]
				insert.File(path, filename)
			} else {
				pathName := args[0]
				insert.Password(pathName)
			}
		},
	}
	showCmd = &cobra.Command{
		Use:     "show",
		Example: "passgo show money/bank.com",
		Short:   "Print the password of a passgo entry.",
		Args:    cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			path := args[0]
			show.Site(path, copyPass)
		},
	}
	generateCmd = &cobra.Command{
		Use:     "generate",
		Short:   "Generate a secure password",
		Example: "passgo generate",
		Long: `Prints a randomly generated password. The length of this password defaults
to 24. If a password length is specified as greater than 2048 then generate
will fail.`,
		Args: cobra.RangeArgs(0, 1),
		Run: func(cmd *cobra.Command, args []string) {
			pwlen := -1
			if len(args) != 0 {
				pwlenStr := args[0]
				pwlenint, err := strconv.Atoi(pwlenStr)
				if err != nil {
					pwlen = -1
				} else {
					pwlen = pwlenint
				}
			}
			pass := generate.Generate(pwlen)
			fmt.Println(pass)
		},
	}
	findCmd = &cobra.Command{
		Use:     "find",
		Aliases: []string{"ls"},
		Example: "passgo find bank.com",
		Short:   "Find a site that contains the site-path.",
		Long: `Prints all sites that contain the site-path. Used to print just
one group or all sites that contain a certain word in the group or name`,
		Args: cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			path := args[0]
			show.Find(path)
		},
	}
	renameCmd = &cobra.Command{
		Use:     "rename",
		Short:   "Rename an entry in the password vault",
		Example: "passgo rename money/bank.com",
		Args:    cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			path := args[0]
			edit.Rename(path)
		},
	}
	editCmd = &cobra.Command{
		Use:     "edit",
		Aliases: []string{"update"},
		Short:   "Change the password of a site in the vault.",
		Example: "passgo edit money/bank.com",
		Args:    cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			path := args[0]
			edit.Edit(path)
		},
	}
	removeCmd = &cobra.Command{
		Use:     "remove",
		Aliases: []string{"rm"},
		Example: "passgo remove money/bank.com",
		Short:   "Remove a site from the password vault by specifying the entire site-path.",
		Args:    cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			path := args[0]
			edit.RemovePassword(path)
		},
	}
)

func init() {
	showCmd.PersistentFlags().BoolVarP(&copyPass, "copy", "c", false, "Copy your password to the clipboard")
	RootCmd.AddCommand(findCmd)
	RootCmd.AddCommand(generateCmd)
	RootCmd.AddCommand(initCmd)
	RootCmd.AddCommand(insertCmd)
	RootCmd.AddCommand(removeCmd)
	RootCmd.AddCommand(editCmd)
	RootCmd.AddCommand(renameCmd)
	RootCmd.AddCommand(showCmd)
	RootCmd.AddCommand(versionCmd)
}

func main() {
	RootCmd.Execute()
}