movies.go

package imdb

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"strings"
)

// Movie supports the lookup of IMDB movies by title.
func Movie(title string) {
	url := "http://www.omdbapi.com?apikey=" + os.Getenv("OMDBKEY") + "&t="
	term := strings.ReplaceAll(title, " ", "+")
	resp, err := http.Get(url + term + "&plot=short")
	if err != nil {
		fmt.Printf("[!] Error in request: %s", err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("[!] Error in reading response body: %s", err)
	}
	cartographer(body)
}

// cartographer is used to handle conversions from byte string to marshaled json bytes.
func cartographer(body []byte) []byte {
	type RatingsType struct {
		Source string `json:"Source"`
		Value  string `json:"Value"`
	}
	type Movies struct {
		Title      string `json:"Title"`
		Year       string `json:"Year"`
		Rated      string `json:"Rated"`
		Released   string `json:"Released"`
		Runtime    string `json:"Runtime"`
		Genre      string `json:"Genre"`
		Director   string `json:"Director"`
		Writer     string `json:"Writer"`
		Actors     string `json:"Actors"`
		Plot       string `json:"Plot"`
		Language   string `json:"Language"`
		Country    string `json:"Country"`
		Awards     string `json:"Awards"`
		Poster     string `json:"Poster"`
		Ratings    []RatingsType
		Metascore  string `json:"Metascore"`
		ImdbRating string `json:"imdbRating"`
		ImdbVotes  string `json:"imdbVotes"`
		ImdbID     string `json:"imdbID"`
		Type       string `json:"Type"`
		DVD        string `json:"DVD"`
		BoxOffice  string `json:"BoxOffice"`
		Production string `json:"Production"`
		Website    string `json:"Website"`
		Response   string `json:"Response"`
	}
	data := &Movies{}
	err := json.Unmarshal(body, data)
	if err != nil {
		fmt.Println("[!] Error unmarshaling: ", err)
	}
	b, _ := json.Marshal(data)
	return b
}