/
generate.go
66 lines (52 loc) · 1.32 KB
/
generate.go
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
package commands
import (
"fmt"
"os"
"strconv"
"time"
"code.cloudfoundry.org/gorouter/routeservice"
"code.cloudfoundry.org/gorouter/test_util/rss/common"
"github.com/codegangsta/cli"
)
func GenerateSignature(c *cli.Context) {
url := c.String("url")
if url == "" {
cli.ShowCommandHelp(c, "generate")
os.Exit(1)
}
crypto, err := common.CreateCrypto(c)
if err != nil {
os.Exit(1)
}
signatureContents, err := createSigFromArgs(c)
if err != nil {
os.Exit(1)
}
sigEncoded, metaEncoded, err := routeservice.BuildSignatureAndMetadata(crypto, &signatureContents)
if err != nil {
fmt.Printf("Failed to create signature: %s", err.Error())
os.Exit(1)
}
fmt.Printf("Encoded Signature:\n%s\n\n", sigEncoded)
fmt.Printf("Encoded Metadata:\n%s\n\n", metaEncoded)
}
func createSigFromArgs(c *cli.Context) (routeservice.SignatureContents, error) {
signatureContents := routeservice.SignatureContents{}
url := c.String("url")
var sigTime time.Time
timeStr := c.String("time")
if timeStr != "" {
unix, err := strconv.ParseInt(timeStr, 10, 64)
if err != nil {
fmt.Printf("Invalid time format: %s", timeStr)
return signatureContents, err
}
sigTime = time.Unix(unix, 0)
} else {
sigTime = time.Now()
}
return routeservice.SignatureContents{
RequestedTime: sigTime,
ForwardedUrl: url,
}, nil
}