-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
48 lines (39 loc) · 969 Bytes
/
main.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
package main
import (
"flag"
"log"
"net/http"
"net/url"
"github.com/stinkyfingers/fileEncryption/request"
"github.com/stinkyfingers/fileEncryption/respond"
)
var (
uri = flag.String("uri", "http://localhost:9000", "-uri=filelocation")
fileToGet = flag.String("fetch", "test.txt", "-fetch=<filename_to_get>")
fileToWrite = flag.String("write", "temp.txt", "-write=<file_to_write>")
port = flag.String("port", "9000", "-port=<port>")
)
func main() {
flag.Parse()
// run response server
go server()
// parse url & query
u, err := url.Parse(*uri)
if err != nil {
log.Fatal(err)
}
q := u.Query()
q.Add("file", *fileToGet)
u.RawQuery = q.Encode()
// get & decrypt file
err = request.GetFileHttp(u.String(), *fileToWrite)
if err != nil {
log.Fatal(err)
}
}
// server for the responder
func server() {
mux := http.NewServeMux()
mux.HandleFunc("/", respond.HandleRequest)
log.Fatal(http.ListenAndServe(":"+*port, mux))
}