package models

import (
	"errors"
	"strconv"
)

func (m *BotModel) GetBytes(path []string) ([]byte, error) {
	var err error
	var v []byte
	// Value is not cached, try to pull it from the DB
	if len(path) > 2 {
		path, key := path[:len(path)-1], path[len(path)-1]
		v, err = m.db.GetBytes(path, key)
		if err != nil {
			return nil, err
		}
	}
	return v, nil
}

func (m *BotModel) SetBytes(path []string, val []byte) error {
	if len(path) > 1 {
		path, key := path[:len(path)-1], path[len(path)-1]
		err := m.db.SetBytes(path, key, val)
		if err != nil {
			return err
		}
		return nil
	}
	return errors.New("Invalid path")
}

func (m *BotModel) GetString(path []string) (string, error) {
	bts, err := m.GetBytes(path)
	if err != nil {
		return "", err
	}
	if len(bts) == 0 {
		return "", nil
	}
	return string(bts), nil
}

func (m *BotModel) SetString(path []string, val string) error {
	return m.SetBytes(path, []byte(val))
}

func (m *BotModel) GetInt(path []string) (int, error) {
	bts, err := m.GetBytes(path)
	if err != nil {
		return 0, err
	}
	if len(bts) == 0 {
		return 0, nil
	}
	return strconv.Atoi(string(bts))
}

func (m *BotModel) SetInt(path []string, val int) error {
	return m.SetString(path, strconv.Itoa(val))
}