package repository

import (
	"time"

	"gat1400Exchange/models"
	"gat1400Exchange/vo"
)

type ApeRepository struct {
}

func NewApeRepository() ApeRepository {
	return ApeRepository{}
}

func (a *ApeRepository) Keepalive(id string) error {
	var ape models.Ape

	// 设备存在
	if err := ape.FindById(id); err != nil {
		return nil
	}

	return ape.Keepalive()
}

func (a *ApeRepository) Create(id string) error {
	var ape models.Ape

	// 设备存在
	if err := ape.FindById(id); err == nil {
		return nil
	}

	ape.Id = id
	ape.Name = id
	ape.HeartbeatTime = time.Now().Format("2006-01-02 15:04:05")
	ape.Ext = vo.Ape{
		ApeID:            id,
		Name:             "",
		Model:            "",
		IPAddr:           "",
		IPV6Addr:         "",
		Port:             0,
		Longitude:        0,
		Latitude:         0,
		PlaceCode:        "",
		Place:            "",
		OrgCode:          "",
		CapDirection:     0,
		MonitorDirection: "",
		MonitorAreaDesc:  "",
		IsOnline:         "2",
		OwnerApsID:       "",
		UserID:           "",
		Password:         "",
		FunctionType:     "2",
	}

	return ape.Save()
}

func (a *ApeRepository) List() ([]models.Ape, error) {
	var ape models.Ape

	return ape.FindAll()
}

func (a *ApeRepository) Update(req *vo.Ape) error {
	var ape models.Ape

	err := ape.FindById(req.ApeID)
	if err != nil {
		return err
	}

	ape.Name = req.Name
	ape.Ext = *req

	return ape.Save()
}