init

pkg/hashid/hash.go

@@ -0,0 +1,70 @@
+package hashid
+
+import (
+	"errors"
+
+	"github.com/cloudreve/Cloudreve/v3/pkg/conf"
+	"github.com/speps/go-hashids"
+)
+
+// ID类型
+const (
+	ShareID  = iota // 分享
+	UserID          // 用户
+	FileID          // 文件ID
+	FolderID        // 目录ID
+	TagID           // 标签ID
+	PolicyID        // 存储策略ID
+	SourceLinkID
+)
+
+var (
+	// ErrTypeNotMatch ID类型不匹配
+	ErrTypeNotMatch = errors.New("mismatched ID type.")
+)
+
+// HashEncode 对给定数据计算HashID
+func HashEncode(v []int) (string, error) {
+	hd := hashids.NewData()
+	hd.Salt = conf.SystemConfig.HashIDSalt
+
+	h, err := hashids.NewWithData(hd)
+	if err != nil {
+		return "", err
+	}
+
+	id, err := h.Encode(v)
+	if err != nil {
+		return "", err
+	}
+	return id, nil
+}
+
+// HashDecode 对给定数据计算原始数据
+func HashDecode(raw string) ([]int, error) {
+	hd := hashids.NewData()
+	hd.Salt = conf.SystemConfig.HashIDSalt
+
+	h, err := hashids.NewWithData(hd)
+	if err != nil {
+		return []int{}, err
+	}
+
+	return h.DecodeWithError(raw)
+
+}
+
+// HashID 计算数据库内主键对应的HashID
+func HashID(id uint, t int) string {
+	v, _ := HashEncode([]int{int(id), t})
+	return v
+}
+
+// DecodeHashID 计算HashID对应的数据库ID
+func DecodeHashID(id string, t int) (uint, error) {
+	v, _ := HashDecode(id)
+	if len(v) != 2 || v[1] != t {
+		return 0, ErrTypeNotMatch
+	}
+	return uint(v[0]), nil
+}