Add a shortenification

Author: Artur Kh

go-gist.go

@@ -53,7 +53,7 @@ func main() {
 	filetype := flag.String("t", "", "Sets the file extension and syntax type.")
 	privateFlag := flag.Bool("p", false, "Indicates whether the gist is private.")
 	desc := flag.String("d", "", "A description of the gist.")
-	shortenFlag := flag.Bool("-s", false, "Shorten the gist URL using git.io.")
+	shortenFlag := flag.Bool("s", false, "Shorten the gist URL using git.io.")
 	uid := flag.String("u", "", "Update an existing gist. Takes ID as an argument.")
 	anonymousFlag := flag.Bool("a", false, "Create an anonymous gist.")
 	copyFlag := flag.Bool("c", false, "Copy the resulting URL to the clipboard.")

util.go

@@ -4,9 +4,13 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"net/http"
+	neturl "net/url"
 	"strings"
 )
 
+const GitIOURL = "http://git.io"
+
 func GistParseError(body []byte) error {
 	var err map[string]json.RawMessage
 	if err := json.Unmarshal(body, &err); err != nil {
@@ -32,5 +36,15 @@ func GistParseError(body []byte) error {
 }
 
 func Shorten(url string) (string, error) {
-	return "", nil
+	resp, err := http.PostForm(GitIOURL, neturl.Values{"url": {url}})
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+
+	if resp.StatusCode == 201 {
+		url = resp.Header.Get("Location")
+	}
+
+	return url, nil
 }