package service

import (
	"aps_crm/model"
	"aps_crm/model/request"
	"aps_crm/pkg/ecode"
)

type FileService struct{}

func NewFileService() FileService {
	return FileService{}
}

func (FileService) GetFile(id uint) (*model.File, int) {
	file, err := model.NewFileSearch().SetId(id).First()
	if err != nil {
		return nil, ecode.DBErr
	}

	return file, ecode.OK
}

func (FileService) AddFile(fileRecord *model.File) int {

	err := model.NewFileSearch().Create(fileRecord)
	if err != nil {
		return ecode.DBErr
	}

	return ecode.OK
}

func (FileService) DeleteFile(id uint) int {
	err := model.NewFileSearch().SetId(id).Delete()
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}

func (FileService) GetFileList() ([]*model.File, int64, int) {
	list, total, err := model.NewFileSearch().Find()
	if err != nil {
		return nil, 0, ecode.DBErr
	}

	return list, total, ecode.OK
}

func (FileService) UpdateFiles(Files []*request.UpdateFile) int {
	for _, v := range Files {
		// check File exist
		_, err := model.NewFileSearch().SetId(v.Id).First()
		if err != nil {
			return ecode.DBErr
		}

		err = model.NewFileSearch().SetId(v.Id).Updates(map[string]interface{}{})
		if err != nil {
			return ecode.DBErr
		}
	}

	return ecode.OK
}

func (FileService) UpdateFile(file *model.File) int {
	err := model.NewFileSearch().SetId(file.ID).Update(file)
	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}