package main

import (
	"bytes"
	"encoding/json"
	"net/http"

	"github.com/zenazn/goji/web"
)

type shortenedURL struct {
	Kind    string `json:"kind"`
	ID      string `json:"id"`
	LongURL string `json:"longUrl"`
}

func shortURLHandler(c web.C, w http.ResponseWriter, r *http.Request) {
	fileName := c.URLParams["name"]

	err := checkFile(fileName)
	if err == NotFoundErr {
		notFoundHandler(c, w, r)
		return
	}

	metadata, err := metadataRead(fileName)
	if err != nil {
		oopsHandler(c, w, r, RespAUTO, "Corrupt metadata.")
		return
	}

	if metadata.ShortURL == "" {
		url, err := shortenURL(getSiteURL(r) + fileName)
		if err != nil {
			oopsHandler(c, w, r, RespAUTO, "Something went wrong")
			return
		}

		metadata.ShortURL = url

		err = metadataWrite(fileName, &metadata)
		if err != nil {
			oopsHandler(c, w, r, RespAUTO, "Corrupt metadata.")
			return
		}
	}

	js, _ := json.Marshal(map[string]string{
		"shortUrl": metadata.ShortURL,
	})
	w.Write(js)
	return
}

func shortenURL(url string) (string, error) {
	apiURL := "https://www.googleapis.com/urlshortener/v1/url"
	if Config.googleShorterAPIKey != "" {
		apiURL += "?key=" + Config.googleShorterAPIKey
	}

	jsonStr, _ := json.Marshal(shortenedURL{LongURL: url})

	req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(jsonStr))
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	shortened := new(shortenedURL)
	err = json.NewDecoder(resp.Body).Decode(shortened)
	if err != nil {
		return "", err
	}

	return shortened.ID, err
}