89 lines
1.3 KiB
Go
89 lines
1.3 KiB
Go
|
package jsons
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
"encoding/json"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
// The package implements basic types
|
||
|
|
// to be used in structures that can be parsed
|
||
|
|
// from the poorly structured JSON. (Like ones that make IDs strings)
|
||
|
|
|
||
|
|
func parseInt(str string, n int) (int64, error) {
|
||
|
|
if str == "" {
|
||
|
|
return 0, EmptyStringErr
|
||
|
|
}
|
||
|
|
|
||
|
|
if str[0] == '"' {
|
||
|
|
str = str[1:]
|
||
|
|
}
|
||
|
|
|
||
|
|
if str == "" {
|
||
|
|
return 0, EmptyStringErr
|
||
|
|
}
|
||
|
|
|
||
|
|
if str[len(str)-1] == '"' {
|
||
|
|
str = str[:len(str)-1]
|
||
|
|
}
|
||
|
|
|
||
|
|
v, err := strconv.ParseInt(str, 10, n)
|
||
|
|
if err != nil {
|
||
|
|
return 0, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return v, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type Int int
|
||
|
|
func(ji *Int) UnmarshalJSON(bts []byte) error {
|
||
|
|
v, err := parseInt(string(bts), 32)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
*ji = Int(v)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type Int64 int64
|
||
|
|
func(ji *Int64) UnmarshalJSON(bts []byte) error {
|
||
|
|
v, err := parseInt(string(bts), 64)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
*ji = Int64(v)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type ArrayMap[V any] map[int] V
|
||
|
|
func (jam *ArrayMap[V]) UnmarshalJSON(bts []byte) error {
|
||
|
|
*jam = make(ArrayMap[V])
|
||
|
|
am := *jam
|
||
|
|
j := map[string] json.RawMessage{}
|
||
|
|
err := json.Unmarshal(bts, &j)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
v := new(V)
|
||
|
|
for jk, jv := range j {
|
||
|
|
// Getting the key from string.
|
||
|
|
k, err := strconv.ParseInt(jk, 10, 32)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
err = json.Unmarshal([]byte(jv), v)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
am[int(k)] = *v
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|