Skip to content

Commit

Permalink
Add binary upload mode
Browse files Browse the repository at this point in the history
sandeepmistry committed Dec 20, 2016

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent 70d3dd4 commit 48c3355
Showing 1 changed file with 18 additions and 6 deletions.
24 changes: 18 additions & 6 deletions main.go
Original file line number Diff line number Diff line change
@@ -26,6 +26,7 @@ var (
uploadEndpoint = flag.String("upload", "", "Upload endpoint")
resetEndpoint = flag.String("reset", "", "Upload endpoint")
syncEndpoint = flag.String("sync", "", "Upload endpoint")
binMode = flag.Bool("b", false, "Upload binary mode")
verbose = flag.Bool("v", true, "Verbose flag")
quiet = flag.Bool("q", false, "Quiet flag")
useSsl = flag.String("ssl", "", "SSL flag")
@@ -120,11 +121,18 @@ func main() {
}
defer f.Close()

str := StreamToString(f)
re := regexp.MustCompile(`\r?\n`)
str = re.ReplaceAllString(str, "")
var sketchData *bytes.Buffer

req, err := http.NewRequest("POST", httpheader+*networkAddress+":"+*networkPort+*uploadEndpoint, bytes.NewBufferString(str))
if *binMode {
sketchData = StreamToBytes(f)
} else {
str := StreamToString(f)
re := regexp.MustCompile(`\r?\n`)
str = re.ReplaceAllString(str, "")
sketchData = bytes.NewBufferString(str)
}

req, err := http.NewRequest("POST", httpheader+*networkAddress+":"+*networkPort+*uploadEndpoint, sketchData)
if err != nil {
if *verbose {
fmt.Println("Error sending sketch file")
@@ -173,8 +181,12 @@ func main() {
}
}

func StreamToString(stream io.Reader) string {
func StreamToBytes(stream io.Reader) *bytes.Buffer {
buf := new(bytes.Buffer)
buf.ReadFrom(stream)
return buf.String()
return buf
}

func StreamToString(stream io.Reader) string {
return StreamToBytes(stream).String()
}

0 comments on commit 48c3355

Please sign in to comment.