package internal

import (
	"encoding/json"
	"strconv"
	"time"
)

type textTime struct {
	time.Time
}

func (t textTime) MarshalJSON() ([]byte, error) {
	return json.Marshal(t.Format("2006-01-02 15:04:05 -0700"))
}
func (t *textTime) UnmarshalJSON(b []byte) error {
	s := string(b)
	t2, err := time.Parse(`"2006-01-02 15:04:05 -0700"`, s)
	*t = textTime{t2}
	return err
}

type stampTime struct {
	time.Time
}

func (t stampTime) MarshalJSON() ([]byte, error) {
	return json.Marshal(t.Unix())
}
func (t *stampTime) UnmarshalJSON(b []byte) error {
	ts, err := strconv.Atoi(string(b))
	if err != nil {
		return err
	}
	*t = stampTime{time.Unix(int64(ts), 0)}
	return err
}

// WebMirrorStatus is the mirror status to be shown in the web page
type WebMirrorStatus struct {
	Name          string     `json:"name"`
	IsMaster      bool       `json:"is_master"`
	Status        SyncStatus `json:"status"`
	LastUpdate    textTime   `json:"last_update"`
	LastUpdateTs  stampTime  `json:"last_update_ts"`
	LastStarted   textTime   `json:"last_started"`
	LastStartedTs stampTime  `json:"last_started_ts"`
	LastEnded     textTime   `json:"last_ended"`
	LastEndedTs   stampTime  `json:"last_ended_ts"`
	Scheduled     textTime   `json:"next_schedule"`
	ScheduledTs   stampTime  `json:"next_schedule_ts"`
	Upstream      string     `json:"upstream"`
	Size          string     `json:"size"` // approximate size
}

func BuildWebMirrorStatus(m MirrorStatus) WebMirrorStatus {
	return WebMirrorStatus{
		Name:          m.Name,
		IsMaster:      m.IsMaster,
		Status:        m.Status,
		LastUpdate:    textTime{m.LastUpdate},
		LastUpdateTs:  stampTime{m.LastUpdate},
		LastStarted:   textTime{m.LastStarted},
		LastStartedTs: stampTime{m.LastStarted},
		LastEnded:     textTime{m.LastEnded},
		LastEndedTs:   stampTime{m.LastEnded},
		Scheduled:     textTime{m.Scheduled},
		ScheduledTs:   stampTime{m.Scheduled},
		Upstream:      m.Upstream,
		Size:          m.Size,
	}
}