sqlite.go

package wife

import (
	"strconv"
	"sync"
	"time"

	sql "github.com/FloatTech/sqlite"
)

// BlackListStruct (Example: blacklist_1292581422)
type BlackListStruct struct {
	BlackList int64 `db:"blacklist"`
}

// DisabledListStruct (Example: disabled_1292581422)
type DisabledListStruct struct {
	DisabledList int64 `db:"disabledlist"`
}

// OrderListStruct (Example: orderlist_1145141919180)
type OrderListStruct struct {
	OrderPerson  int64  `db:"order"`
	TargerPerson int64  `db:"target"`
	Time         string `db:"time"`
}

// GlobalDataStruct (Example: grouplist_1292581422 )
type GlobalDataStruct struct {
	PairKey  string `db:"pairkey"`
	UserID   int64  `db:"userid"`
	TargetID int64  `db:"targetid"`
	Time     string `db:"time"`
}

// PairKeyStruct pairkey is used to check the list.
type PairKeyStruct struct {
	PairKey  string `db:"pairkey"`
	StatusID int64  `db:"statusid"`
}

// ChosenList User Refer To Theirs username and save it by using the quick additon (table: chosenList) be careful that it's global.
type ChosenList struct {
	IdentifyID string `db:"id"`
	UserID     int64  `db:"userid"`
	TargetName string `db:"targetname"`
	TargetID   int64  `db:"targetid"`
}

var (
	marryList   = &sql.Sqlite{}
	marryLocker = sync.Mutex{}
)

func init() {
	// init UserDataBase
	marryList.DBPath = engine.DataFolder() + "data.db"
	err := marryList.Open(time.Hour * 24)
	if err != nil {
		panic(err)
	}
}

/*
SQL HANDLER
*/

// FormatInsertUserGlobalMarryList Format Insert
func FormatInsertUserGlobalMarryList(UserID int64, targetID int64, PairKeyRaw string) *GlobalDataStruct {
	return &GlobalDataStruct{PairKey: PairKeyRaw, UserID: UserID, TargetID: targetID, Time: strconv.FormatInt(time.Now().Unix(), 10)}
}

// FormatPairKey Format PairKey
func FormatPairKey(PairKeyRaw string, statusID int64) *PairKeyStruct {
	return &PairKeyStruct{PairKey: PairKeyRaw, StatusID: statusID}
}

// FormatBlackList Format BlackList
func FormatBlackList(blockID int64) *BlackListStruct {
	return &BlackListStruct{BlackList: blockID}
}

// FormatDisabledList Format DisabledList
func FormatDisabledList(disabledID int64) *DisabledListStruct {
	return &DisabledListStruct{DisabledList: disabledID}
}

// FormatOrderList Format OrderList
func FormatOrderList(orderPersonal int64, targetPersonal int64, time string) *OrderListStruct {
	return &OrderListStruct{OrderPerson: orderPersonal, TargerPerson: targetPersonal, Time: time}
}

// InsertUserGlobalMarryList no check,use it with caution // Only insert and do nothing.
func InsertUserGlobalMarryList(db *sql.Sqlite, groupID int64, UserID int64, targetID int64, StatusID int64, PairKeyRaw string) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	formatList := FormatInsertUserGlobalMarryList(UserID, targetID, PairKeyRaw)
	err := db.Insert("grouplist_"+strconv.FormatInt(groupID, 10), formatList)
	if err != nil {
		_ = db.Create("grouplist_"+strconv.FormatInt(groupID, 10), &GlobalDataStruct{})
		_ = db.Insert("grouplist_"+strconv.FormatInt(groupID, 10), formatList)
	}
	// throw key
	err = db.Insert("pairkey_"+strconv.FormatInt(groupID, 10), FormatPairKey(PairKeyRaw, StatusID))
	if err != nil {
		_ = db.Create("pairkey_"+strconv.FormatInt(groupID, 10), &PairKeyStruct{})
		_ = db.Insert("pairkey_"+strconv.FormatInt(groupID, 10), FormatPairKey(PairKeyRaw, StatusID))
	}
	return err
}

// RemoveUserGlobalMarryList just remove it,it will persist the key and change it to Type4.
func RemoveUserGlobalMarryList(db *sql.Sqlite, pairKey string, groupID int64) bool {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	// first find the key list.
	var pairKeyNeed PairKeyStruct
	err := db.Find("pairkey_"+strconv.FormatInt(groupID, 10), &pairKeyNeed, "where pairkey is '"+pairKey+"'")
	if err != nil {
		// cannnot find, don't need to remove.
		return false
	}
	// if found.
	getThisKey := pairKeyNeed.PairKey
	err = db.Del("pairkey_"+strconv.FormatInt(groupID, 10), "where pairkey is '"+getThisKey+"'")
	if err != nil {
		// cannnot find, don't need to remove.
		return false
	}
	err = db.Del("grouplist_"+strconv.FormatInt(groupID, 10), "where pairkey is '"+getThisKey+"'")
	if err != nil {
		// cannnot find, don't need to remove.
		return false
	}
	err = db.Insert("pairkey_"+strconv.FormatInt(groupID, 10), FormatPairKey(pairKey, 4))
	return err == nil
	// store? || persist this key and check the next Time.
}

// CustomRemoveUserGlobalMarryList just remove it,it will persist the key (referKeyStatus)
func CustomRemoveUserGlobalMarryList(db *sql.Sqlite, pairKey string, groupID int64, statusID int64) bool {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	// first find the key list.
	var pairKeyNeed PairKeyStruct
	err := db.Find("pairkey_"+strconv.FormatInt(groupID, 10), &pairKeyNeed, "where pairkey is '"+pairKey+"'")
	if err != nil {
		// cannnot find, don't need to remove.
		return false
	}
	// if found.
	getThisKey := pairKeyNeed.PairKey
	_ = db.Del("pairkey_"+strconv.FormatInt(groupID, 10), "where pairkey is '"+getThisKey+"'")
	_ = db.Del("grouplist_"+strconv.FormatInt(groupID, 10), "where pairkey is '"+getThisKey+"'")
	// store? || persist this key and check the next Time.
	_ = db.Insert("pairkey_"+strconv.FormatInt(groupID, 10), FormatPairKey(pairKey, statusID))
	return err == nil
}

/*
// CheckThisKeyStatus check this key status.
func CheckThisKeyStatus(db *sql.Sqlite, pairKey string, groupID int64) int64 {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	// first find the key list.
	var pairKeyNeed PairKeyStruct
	err := db.Find("pairkey_"+strconv.FormatInt(groupID, 10), &pairKeyNeed, "where pairkey is "+pairKey)
	if err != nil {
		// cannnot find, don't need to remove.
		return -1
	}
	return pairKeyNeed.StatusID
}

*/

// AddBlackList add blacklist
func AddBlackList(db *sql.Sqlite, userID int64, targetID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var blackListNeed BlackListStruct
	err := db.Find("blacklist_"+strconv.FormatInt(userID, 10), &blackListNeed, "where blacklist is '"+strconv.FormatInt(targetID, 10)+"'")
	if err != nil {
		// add it, not sure then init this and add.
		_ = db.Create("blacklist_"+strconv.FormatInt(userID, 10), &BlackListStruct{})
		err = db.Insert("blacklist_"+strconv.FormatInt(userID, 10), FormatBlackList(targetID))
		return err
	}
	// find this so don't do it.
	return err
}

// DeleteBlackList delete blacklist.
func DeleteBlackList(db *sql.Sqlite, userID int64, targetID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var blackListNeed BlackListStruct
	err := db.Find("blacklist_"+strconv.FormatInt(userID, 10), &blackListNeed, "where blacklist is '"+strconv.FormatInt(targetID, 10)+"'")
	if err != nil {
		// not init or didn't find.
		return err
	}
	_ = db.Del("blacklist_"+strconv.FormatInt(userID, 10), "where blacklist is '"+strconv.FormatInt(targetID, 10)+"'")
	return err
}

// CheckTheBlackListIsExistedToThisPerson check the person is blocked.
func CheckTheBlackListIsExistedToThisPerson(db *sql.Sqlite, userID int64, targetID int64) bool {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var blackListNeed BlackListStruct
	err := db.Find("blacklist_"+strconv.FormatInt(userID, 10), &blackListNeed, "where blacklist is '"+strconv.FormatInt(targetID, 10)+"'")
	if err != nil {
		return true
	}
	if blackListNeed.BlackList == targetID {
		return false
	}
	return true
}

// AddDisabledList add disabledList
func AddDisabledList(db *sql.Sqlite, userID int64, groupID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var disabledListNeed DisabledListStruct
	err := db.Find("disabled_"+strconv.FormatInt(userID, 10), &disabledListNeed, "where disabledlist is '"+strconv.FormatInt(groupID, 10)+"'")
	if err != nil {
		// add it, not sure then init this and add.
		_ = db.Create("disabled_"+strconv.FormatInt(userID, 10), &DisabledListStruct{})
		_ = db.Insert("disabled_"+strconv.FormatInt(userID, 10), FormatDisabledList(groupID))
		return err
	}
	// find this so don't do it.
	return err
}

// DeleteDisabledList delete disabledlist.
func DeleteDisabledList(db *sql.Sqlite, userID int64, groupID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var disabledListNeed DisabledListStruct
	err := db.Find("disabled_"+strconv.FormatInt(userID, 10), &disabledListNeed, "where disabledlist is '"+strconv.FormatInt(groupID, 10)+"'")
	if err != nil {
		// not init or didn't find.
		return err
	}
	_ = db.Del("disabled_"+strconv.FormatInt(userID, 10), "where disabledlist is '"+strconv.FormatInt(groupID, 10)+"'")
	return err
}

// CheckDisabledListIsExistedInThisGroup Check this group.
func CheckDisabledListIsExistedInThisGroup(db *sql.Sqlite, userID int64, groupID int64) bool {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var disabledListNeed DisabledListStruct
	err := db.Find("disabled_"+strconv.FormatInt(userID, 10), &disabledListNeed, "where disabledlist is '"+strconv.FormatInt(groupID, 10)+"'")
	if err != nil {
		// not init or didn't find.
		return true
	}
	if disabledListNeed.DisabledList == groupID {
		return false
	}
	return true
}

// AddOrderToList add order.
func AddOrderToList(db *sql.Sqlite, userID int64, targetID int64, time string, groupID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var addOrderListNeed OrderListStruct
	err := db.Find("orderlist_"+strconv.FormatInt(groupID, 10), &addOrderListNeed, "where order is '"+strconv.FormatInt(userID, 10)+"'")
	if err != nil {
		// create and insert.
		_ = db.Create("orderlist_"+strconv.FormatInt(groupID, 10), &OrderListStruct{})
		_ = db.Insert("orderlist_"+strconv.FormatInt(groupID, 10), FormatOrderList(userID, targetID, time))
		return err
	}
	// find it.
	return err
}

// RemoveOrderToList remove it.
func RemoveOrderToList(db *sql.Sqlite, userID int64, groupID int64) error {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var addOrderListNeed OrderListStruct
	err := db.Find("orderlist_"+strconv.FormatInt(groupID, 10), &addOrderListNeed, "where order is '"+strconv.FormatInt(userID, 10)+"'")
	if err != nil {
		return err
	}
	err = db.Del("orderlist_"+strconv.FormatInt(groupID, 10), "where order is '"+strconv.FormatInt(userID, 10)+"'")
	return err
}

/*
// CheckThisOrderList getList
func CheckThisOrderList(db *sql.Sqlite, userID int64, groupID int64) (OrderUser int64, TargetUSer int64, time string) {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	var addOrderListNeed OrderListStruct
	err := db.Find("orderlist_"+strconv.FormatInt(groupID, 10), &addOrderListNeed, "where order is "+strconv.FormatInt(userID, 10))
	if err != nil {
		return -1, -1, ""
	}
	OrderUser = addOrderListNeed.OrderPerson
	TargetUSer = addOrderListNeed.OrderPerson
	time = addOrderListNeed.Time
	return OrderUser, TargetUSer, time
}

*/

// GetTheGroupList Get this group.
func GetTheGroupList(gid int64) (list [][2]string, num int) {
	marryLocker.Lock()
	defer marryLocker.Unlock()
	gidStr := strconv.FormatInt(gid, 10)
	getNum, _ := marryList.Count("grouplist_" + gidStr)
	if getNum == 0 {
		return nil, 0
	}
	var info GlobalDataStruct
	list = make([][2]string, 0, num)
	_ = marryList.FindFor("grouplist_"+gidStr, &info, "GROUP BY userid", func() error {
		getInfoSlice := [2]string{
			strconv.FormatInt(info.UserID, 10),
			strconv.FormatInt(info.TargetID, 10),
		}
		list = append(list, getInfoSlice)
		return nil
	})
	return list, getNum
}