package model

import (
	"apsClient/pkg/sqlitex"
	"encoding/json"
	"fmt"
	"github.com/jinzhu/gorm"
)

type (
	ProcessModelAddress struct {
		Name    string `json:"name,omitempty"`
		Address int    `json:"address,omitempty"`
	}

	// ProcessModelPlcAddress 工艺参数plc读写地址
	ProcessModelPlcAddress struct {
		CommonModel
		AddressList []*ProcessModelAddress `json:"addressList" gorm:"-"`
		Address     string                 `json:"-" gorm:"address"`
		DeviceID    string                 `json:"deviceId" gorm:"column:device_id;index;type:varchar(255)"`
	}

	ProcessModelPlcAddressSearch struct {
		ProcessModelPlcAddress
		Order     string
		PageNum   int
		PageSize  int
		Keyword   string
		Orm       *gorm.DB
		DeviceIDs []string
	}
)

func (slf *ProcessModelPlcAddress) TableName() string {
	return "process_model_plc_address"
}

func (slf *ProcessModelPlcAddress) AfterFind(*gorm.DB) error {
	if slf.Address != "" {
		return json.Unmarshal([]byte(slf.Address), &slf.AddressList)
	}
	return nil
}

func NewProcessModelPlcAddressSearch() *ProcessModelPlcAddressSearch {
	return &ProcessModelPlcAddressSearch{Orm: sqlitex.GetDB()}
}

func (slf *ProcessModelPlcAddressSearch) SetOrm(tx *gorm.DB) *ProcessModelPlcAddressSearch {
	slf.Orm = tx
	return slf
}

func (slf *ProcessModelPlcAddressSearch) SetOrder(order string) *ProcessModelPlcAddressSearch {
	slf.Order = order
	return slf
}

func (slf *ProcessModelPlcAddressSearch) SetID(id uint) *ProcessModelPlcAddressSearch {
	slf.ID = id
	return slf
}

func (slf *ProcessModelPlcAddressSearch) SetDeviceID(id string) *ProcessModelPlcAddressSearch {
	slf.DeviceID = id
	return slf
}

func (slf *ProcessModelPlcAddressSearch) SetDeviceIDs(ids []string) *ProcessModelPlcAddressSearch {
	slf.DeviceIDs = ids
	return slf
}

func (slf *ProcessModelPlcAddressSearch) build() *gorm.DB {
	var db = slf.Orm.Table(slf.TableName())

	if slf.ID != 0 {
		db = db.Where("id = ?", slf.ID)
	}

	if slf.Order != "" {
		db = db.Order(slf.Order)
	}

	if slf.DeviceID != "" {
		db = db.Where("device_id = ?", slf.DeviceID)
	}

	if len(slf.DeviceIDs) > 0 {
		db = db.Where("device_id in (?)", slf.DeviceIDs)
	}

	return db
}

// Create 单条插入
func (slf *ProcessModelPlcAddressSearch) Create(record *ProcessModelPlcAddress) error {
	var db = slf.build()

	if err := db.Create(record).Error; err != nil {
		return fmt.Errorf("create err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ProcessModelPlcAddressSearch) Save(record *ProcessModelPlcAddress) error {
	var db = slf.build()

	if err := db.Omit("CreatedAt").Save(record).Error; err != nil {
		return fmt.Errorf("save err: %v, record: %+v", err, record)
	}

	return nil
}

func (slf *ProcessModelPlcAddressSearch) UpdateByMap(upMap map[string]interface{}) error {
	var (
		db = slf.build()
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by map err: %v, upMap: %+v", err, upMap)
	}

	return nil
}

func (slf *ProcessModelPlcAddressSearch) UpdateByQuery(query string, args []interface{}, upMap map[string]interface{}) error {
	var (
		db = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Updates(upMap).Error; err != nil {
		return fmt.Errorf("update by query err: %v, query: %s, args: %+v, upMap: %+v", err, query, args, upMap)
	}

	return nil
}

func (slf *ProcessModelPlcAddressSearch) Delete() error {
	var db = slf.build()

	if err := db.Unscoped().Delete(&ProcessModelPlcAddress{}).Error; err != nil {
		return err
	}

	return nil
}

func (slf *ProcessModelPlcAddressSearch) First() (*ProcessModelPlcAddress, error) {
	var (
		record = new(ProcessModelPlcAddress)
		db     = slf.build()
	)

	if err := db.First(record).Error; err != nil {
		return record, err
	}

	return record, nil
}

func (slf *ProcessModelPlcAddressSearch) Find() ([]*ProcessModelPlcAddress, int64, error) {
	var (
		records = make([]*ProcessModelPlcAddress, 0)
		total   int64
		db      = slf.build()
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find count err: %v", err)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find records err: %v", err)
	}

	return records, total, nil
}

func (slf *ProcessModelPlcAddressSearch) FindNotTotal() ([]*ProcessModelPlcAddress, error) {
	var (
		records = make([]*ProcessModelPlcAddress, 0)
		db      = slf.build()
	)

	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find records err: %v", err)
	}

	return records, nil
}

// FindByQuery 指定条件查询.
func (slf *ProcessModelPlcAddressSearch) FindByQuery(query string, args []interface{}) ([]*ProcessModelPlcAddress, int64, error) {
	var (
		records = make([]*ProcessModelPlcAddress, 0)
		total   int64
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Count(&total).Error; err != nil {
		return records, total, fmt.Errorf("find by query count err: %v", err)
	}
	if err := db.Find(&records).Error; err != nil {
		return records, total, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, total, nil
}

// FindByQueryNotTotal 指定条件查询&不查询总条数.
func (slf *ProcessModelPlcAddressSearch) FindByQueryNotTotal(query string, args []interface{}) ([]*ProcessModelPlcAddress, error) {
	var (
		records = make([]*ProcessModelPlcAddress, 0)
		db      = slf.Orm.Table(slf.TableName()).Where(query, args...)
	)

	if err := db.Find(&records).Error; err != nil {
		return records, fmt.Errorf("find by query records err: %v, query: %s, args: %+v", err, query, args)
	}

	return records, nil
}