134 lines
2.6 KiB
Go
134 lines
2.6 KiB
Go
|
package helpers
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/base64"
|
||
|
|
"encoding/binary"
|
||
|
|
"encoding/gob"
|
||
|
|
"encoding/hex"
|
||
|
|
"encoding/json"
|
||
|
|
"sort"
|
||
|
|
|
||
|
|
wnsUtils "git.vdb.to/cerc-io/laconic2d/utils"
|
||
|
|
set "github.com/deckarep/golang-set"
|
||
|
|
)
|
||
|
|
|
||
|
|
func StringToBytes(val string) []byte {
|
||
|
|
return []byte(val)
|
||
|
|
}
|
||
|
|
|
||
|
|
func BytesToString(val []byte) string {
|
||
|
|
return string(val)
|
||
|
|
}
|
||
|
|
|
||
|
|
func StrArrToBytesArr(val []string) ([]byte, error) {
|
||
|
|
buffer := &bytes.Buffer{}
|
||
|
|
|
||
|
|
err := gob.NewEncoder(buffer).Encode(val)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return buffer.Bytes(), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func BytesArrToStringArr(val []byte) ([]string, error) {
|
||
|
|
buffer := bytes.NewReader(val)
|
||
|
|
var v []string
|
||
|
|
err := gob.NewDecoder(buffer).Decode(&v)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return v, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func Int64ToBytes(num int64) []byte {
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
_ = binary.Write(buf, binary.BigEndian, num)
|
||
|
|
return buf.Bytes()
|
||
|
|
}
|
||
|
|
|
||
|
|
func MustMarshalJSON[T any](val T) (bytes []byte) {
|
||
|
|
bytes, err := json.Marshal(val)
|
||
|
|
if err != nil {
|
||
|
|
panic("JSON marshal error:" + err.Error())
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func MustUnmarshalJSON[T any](bytes []byte) T {
|
||
|
|
var val T
|
||
|
|
err := json.Unmarshal(bytes, &val)
|
||
|
|
if err != nil {
|
||
|
|
panic("JSON unmarshal error:" + err.Error())
|
||
|
|
}
|
||
|
|
return val
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetCid gets the content ID.
|
||
|
|
func GetCid(content []byte) (string, error) {
|
||
|
|
return wnsUtils.CIDFromJSONBytes(content)
|
||
|
|
}
|
||
|
|
|
||
|
|
// BytesToBase64 encodes a byte array as a base64 string.
|
||
|
|
func BytesToBase64(bytes []byte) string {
|
||
|
|
return base64.StdEncoding.EncodeToString(bytes)
|
||
|
|
}
|
||
|
|
|
||
|
|
// BytesFromBase64 decodes a byte array from a base64 string.
|
||
|
|
func BytesFromBase64(str string) []byte {
|
||
|
|
bytes, err := base64.StdEncoding.DecodeString(str)
|
||
|
|
if err != nil {
|
||
|
|
panic("Error decoding string to bytes.")
|
||
|
|
}
|
||
|
|
|
||
|
|
return bytes
|
||
|
|
}
|
||
|
|
|
||
|
|
// BytesToHex encodes a byte array as a hex string.
|
||
|
|
func BytesToHex(bytes []byte) string {
|
||
|
|
return hex.EncodeToString(bytes)
|
||
|
|
}
|
||
|
|
|
||
|
|
// BytesFromHex decodes a byte array from a hex string.
|
||
|
|
func BytesFromHex(str string) []byte {
|
||
|
|
bytes, err := hex.DecodeString(str)
|
||
|
|
if err != nil {
|
||
|
|
panic("Error decoding hex to bytes.")
|
||
|
|
}
|
||
|
|
|
||
|
|
return bytes
|
||
|
|
}
|
||
|
|
|
||
|
|
func SetToSlice(set set.Set) []string {
|
||
|
|
names := []string{}
|
||
|
|
|
||
|
|
for name := range set.Iter() {
|
||
|
|
if name, ok := name.(string); ok && name != "" {
|
||
|
|
names = append(names, name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
sort.SliceStable(names, func(i, j int) bool { return names[i] < names[j] })
|
||
|
|
|
||
|
|
return names
|
||
|
|
}
|
||
|
|
|
||
|
|
func SliceToSet(names []string) set.Set {
|
||
|
|
set := set.NewThreadUnsafeSet()
|
||
|
|
|
||
|
|
for _, name := range names {
|
||
|
|
if name != "" {
|
||
|
|
set.Add(name)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return set
|
||
|
|
}
|
||
|
|
|
||
|
|
func AppendUnique(list []string, element string) []string {
|
||
|
|
set := SliceToSet(list)
|
||
|
|
set.Add(element)
|
||
|
|
|
||
|
|
return SetToSlice(set)
|
||
|
|
}
|