Add timeout option

Author: Montana Flynn

README.md

@@ -192,6 +192,7 @@ module.exports.info = {
 | --------------- | ------- | ------------------------------------------------------------------------ |
 | `errorChecking` | `false` | add error checking for request, response and body                        |
 | `printBody`     | `true`  | include code to print the body as a string                               |
+| `timeout`       | `-1`    | sets a request timeout in seconds (requires go 1.3+)                     |
 
 ### HTTPie
 

src/targets/go/native.js

@@ -10,7 +10,8 @@ module.exports = function (options) {
   // Define Options
   var opts = util._extend({
     checkErrors: false,
-    printBody: true
+    printBody: true,
+    timeout: -1
   }, options);
 
   // Set some shortcuts 
@@ -39,6 +40,7 @@ module.exports = function (options) {
   code.push('package main\n');
   code.push('import (');
   code.push('\t"fmt"');
+  if (opts.timeout) code.push('\t"time"');
   if (bodyPresent) code.push('\t"strings"');
   code.push('\t"net/http"');
   if (opts.printBody) code.push('\t"io/ioutil"');
@@ -47,13 +49,20 @@ module.exports = function (options) {
   code.push('func main() {');
 
   // Create client
-  code.push('\tclient := &http.Client{}');
+  if (opts.timeout) {
+    code.push('\tclient := http.Client{');
+    code.push('\t\tTimeout: time.Duration(' + opts.timeout + ' * time.Second),');
+    code.push('\t}');
+  } else {
+    code.push('\tclient := &http.Client{}');
+  }
   code.push('\turl := "' + req.url + '"');
 
   // If we have body content or not create the var and reader or nil
   if (bodyPresent) {
-    code.push('\tpayload := ' + JSON.stringify(this.source.postData.text));
-    req.body = 'strings.NewReader(payload)';
+    var reqBody = JSON.stringify(this.source.postData.text);
+    code.push('\tpayload := strings.NewReader(' + reqBody + ')');
+    req.body = 'payload';
   } else {
     req.body = 'nil';
   }

test/fixtures/output/go/native/application-form-encoded.go

@@ -10,8 +10,8 @@ import (
 func main() {
 	client := &http.Client{}
 	url := "http://mockbin.com/har"
-	payload := "foo=bar&hello=world"
-	req, _ := http.NewRequest("POST", url, strings.NewReader(payload))
+	payload := strings.NewReader("foo=bar&hello=world")
+	req, _ := http.NewRequest("POST", url, payload)
 	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
 	res, _ := client.Do(req)
 	defer res.Body.Close()

test/fixtures/output/go/native/application-json.go

@@ -10,8 +10,8 @@ import (
 func main() {
 	client := &http.Client{}
 	url := "http://mockbin.com/har"
-	payload := "{\"foo\": \"bar\"}"
-	req, _ := http.NewRequest("POST", url, strings.NewReader(payload))
+	payload := strings.NewReader("{\"foo\": \"bar\"}")
+	req, _ := http.NewRequest("POST", url, payload)
 	req.Header.Add("Content-Type", "application/json")
 	res, _ := client.Do(req)
 	defer res.Body.Close()

test/fixtures/output/go/native/full.go

@@ -10,8 +10,8 @@ import (
 func main() {
 	client := &http.Client{}
 	url := "http://mockbin.com/har?baz=abc&foo=bar&foo=baz"
-	payload := "foo=bar"
-	req, _ := http.NewRequest("POST", url, strings.NewReader(payload))
+	payload := strings.NewReader("foo=bar")
+	req, _ := http.NewRequest("POST", url, payload)
 	req.Header.Add("Accept", "application/json")
 	req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
 	req.Header.Add("Cookie", "foo=bar; bar=baz")