package web

import (
	"math"
	"net/http"
	"strings"
	"time"

	"github.com/gorilla/mux"
	"gopkg.in/mgo.v2"
	"gopkg.in/mgo.v2/bson"

	"github.com/shunfei/cronsun"
)

type JobLog struct{}

func (jl *JobLog) GetDetail(ctx *Context) {
	vars := mux.Vars(ctx.R)
	id := strings.TrimSpace(vars["id"])
	if len(id) == 0 {
		outJSONWithCode(ctx.W, http.StatusBadRequest, "empty log id.")
		return
	}

	if !bson.IsObjectIdHex(id) {
		outJSONWithCode(ctx.W, http.StatusBadRequest, "invalid ObjectId.")
		return
	}

	logDetail, err := cronsun.GetJobLogById(bson.ObjectIdHex(id))
	if err != nil {
		statusCode := http.StatusInternalServerError
		if err == mgo.ErrNotFound {
			statusCode = http.StatusNotFound
			err = nil
		}
		outJSONWithCode(ctx.W, statusCode, err)
		return
	}

	outJSON(ctx.W, logDetail)
}

func (jl *JobLog) GetList(ctx *Context) {
	nodes := getStringArrayFromQuery("nodes", ",", ctx.R)
	names := getStringArrayFromQuery("names", ",", ctx.R)
	ids := getStringArrayFromQuery("ids", ",", ctx.R)
	begin := getTime(ctx.R.FormValue("begin"))
	end := getTime(ctx.R.FormValue("end"))
	page := getPage(ctx.R.FormValue("page"))
	failedOnly := ctx.R.FormValue("failedOnly") == "true"
	pageSize := getPageSize(ctx.R.FormValue("pageSize"))
	sort := "-beginTime"
	if ctx.R.FormValue("sort") == "1" {
		sort = "beginTime"
	}

	query := bson.M{}
	if len(nodes) > 0 {
		query["node"] = bson.M{"$in": nodes}
	}

	if len(ids) > 0 {
		query["jobId"] = bson.M{"$in": ids}
	}

	if len(names) > 0 {
		var search []bson.M
		for _, k := range names {
			k = strings.TrimSpace(k)
			if len(k) == 0 {
				continue
			}
			search = append(search, bson.M{"name": bson.M{"$regex": bson.RegEx{Pattern: k, Options: "i"}}})
		}
		query["$or"] = search
	}

	if !begin.IsZero() {
		query["beginTime"] = bson.M{"$gte": begin}
	}
	if !end.IsZero() {
		query["endTime"] = bson.M{"$lt": end.Add(time.Hour * 24)}
	}

	if failedOnly {
		query["success"] = false
	}

	var pager struct {
		Total int               `json:"total"`
		List  []*cronsun.JobLog `json:"list"`
	}
	var err error
	if ctx.R.FormValue("latest") == "true" {
		var latestLogList []*cronsun.JobLatestLog
		latestLogList, pager.Total, err = cronsun.GetJobLatestLogList(query, page, pageSize, sort)
		for i := range latestLogList {
			latestLogList[i].JobLog.Id = bson.ObjectIdHex(latestLogList[i].RefLogId)
			pager.List = append(pager.List, &latestLogList[i].JobLog)
		}
	} else {
		pager.List, pager.Total, err = cronsun.GetJobLogList(query, page, pageSize, sort)
	}
	if err != nil {
		outJSONWithCode(ctx.W, http.StatusInternalServerError, err.Error())
		return
	}

	pager.Total = int(math.Ceil(float64(pager.Total) / float64(pageSize)))
	outJSON(ctx.W, pager)
}