package team

import (
	"github.com/shylinux/icebergs"
	"github.com/shylinux/icebergs/base/web"
	"github.com/shylinux/toolkits"

	"encoding/csv"
	"os"
	"time"
)

const (
	ZONE = "zone"
	PLAN = "plan"
	TASK = "task"
	MISS = "miss"

	EXPORT = "usr/export/web.team/task.csv"
)

const (
	StatusPrepare = "prepare"
	StatusProcess = "process"
	StatusCancel  = "cancel"
	StatusFinish  = "finish"
)

func _task_list(m *ice.Message, zone string, id string, field ...interface{}) {
	m.Richs(TASK, nil, kit.Select("*", zone), func(key string, val map[string]interface{}) {
		if zone = kit.Format(kit.Value(val, "meta.zone")); id == "" {
			m.Grows(TASK, kit.Keys("hash", key), "", "", func(index int, value map[string]interface{}) {
				m.Push("zone", zone)
				m.Push(zone, value)
			})
			return
		}
		m.Grows(TASK, kit.Keys("hash", key), "id", id, func(index int, value map[string]interface{}) {
			m.Push("detail", value)
		})
	})
}

func _task_create(m *ice.Message, zone string) {
	if m.Richs(TASK, nil, zone, nil) == nil {
		m.Rich(TASK, nil, kit.Data(ZONE, zone))
		m.Log_CREATE("zone", zone)
	}
}
func _task_export(m *ice.Message, file string) {
	f, p, e := kit.Create(file)
	m.Assert(e)
	defer f.Close()

	w := csv.NewWriter(f)
	defer w.Flush()

	m.Assert(w.Write([]string{
		"zone", "id", "time",
		"type", "name", "text",
		"level", "status", "score",
		"begin_time", "close_time",
		"extra",
	}))
	count := 0
	m.Option("cache.limit", -2)
	m.Richs(TASK, nil, "*", func(key string, val map[string]interface{}) {
		m.Grows(TASK, kit.Keys("hash", key), "", "", func(index int, value map[string]interface{}) {
			m.Assert(w.Write(kit.Simple(
				kit.Format(kit.Value(val, "meta.zone")),
				kit.Format(value["id"]),
				kit.Format(value["time"]),
				kit.Format(value["type"]),
				kit.Format(value["name"]),
				kit.Format(value["text"]),
				kit.Format(value["level"]),
				kit.Format(value["status"]),
				kit.Format(value["score"]),
				kit.Format(value["begin_time"]),
				kit.Format(value["close_time"]),
				kit.Format(value["extra"]),
			)))
			count++
		})
	})
	m.Log_EXPORT("file", p, "count", count)
}
func _task_import(m *ice.Message, file string) {
	f, e := os.Open(file)
	m.Assert(e)
	defer f.Close()

	r := csv.NewReader(f)
	heads, _ := r.Read()
	count := 0
	for {
		lines, e := r.Read()
		if e != nil {
			break
		}

		zone := ""
		data := kit.Dict()
		for i, k := range heads {
			switch k {
			case "zone":
				zone = lines[i]
			case "id":
				continue
			case "extra":
				kit.Value(data, k, kit.UnMarshal(lines[i]))
			default:
				kit.Value(data, k, lines[i])
			}
		}

		_task_create(m, zone)
		m.Richs(TASK, nil, zone, func(key string, value map[string]interface{}) {
			id := m.Grow(TASK, kit.Keys("hash", key), data)
			m.Log_INSERT("zone", zone, "id", id)
			count++
		})
	}
	m.Log_IMPORT("file", file, "count", count)
}

func _task_insert(m *ice.Message, zone, kind, name, text, begin_time, close_time string, arg ...string) {
	m.Richs(TASK, nil, zone, func(key string, value map[string]interface{}) {
		id := m.Grow(TASK, kit.Keys("hash", key), kit.Dict(
			kit.MDB_TYPE, kind, kit.MDB_NAME, name, kit.MDB_TEXT, text,
			"begin_time", begin_time, "close_time", begin_time,
			"status", StatusPrepare, "level", 3, "score", 3,
			kit.MDB_EXTRA, kit.Dict(arg),
		))
		m.Log_INSERT("zone", zone, "id", id, "type", kind, "name", name)
		m.Echo("%d", id)
	})
}
func _task_modify(m *ice.Message, zone, id, pro, set, old string) {
	m.Richs(TASK, nil, kit.Select("*", zone), func(key string, val map[string]interface{}) {
		m.Grows(TASK, kit.Keys("hash", key), "id", id, func(index int, value map[string]interface{}) {
			switch pro {
			case "id", "time":
				m.Info("not allow %v", key)
			case "status":
				if value["status"] == set {
					break
				}
				switch value["status"] {
				case StatusCancel, StatusFinish:
					m.Info("not allow %v", key)
					return
				}
				switch set {
				case StatusProcess:
					kit.Value(value, "begin_time", m.Time())
				case StatusCancel, StatusFinish:
					kit.Value(value, "close_time", m.Time())
				}
				fallthrough
			default:
				m.Log_MODIFY("zone", zone, "id", id, "key", pro, "value", set, "old", old)
				kit.Value(value, pro, set)
			}
		})
	})
}
func _task_delete(m *ice.Message, zone, id string) {
	m.Richs(TASK, nil, kit.Select("*", zone), func(key string, val map[string]interface{}) {
		m.Grows(TASK, kit.Keys("hash", key), "id", id, func(index int, value map[string]interface{}) {
			m.Log_DELETE("zone", zone, "id", id)
			kit.Value(value, "status", "cancel")
		})
	})
}

var Index = &ice.Context{Name: "team", Help: "团队中心",
	Configs: map[string]*ice.Config{
		TASK: {Name: "task", Help: "task", Value: kit.Data(kit.MDB_SHORT, "zone")},
	},
	Commands: map[string]*ice.Command{
		ice.ICE_INIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) { m.Load() }},
		ice.ICE_EXIT: {Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) { m.Save("task") }},

		PLAN: {Name: "plan scale:select=day|week|month|year|long begin_time=@date end_time=@date auto", Help: "计划", Meta: kit.Dict(
			"display", "/plugin/local/team/miss.js", "detail", []string{"prepare", "process", "finish", "cancel"},
		), Action: map[string]*ice.Action{
			"insert": {Name: "insert zone type name text begin_time end_time", Help: "添加", Hand: func(m *ice.Message, arg ...string) {
				_task_create(m, arg[0])
				_task_insert(m, arg[0], arg[1], arg[2], arg[3], arg[4], arg[5])
			}},
			"delete": {Name: "delete key value", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
				_task_delete(m, m.Option("zone"), m.Option("id"))
			}},
			"modify": {Name: "modify key value old", Help: "编辑", Hand: func(m *ice.Message, arg ...string) {
				switch arg[0] {
				case "status":

				}
				_task_modify(m, m.Option("zone"), m.Option("id"), arg[0], arg[1], arg[2])
			}},
			"import": {Name: "import file...", Help: "导入", Hand: func(m *ice.Message, arg ...string) {
				_task_import(m, kit.Select(EXPORT, arg, 0))
			}},
			"export": {Name: "export file...", Help: "导出", Hand: func(m *ice.Message, arg ...string) {
				_task_export(m, kit.Select(EXPORT, arg, 0))
			}},
		}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
			begin_time := time.Now()
			if len(arg) > 1 {
				begin_time, _ = time.ParseInLocation("2006-01-02 15:04:05", arg[1], time.Local)
			}
			end_time := begin_time

			switch begin_time = begin_time.Add(-time.Duration(begin_time.UnixNano()) % (24 * time.Hour)); arg[0] {
			case "day":
				end_time = begin_time.AddDate(0, 0, 1)
			case "week":
				begin_time = begin_time.AddDate(0, 0, -int(begin_time.Weekday()))
				end_time = begin_time.AddDate(0, 0, 7)
			case "month":
				begin_time = begin_time.AddDate(0, 0, -begin_time.Day()+1)
				end_time = begin_time.AddDate(0, 1, 0)
			case "year":
				begin_time = begin_time.AddDate(0, 0, -begin_time.YearDay()+1)
				end_time = begin_time.AddDate(1, 0, 0)
			case "long":
				begin_time = begin_time.AddDate(0, 0, -begin_time.YearDay()+1)
				begin_time = begin_time.AddDate(-begin_time.Year()%5, 0, 0)
				end_time = begin_time.AddDate(5, 0, 0)
			}

			begin_time = begin_time.Add(-8 * time.Hour)
			end_time = end_time.Add(-8 * time.Hour)
			m.Debug("begin: %v end: %v", begin_time, end_time)

			m.Richs(TASK, nil, kit.Select("*", m.Option("zone")), func(key string, val map[string]interface{}) {
				zone := kit.Format(kit.Value(val, "meta.zone"))
				m.Grows(TASK, kit.Keys("hash", key), "", "", func(index int, value map[string]interface{}) {
					begin, _ := time.ParseInLocation("2006-01-02 15:04:05", kit.Format(value["begin_time"]), time.Local)
					if begin_time.Before(begin) && begin.Before(end_time) {
						m.Push("zone", zone)
						m.Push(zone, value)
					}
				})
			})
		}},
		TASK: {Name: "task zone=auto id=auto auto", Help: "任务", Action: map[string]*ice.Action{
			"delete": {Name: "delete key value", Help: "删除", Hand: func(m *ice.Message, arg ...string) {
				_task_delete(m, m.Option("zone"), m.Option("id"))
			}},
			"modify": {Name: "modify key value old", Help: "编辑", Hand: func(m *ice.Message, arg ...string) {
				_task_modify(m, m.Option("zone"), m.Option("id"), arg[0], arg[1], arg[2])
			}},
		}, Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
			if len(arg) < 3 {
				_task_list(m, kit.Select("", arg, 0), kit.Select("", arg, 1))
				return
			}

			if m.Richs(TASK, nil, arg[0], nil) == nil {
				_task_create(m, arg[0])
			}
			if len(arg) == 5 {
				arg = append(arg, m.Time())
			}
			if len(arg) == 6 {
				arg = append(arg, m.Time("1h"))
			}

			_task_insert(m, arg[0], arg[2], arg[3], arg[4], arg[5], arg[6], arg[7:]...)
		}},
		MISS: {Name: "miss", Help: "miss", Hand: func(m *ice.Message, c *ice.Context, cmd string, arg ...string) {
		}},
	},
}

func init() { web.Index.Register(Index, &web.Frame{}) }