package repository

import (
	"context"

	"next-terminal/server/model"
)

var SessionCommandRepository = new(sessionCommandRepository)

type sessionCommandRepository struct {
	baseRepository
}

func (r sessionCommandRepository) FindBySessionId(c context.Context, sessionId string) (o []model.SessionCommand, err error) {
	err = r.GetDB(c).Where("session_id = ?", sessionId).Order("created asc").Find(&o).Error
	return
}

func (r sessionCommandRepository) FindByPage(c context.Context, pageIndex, pageSize int, sessionId string) (o []model.SessionCommand, total int64, err error) {
	m := model.SessionCommand{}
	db := r.GetDB(c).Table(m.TableName())
	dbCounter := r.GetDB(c).Table(m.TableName())

	if sessionId != "" {
		db = db.Where("session_id = ?", sessionId)
		dbCounter = dbCounter.Where("session_id = ?", sessionId)
	}

	err = dbCounter.Count(&total).Error
	if err != nil {
		return nil, 0, err
	}

	err = db.Order("created desc").Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&o).Error
	if o == nil {
		o = make([]model.SessionCommand, 0)
	}
	return
}

func (r sessionCommandRepository) Create(c context.Context, o *model.SessionCommand) error {
	return r.GetDB(c).Create(o).Error
}

func (r sessionCommandRepository) DeleteBySessionId(c context.Context, sessionId string) error {
	return r.GetDB(c).Where("session_id = ?", sessionId).Delete(&model.SessionCommand{}).Error
}