func TestSignatureApplicationAuthorized(t *testing.T) {
ts := newHTTPServer()
defer ts.Close()
defer ts.CloseClientConnections()
content := `{
"debug": true,
"port": 3001,
"secret_key": "dummy"
}`
app, err := NewFromConfig(content)
assert.Nil(t, err)
u, _ := url.Parse(ts.URL + "/avatar.png")
params := fmt.Sprintf("h=100&op=resize&url=%s&w=100", u.String())
values, err := url.ParseQuery(params)
assert.Nil(t, err)
sig := signature.Sign("dummy", values.Encode())
location := fmt.Sprintf("http://example.com/display?%s&sig=%s", params, sig)
request, _ := http.NewRequest("GET", location, nil)
negroni := app.InitRouter()
res := httptest.NewRecorder()
negroni.ServeHTTP(res, request)
assert.Equal(t, 200, res.Code)
}
func main() {
app := cli.NewApp()
app.Name = "picfit"
app.Author = "thoas"
app.Email = "*****@*****.**"
app.Usage = "Display, manipulate, transform and cache your images"
app.Version = application.Version
app.Flags = []cli.Flag{
cli.StringFlag{
Name: "config, c",
Usage: "Config file path",
EnvVar: "PICFIT_CONFIG_PATH",
},
}
app.Commands = []cli.Command{
{
Name: "version",
ShortName: "v",
Usage: "Retrieve the version number",
Action: func(c *cli.Context) {
fmt.Printf("picfit %s\n", application.Version)
},
},
{
Name: "signature",
ShortName: "s",
Usage: "Verify that your client application is generating correct signatures",
Flags: []cli.Flag{
cli.StringFlag{
Name: "key",
Usage: "The signing key",
},
},
Action: func(c *cli.Context) {
key := c.String("key")
if key == "" {
fmt.Fprintf(os.Stderr, "You must provide a key\n")
os.Exit(1)
}
if len(c.Args()) < 1 {
fmt.Fprintf(os.Stderr, "You must provide a Query String\n")
os.Exit(1)
}
qs := c.Args()[0]
sig := signature.Sign(key, qs)
appended := signature.AppendSign(key, qs)
fmt.Fprintf(os.Stdout, "Query String: %s\n", qs)
fmt.Fprintf(os.Stdout, "Signature: %s\n", sig)
fmt.Fprintf(os.Stdout, "Signed Query String: %s\n", appended)
},
},
}
app.Action = func(c *cli.Context) {
config := c.String("config")
if config != "" {
if _, err := os.Stat(config); err != nil {
fmt.Fprintf(os.Stderr, "Can't find config file `%s`\n", config)
os.Exit(1)
}
} else {
fmt.Fprintf(os.Stderr, "Can't find config file\n")
os.Exit(1)
}
err := application.Run(config)
if err != nil {
fmt.Fprint(os.Stderr, err)
os.Exit(1)
}
}
app.Run(os.Args)
}