aboutsummaryrefslogtreecommitdiff
path: root/cmd_generate.go
blob: 57391f6cebd45acefcc46d981e6ad30e2880fd10 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
package main

import (
	"context"
	"encoding"
	"flag"
	"log"
	"os"

	"dim13.org/signify/ask"
	"dim13.org/signify/b64file"
	"dim13.org/signify/key"
	"github.com/google/subcommands"
)

// Usage: signify -G [-n] [-c comment] -p pubkey -s seckey

type generateCmd struct {
	nopass  bool
	comment string
	pubFile string
	secFile string
}

func (m *generateCmd) Name() string     { return "generate" }
func (m *generateCmd) Synopsis() string { return "generate key" }
func (m *generateCmd) Usage() string {
	return "generate [-n] [-c comment] -p pubkey -s seckey\n"
}

func (m *generateCmd) SetFlags(f *flag.FlagSet) {
	f.BoolVar(&m.nopass, "n", false, "no key passphrase")
	f.StringVar(&m.comment, "c", "signify", "comment")
	f.StringVar(&m.pubFile, "p", "", "public key file (required)")
	f.StringVar(&m.secFile, "s", "", "secret key file (required)")
}

func (m *generateCmd) Execute(ctx context.Context, f *flag.FlagSet, args ...interface{}) subcommands.ExitStatus {
	if m.pubFile == "" || m.secFile == "" {
		f.Usage()
		return subcommands.ExitUsageError
	}
	if err := ValidateNames(m.pubFile, m.secFile); err != nil {
		log.Println(err)
		return subcommands.ExitFailure
	}

	var der key.Deriver
	if !m.nopass {
		der = ask.Confirmed{}
	}

	pubKey, secKey, err := key.Generate(der)
	if err != nil {
		log.Println(err)
		return subcommands.ExitFailure
	}

	if err := m.writeFile(m.secFile, "secret key", secKey, ModeSec); err != nil {
		log.Println(err)
		return subcommands.ExitFailure
	}
	if err := m.writeFile(m.pubFile, "public key", pubKey, ModePub); err != nil {
		log.Println(err)
		return subcommands.ExitFailure
	}

	return subcommands.ExitSuccess
}

func (m *generateCmd) writeFile(fname, comment string, u encoding.BinaryMarshaler, mode os.FileMode) error {
	fd, err := Create(fname, mode)
	if err != nil {
		return err
	}
	defer fd.Close()
	return b64file.Encode(fd, u, m.comment+" "+comment, nil)
}