getIDHandler.go

package server

import (
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"github.com/hoshinonyaruko/gensokyo/idmap"
	"github.com/hoshinonyaruko/gensokyo/mylog"
)

func GetIDHandler(c *gin.Context) {
	idOrRow := c.Query("id")
	typeVal, err := strconv.Atoi(c.Query("type"))

	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"error": "invalid type"})
		return
	}

	switch typeVal {
	case 1:
		newRow, err := idmap.StoreIDv2(idOrRow)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"row": newRow})

	case 2:
		id, err := idmap.RetrieveRowByIDv2(idOrRow)
		if err == idmap.ErrKeyNotFound {
			c.JSON(http.StatusNotFound, gin.H{"error": "ID not found"})
			return
		} else if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"id": id})

	case 3:
		// 存储
		section := c.Query("id")
		subtype := c.Query("subtype")
		value := c.Query("value")
		err := idmap.WriteConfigv2(section, subtype, value)
		if err != nil {
			mylog.Printf(err.Error())
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"status": "success"})

	case 4:
		// 获取值
		section := c.Query("id")
		subtype := c.Query("subtype")
		value, err := idmap.ReadConfigv2(section, subtype)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"value": value})

	case 5:
		oldRowValue, err := strconv.ParseInt(c.Query("oldRowValue"), 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid oldRowValue"})
			return
		}

		newRowValue, err := strconv.ParseInt(c.Query("newRowValue"), 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid newRowValue"})
			return
		}

		err = idmap.UpdateVirtualValuev2(oldRowValue, newRowValue)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"status": "success"})

	case 6:
		virtualValue, err := strconv.ParseInt(c.Query("virtualValue"), 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid virtualValue"})
			return
		}

		virtual, real, err := idmap.RetrieveRealValuev2(virtualValue)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"virtual": virtual, "real": real})
	case 7:
		realValue := c.Query("id")
		if realValue == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "invalid id"})
			return
		}

		_, virtualValue, err := idmap.RetrieveVirtualValuev2(realValue)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"virtual": virtualValue})
	case 8:
		// 调用新的 StoreIDv2Pro
		subid := c.Query("subid")
		if subid == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "subid parameter is required for type 8"})
			return
		}
		newRow, newSubRow, err := idmap.StoreIDv2Pro(idOrRow, subid)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"row": newRow, "subRow": newSubRow})

	case 9:
		// 调用新的 RetrieveRowByIDv2Pro
		subid := c.Query("subid")
		if subid == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "subid parameter is required for type 9"})
			return
		}
		id, subid, err := idmap.RetrieveRowByIDv2Pro(idOrRow, subid)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"id": id, "subid": subid})
	case 10:
		subid := c.Query("subid")
		if idOrRow == "" || subid == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "id and subid parameters are required for type 10"})
			return
		}

		firstValue, secondValue, err := idmap.RetrieveVirtualValuev2Pro(idOrRow, subid) // 确保函数名称正确
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{"firstValue": firstValue, "secondValue": secondValue})
	case 11:
		subid := c.Query("subid")
		if idOrRow == "" || subid == "" {
			c.JSON(http.StatusBadRequest, gin.H{"error": "id and subid parameters are required for type 11"})
			return
		}
		var virtualValue, virtualValueSub int64
		virtualValue, err = strconv.ParseInt(idOrRow, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values idOrRow"})
			return
		}
		virtualValueSub, err = strconv.ParseInt(subid, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values subid"})
			return
		}
		firstRealValue, secondRealValue, err := idmap.RetrieveRealValuesv2Pro(virtualValue, virtualValueSub)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{"firstRealValue": firstRealValue, "secondRealValue": secondRealValue})
	case 12:
		oldVirtualValue1Str := c.Query("oldVirtualValue1")
		newVirtualValue1Str := c.Query("newVirtualValue1")
		oldVirtualValue2Str := c.Query("oldVirtualValue2")
		newVirtualValue2Str := c.Query("newVirtualValue2")
		var oldVirtualValue1, newVirtualValue1, oldVirtualValue2, newVirtualValue2 int64
		// 将字符串转换为int64
		oldVirtualValue1, err = strconv.ParseInt(oldVirtualValue1Str, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values oldVirtualValue1"})
			return
		}
		newVirtualValue1, err = strconv.ParseInt(newVirtualValue1Str, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values newVirtualValue1"})
			return
		}
		oldVirtualValue2, err = strconv.ParseInt(oldVirtualValue2Str, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values oldVirtualValue2"})
			return
		}
		newVirtualValue2, err = strconv.ParseInt(newVirtualValue2Str, 10, 64)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid input values newVirtualValue2"})
			return
		}
		err = idmap.UpdateVirtualValuev2Pro(oldVirtualValue1, newVirtualValue1, oldVirtualValue2, newVirtualValue2)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{"message": "Virtual values updated successfully"})
	case 13:
		newRow, err := idmap.SimplifiedStoreIDv2(idOrRow)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"row": newRow})
	case 14:
		id := c.Query("id")
		keys, err := idmap.FindSubKeysByIdPro(id)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		c.JSON(http.StatusOK, gin.H{"keys": keys})
	case 15:
		// 删除
		// 从请求中获取参数
		section := c.Query("id")
		subtype := c.Query("subtype")

		// 调用DeleteConfigv2来删除配置
		err := idmap.DeleteConfigv2(section, subtype)
		if err != nil {
			// 如果有错误，记录并返回错误信息
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		// 如果删除成功，返回成功响应
		c.JSON(http.StatusOK, gin.H{"status": "success"})
	}

}