image-service.go

package main

import (
	_"fmt"
	"strings"
	"io"
	"io/ioutil"
	"net/http"
	_"net/url"
	"os"
)


type queryParams struct {
	query string
}

func health(w http.ResponseWriter, r *http.Request) {
	io.WriteString(w, "I am alive!")
}

func (s queryParams) createSeachURL() string {
	
	searchURL := []string{"https://www.googleapis.com/customsearch/v1?"}
	query := make(map[string]string)
	query["q"] = s.query
	query["searchType"] = "image"
	query["fields"] = "items(link)"
	// add your google api keys
	query["cx"] = ""
	query["key"] = ""
	// override defaults
	if os.Getenv("GOOGLE_CX") != "" {
		query["cx"] = os.Getenv("GOOGLE_CX")
	}
	if os.Getenv("GOOGLE_KEY") != "" {
		query["key"] = os.Getenv("GOOGLE_KEY")
	}


	for k, v := range query {
		searchURL = append(searchURL, k,"=",v,"&")
	}

	return strings.Join(searchURL, "")

}

func image (w http.ResponseWriter, r *http.Request) {
	//image?search=XXXX
	query := queryParams{r.URL.Query().Get("search")}
	imageSearch := query.createSeachURL()

	request, err := http.Get(imageSearch)
	if err != nil {
		panic(err)
	}
	if request.StatusCode != http.StatusOK {
		panic("http status code not 200")
	} 
	defer request.Body.Close()
	content, _ := ioutil.ReadAll(request.Body)
	w.Header().Set("Content-Type", "application/json")
	w.Write([]byte(content))
}

func main() {
	http.HandleFunc("/", health)
	http.HandleFunc("/image", image)
	http.ListenAndServe(":8888", nil)
}