main.go

package main

import (
    "encoding/json"
    "fmt"
    "html/template"
    "net/http"
    "net/url"
    "os"
)

type Dialog struct {
    Episodename string `json:"episodename"`
    Act         string `json:"act"`
    Scenenumber string `json:"scenenumber"`
    Texttype    string `json:"texttype"`
    Who         string `json:"who"`
    Text        string `json:"text"`
    Speech      string `json:"speech"`
    Released    string `json:"released"`
    Episode     string `json:"episode"`
    Imdbrating  string `json:"imdbrating"`
    Imdbid      string `json:"imdbid"`
    Season      string `json:"season"`
}

type Results struct {
    SearchKey string
    Lines     []Dialog
    Images    map[string]string
}

type TMDBQuery struct {
    MovieResults     []string `json:"-"`
    PersonResults    []string `json:"-"`
    TvResults        []string `json:"-"`
    TvEpisodeResults []struct {
        AirDate        string  `json:"air_date"`
        EpisodeNumber  int     `json:"episode_number"`
        ID             int     `json:"id"`
        Name           string  `json:"name"`
        Overview       string  `json:"overview"`
        ProductionCode string  `json:"production_code"`
        SeasonNumber   int     `json:"season_number"`
        ShowID         int     `json:"show_id"`
        StillPath      string  `json:"still_path"`
        VoteAverage    float64 `json:"vote_average"`
        VoteCount      int     `json:"vote_count"`
    } `json:"tv_episode_results"`
    TvSeasonResults string `json:"-"`
}

var API_KEY string
var tpl = template.Must(template.ParseFiles("index.html"))

func indexHandler(w http.ResponseWriter, r *http.Request) {
    tpl.Execute(w, nil)
}

func searchHandler(w http.ResponseWriter, r *http.Request) {
    u, err := url.Parse(r.URL.String())
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
        w.Write([]byte("Internal server error"))
        return
    }

    params := u.Query()
    searchKey := params.Get("q")

    results := &Results{}
    results.Images = make(map[string]string)
    results.SearchKey = searchKey

    endpoint := fmt.Sprintf("https://tngapi-awicwils6q-ew.a.run.app/?q=%s", url.QueryEscape(searchKey))
    resp, err := http.Get(endpoint)
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
        return
    }

    defer resp.Body.Close()
    if resp.StatusCode != 200 {
        w.WriteHeader(http.StatusInternalServerError)
        return
    }

    err = json.NewDecoder(resp.Body).Decode(&results.Lines)
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
        return
    }

    for _, d := range results.Lines {
        tmdbquery := &TMDBQuery{}
        imdbid := d.Imdbid
        _, ok := results.Images[imdbid]
        if !ok {
            tmdbep := fmt.Sprintf("https://api.themoviedb.org/3/find/%s?api_key=%s&language=en-US&external_source=imdb_id", imdbid, API_KEY)
            tmdbresp, err := http.Get(tmdbep)
            if err != nil {
                w.WriteHeader(http.StatusInternalServerError)
                return
            }
            defer tmdbresp.Body.Close()
            if tmdbresp.StatusCode != 200 {
                w.WriteHeader(http.StatusInternalServerError)
                return
            }

            err = json.NewDecoder(tmdbresp.Body).Decode(&tmdbquery)

            if err != nil {
                w.WriteHeader(http.StatusInternalServerError)
                return
            }
            if len(tmdbquery.TvEpisodeResults) > 0 {
                results.Images[imdbid] = tmdbquery.TvEpisodeResults[0].StillPath
            } else {
                results.Images[imdbid] = "8do6gZErem4wfdPwALiT8agtJfb.jpg"
            }
        }
    }

    err = tpl.Execute(w, results)
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
    }

}

func main() {
    API_KEY = os.Getenv("API_KEY")
    if API_KEY == "" {
        fmt.Println("No API_KEY in environment")
        os.Exit(1)
    }
    mux := http.NewServeMux()
    fs := http.FileServer(http.Dir("assets"))
    mux.Handle("/assets/", http.StripPrefix("/assets/", fs))
    mux.HandleFunc("/search", searchHandler)
    mux.HandleFunc("/", indexHandler)
    http.ListenAndServe(":8080", mux)
}