package index

import (
	"golang.captainalm.com/cityuni-webserver/utils/yaml"
	"html/template"
	"math"
	"time"
)

const dateFormat = "01-2006"

type EntryYaml struct {
	Name               string        `yaml:"name"`
	Content            string        `yaml:"content"`
	StartDate          yaml.DateType `yaml:"startDate"`
	EndDate            yaml.DateType `yaml:"endDate"`
	VideoLocation      string        `yaml:"videoLocation"`
	VideoContentType   string        `yaml:"videoContentType"`
	ThumbnailLocations []string      `yaml:"thumbnailLocations"`
	ImageLocations     []string      `yaml:"imageLocations"`
}

type ImageReference struct {
	ThumbnailLocation string
	ImageLocation     string
}

func (ey EntryYaml) GetStartDate() string {
	return ey.StartDate.Format(dateFormat)
}

func (ey EntryYaml) GetEndDate() string {
	if ey.EndDate.IsZero() {
		return ""
	} else {
		return ey.EndDate.Format(dateFormat)
	}
}

func (ey EntryYaml) GetEndTime() time.Time {
	if ey.EndDate.IsZero() {
		return time.Now()
	} else {
		return ey.EndDate.Time
	}
}

func (ey EntryYaml) GetContent() template.HTML {
	return template.HTML(ey.Content)
}

func (ey EntryYaml) GetDuration() time.Duration {
	return ey.GetEndTime().Sub(ey.StartDate.Time).Truncate(time.Second)
}

func (ey EntryYaml) GetImageCount() int {
	return int(math.Min(float64(len(ey.ThumbnailLocations)), float64(len(ey.ImageLocations))))
}

func (ey EntryYaml) GetImages() []ImageReference {
	toReturn := make([]ImageReference, ey.GetImageCount())
	for i := 0; i < len(ey.ThumbnailLocations) && i < len(ey.ImageLocations); i++ {
		toReturn[i] = ImageReference{
			ThumbnailLocation: ey.ThumbnailLocations[i],
			ImageLocation:     ey.ImageLocations[i],
		}
	}
	return toReturn
}