package ytparser

import (
	"text/template"
	"encoding/json"
	"fmt"
	"net/http"
	"bytes"
	"io"
	"io/ioutil"
	"strings"
	"net/url"
)

const initDataString string = "window[\"ytInitialData\"] = "
const baseUrl string = "https://youtube.com"

type Item struct {
	Id    string
	Title string
	Url   string
	Thumb string
	ChannelTitle string
	ChannelUrl string
}

func (item Item) Format(t *template.Template) string {
	var b strings.Builder
	err := t.Execute(&b, item)
	if err != nil {
		panic(err)
	}
	return b.String()
}

func (item Item) String() string {
	return fmt.Sprintf("id: %s, title: %s, url: %s, thumb: %s", item.Id, item.Title, item.Url, item.Thumb)
}

func parsejson(data string) ([]Item, error) {
	//fmt.Println(data)
	dec := json.NewDecoder(strings.NewReader(data))


	depth := 0
	isArray := false
	isValue := false
	var items []Item
	var item Item
	var names []string
	nbItems := 0
	for {
		tok, err := dec.Token()
		if err == io.EOF {
			break
		} else if err != nil {
			return nil, err
		}

		switch t := tok.(type) {
		case json.Delim:
			if t == '{' {
				depth++
			} else if t == '}' {
				depth--
				names = names[:depth]
			}
			isArray = t == '['
			isValue = false
		case string:
			if !isArray {
				if !isValue {
					if t == "videoRenderer" {
						if nbItems > 0 {
							items = append(items, item)
						}
						item = Item{}
						nbItems++
					}
					if depth > len(names) {
						names = append(names, t)
					} else {
						names[depth - 1] = t
					}
					//fmt.Println(t, depth, len(names), names[depth - 1])
					isValue = true
				} else {
					//fmt.Println(names[len(names) - 1])
					if names[depth-1] == "videoId" {
						item.Id = t
						item.Url = fmt.Sprintf("https://youtube.com/watch?v=%s", t)
					}
					if depth >= 3 && names[depth-3] == "title" &&
						names[depth-2] == "runs" &&
						names[depth-1] == "text" {
						item.Title = t
					}
					if depth >= 3 && names[depth-3] == "ownerText" &&
						names[depth-2] == "runs" &&
						names[depth-1] == "text" {
						item.ChannelTitle = t
					}
					if depth >= 6 &&
						names[depth-6] == "ownerText" &&
						names[depth-5] == "runs" &&
						names[depth-4] == "navigationEndpoint" &&
						names[depth-3] == "commandMetadata" &&
						names[depth-2] == "webCommandMetadata" &&
						names[depth-1] == "url" {
						item.ChannelUrl = baseUrl + t
					}
					if depth >= 4 &&
						names[depth-4] == "videoRenderer" &&
						names[depth-3] == "thumbnail" &&
						names[depth-2] == "thumbnails" &&
						names[depth-1] == "url" {
							item.Thumb = t
					}
					isValue = false
				}
			}
		default:
		}
	}
	return items, nil
	//fmt.Println(names)
}

func PrintItems(items []Item, format string) {
	t := template.Must(template.New("items").Parse(format))
	for _, i := range items {
		fmt.Println(i.Format(t))
		//fmt.Println(i)
	}
}

func request(query string, page int) (string, error) {
	q := url.QueryEscape(query)
	url := fmt.Sprintf("https://www.youtube.com/results?search_query=%s&page=%d", q, page)
	res, err := http.DefaultClient.Get(url)
	if err != nil {
		return "", err
	}
	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		return "", err
	}
	idx := bytes.Index(body, []byte(initDataString))
	idx += len(initDataString)
	startData := body[idx:]
	idx = bytes.Index(startData, []byte(";\n"))
	startData = startData[:idx]
	//fmt.Println(string(startData))
	return string(startData), nil
}

func Search(query string, page int) ([]Item, error) {
	data, err := request(query, page)
	if err != nil {
		return nil, err
	}
	return parsejson(data)
}