summaryrefslogtreecommitdiff
path: root/cmd/getpipelinebook/main.go
diff options
context:
space:
mode:
authorNick White <git@njw.name>2020-05-26 16:03:17 +0100
committerNick White <git@njw.name>2020-05-26 16:03:17 +0100
commit870fb0f3e01a0771d3839da2461e0cec30282bc0 (patch)
treec8f461ba1b829ce4d91d6eb90fd7d34f1e086845 /cmd/getpipelinebook/main.go
parent3d2bdd1d30ec6d43b4921d29d5a8338d65c48a5a (diff)
parentebc27eef77868fa44daed7cfb0ea129690029da8 (diff)
Merge branch 'local'
Diffstat (limited to 'cmd/getpipelinebook/main.go')
-rw-r--r--cmd/getpipelinebook/main.go12
1 files changed, 10 insertions, 2 deletions
diff --git a/cmd/getpipelinebook/main.go b/cmd/getpipelinebook/main.go
index dc9387f..03e709b 100644
--- a/cmd/getpipelinebook/main.go
+++ b/cmd/getpipelinebook/main.go
@@ -17,7 +17,7 @@ import (
"rescribe.xyz/bookpipeline"
)
-const usage = `Usage: getpipelinebook [-a] [-graph] [-pdf] [-png] [-v] bookname
+const usage = `Usage: getpipelinebook [-c conn] [-a] [-graph] [-pdf] [-png] [-v] bookname
Downloads the pipeline results for a book.
@@ -57,6 +57,7 @@ func getpdfs(conn Pipeliner, l *log.Logger, bookname string) {
func main() {
all := flag.Bool("a", false, "Get all files for book")
+ conntype := flag.String("c", "aws", "connection type ('aws' or 'local')")
graph := flag.Bool("graph", false, "Only download graphs (can be used alongside -pdf)")
binarisedpdf := flag.Bool("binarisedpdf", false, "Only download binarised PDF (can be used alongside -graph)")
colourpdf := flag.Bool("colourpdf", false, "Only download colour PDF (can be used alongside -graph)")
@@ -83,7 +84,14 @@ func main() {
}
var conn Pipeliner
- conn = &bookpipeline.AwsConn{Region: "eu-west-2", Logger: verboselog}
+ switch *conntype {
+ case "aws":
+ conn = &bookpipeline.AwsConn{Region: "eu-west-2", Logger: verboselog}
+ case "local":
+ conn = &bookpipeline.LocalConn{Logger: verboselog}
+ default:
+ log.Fatalln("Unknown connection type")
+ }
verboselog.Println("Setting up AWS session")
err := conn.MinimalInit()