movie.go

package main

import (
	"fmt"
	"os"
	"strings"
	"net/http"
	"net/url"
	"io/ioutil"
	"encoding/json"
	"strconv"
	"io"
	"time"

	"./pup"
	"github.com/fatih/color"
)

type Yts struct {
	SearchResults struct {
		Status string `json:"status"`
		Movies []struct {
			Url string `json:"url"`
			Img string `json:"img"`
			Title string `json:"title"`
			Year string `json:"year"`
		} `json:"data"`
	}

	DownloadResults []struct {
		Href string `json:"href"`
		Text string `json:"text"`
		Title string `json:"Title"`
	}
}

var (
	timeout = time.Duration(5 * time.Second)
	HttpClient = http.Client{
		Timeout: timeout,
	}
	YTS = Yts{}
	MOVIES_VERSION string = "1.0.1"
	language string
)

func UrlEncoded(str string) string {
    u, err := url.Parse(str)
    if err != nil {
        return ""
    }
    return u.String()
}

func moviesResultCheck(length int){
	if length <= 0 {
		color.Set(color.FgMagenta, color.Bold)
		fmt.Println("No movies found.")
		color.Unset()
		os.Exit(0)
	}
}

func main(){
	search := ParseFlages()
	if strings.ToLower(language) == "tamil"{
		SearchTamilMovies(search)
		return
	}
	searchMovie("https://yts.am/ajax/search?query=",search)
	moviesResultCheck(len(YTS.SearchResults.Movies))
	fmt.Printf("%s",listMovies())
	downloadIndex := getConformation(len(YTS.SearchResults.Movies))
	torrentHtml := downloadHTML(YTS.SearchResults.Movies[downloadIndex].Url)
	os.Args = []string{
		"anbuksv",
		"#movie-info > p a json{}",
	}
	nodes := pup.Run(torrentHtml)
	json.Unmarshal(nodes,&YTS.DownloadResults)
	color.Set(color.FgYellow, color.Bold)
	for _,result:= range(YTS.DownloadResults){
		color.New(color.FgYellow,color.Bold).Print("("+ result.Text +") ")
		color.White(result.Href + "\n")
	}
	// fmt.Println(YTS.DownloadResults[len(YTS.DownloadResults)-1].Href) //By default high resolution torrent link will be printed
	color.Unset()
}

func ParseFlages() string {
	cmds := os.Args[1:]
	nonFlagCmds := make([]string, len(cmds))
	n := 0
	for i := 0; i < len(cmds); i++ {
		cmd := cmds[i]
		switch cmd {
		case "-h","--help":
			PrintMoviesHelp(os.Stdout, 0)
		case "--version":
			fmt.Println(MOVIES_VERSION)
			os.Exit(0)
		case "-l","--language":
			language = cmds[i+1]
			i++
		default:
			nonFlagCmds[n] = cmds[i]
			n++
		}
	}
	return strings.Join(nonFlagCmds," ")
}

func PrintMoviesHelp(w io.Writer, exitCode int) {
	helpString := `Usage
    movies Name [flags]
Version
    %s
Flags
    -h --help          display this help
    --version          display version
    -l --language      movie language
`
	fmt.Fprintf(w, helpString, MOVIES_VERSION)
	os.Exit(exitCode)
}

func getConformation(limit int) int {
	var downloadIndex int
	color.Set(color.FgWhite, color.Bold)
	fmt.Print("awesome-movie> ")
	color.Unset()
	fmt.Scan(&downloadIndex)
	downloadIndex = downloadIndex - 1
	if downloadIndex < 0 || downloadIndex >= limit{
		fmt.Println("awesome-movie> Please enter valid number.")
		return getConformation(limit)
	}
	return downloadIndex
}

func onHttpError(err error){
	color.Set(color.FgRed, color.Bold)
	fmt.Println(err)
	color.Unset()
	os.Exit(0)
}

func searchMovie(url string, query string) {
	resp,err := HttpClient.Get(url+UrlEncoded(query))
	if err != nil{
		onHttpError(err)
	}
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	json.Unmarshal([]byte(body),&YTS.SearchResults)
}

func downloadHTML(url string) io.ReadCloser {
	resp,err := HttpClient.Get(url)
	if err != nil {
		onHttpError(err)
	}
	return resp.Body
}

func listMovies() string {
	var _movies string = ""
	for index,movie := range(YTS.SearchResults.Movies){
		_movies = _movies + strconv.Itoa(index+1) + ". " + movie.Title + " (" + movie.Year + ")\n"
	}
	return _movies
}