Compare commits
16 Commits
Author | SHA1 | Date | |
---|---|---|---|
f8c0c0afa0
|
|||
2fbd5cf965
|
|||
75f71fe3db
|
|||
ab1a1ab6f6
|
|||
19ee5019ef
|
|||
42b68507f2
|
|||
9d0047a11e
|
|||
06d81f1682
|
|||
7b8ab03779
|
|||
07cbcf5a0a
|
|||
da41ec3e84
|
|||
592fae25af
|
|||
7968460fa2
|
|||
b808c5727c
|
|||
796f7956b8
|
|||
1e6b92d1d9 |
@@ -1,9 +0,0 @@
|
|||||||
FROM golang:latest
|
|
||||||
|
|
||||||
RUN apt install -y make curl python3 && go install gotest.tools/gotestsum@latest
|
|
||||||
|
|
||||||
COPY . /source
|
|
||||||
|
|
||||||
WORKDIR /source
|
|
||||||
|
|
||||||
CMD ["make", "test"]
|
|
@@ -10,21 +10,27 @@ on: [push]
|
|||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
run_tests:
|
run_tests:
|
||||||
name: Run goext test-suite
|
name: Run goext test-suite
|
||||||
runs-on: bfb-cicd-latest
|
runs-on: bfb-cicd-latest
|
||||||
steps:
|
steps:
|
||||||
|
|
||||||
- name: Check out code
|
- name: Check out code
|
||||||
uses: actions/checkout@v3
|
uses: actions/checkout@v3
|
||||||
|
|
||||||
- name: Build test docker
|
- name: Setup go
|
||||||
id: build_docker
|
uses: actions/setup-go@v4
|
||||||
run: echo "DOCKER_IMG_ID=$(docker build -q . -f .gitea/workflows/Dockerfile_tests || echo __err_build__)" >> $GITHUB_OUTPUT
|
with:
|
||||||
|
go-version-file: '${{ gitea.workspace }}/go.mod'
|
||||||
|
|
||||||
|
- name: Setup packages
|
||||||
|
uses: awalsh128/cache-apt-pkgs-action@latest
|
||||||
|
with:
|
||||||
|
packages: curl python3
|
||||||
|
version: 1.0
|
||||||
|
|
||||||
|
- name: go version
|
||||||
|
run: go version
|
||||||
|
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: docker run --rm "${{ steps.build_docker.outputs.DOCKER_IMG_ID }}"
|
run: cd "${{ gitea.workspace }}" && make test
|
||||||
|
|
||||||
- name: Cleanup
|
|
||||||
if: always()
|
|
||||||
run: docker image rm "${{ steps.build_docker.outputs.DOCKER_IMG_ID }}"
|
|
||||||
|
|
||||||
|
59
README.md
59
README.md
@@ -10,32 +10,33 @@ Potentially needs `export GOPRIVATE="gogs.mikescher.com"`
|
|||||||
|
|
||||||
### Packages:
|
### Packages:
|
||||||
|
|
||||||
| Name | Maintainer | Description |
|
| Name | Maintainer | Description |
|
||||||
|--------------|------------|---------------------------------------------------------------------------------------------------------------|
|
|-------------|------------|---------------------------------------------------------------------------------------------------------------|
|
||||||
| langext | Mike | General uttility/helper functions, (everything thats missing from go standard library) |
|
| langext | Mike | General uttility/helper functions, (everything thats missing from go standard library) |
|
||||||
| mathext | Mike | Utility/Helper functions for math |
|
| mathext | Mike | Utility/Helper functions for math |
|
||||||
| cryptext | Mike | Utility/Helper functions for encryption |
|
| cryptext | Mike | Utility/Helper functions for encryption |
|
||||||
| syncext | Mike | Utility/Helper funtions for multi-threading / mutex / channels |
|
| syncext | Mike | Utility/Helper funtions for multi-threading / mutex / channels |
|
||||||
| dataext | Mike | Various useful data structures |
|
| dataext | Mike | Various useful data structures |
|
||||||
| zipext | Mike | Utility for zip/gzip/tar etc |
|
| zipext | Mike | Utility for zip/gzip/tar etc |
|
||||||
| reflectext | Mike | Utility for golagn reflection |
|
| reflectext | Mike | Utility for golang reflection |
|
||||||
| | | |
|
| fsext | Mike | Utility for filesytem access |
|
||||||
| mongoext | Mike | Utility/Helper functions for mongodb |
|
| | | |
|
||||||
| cursortoken | Mike | MongoDB cursortoken implementation |
|
| mongoext | Mike | Utility/Helper functions for mongodb |
|
||||||
| | | |
|
| cursortoken | Mike | MongoDB cursortoken implementation |
|
||||||
| totpext | Mike | Implementation of TOTP (2-Factor-Auth) |
|
| | | |
|
||||||
| termext | Mike | Utilities for terminals (mostly color output) |
|
| totpext | Mike | Implementation of TOTP (2-Factor-Auth) |
|
||||||
| confext | Mike | Parses environment configuration into structs |
|
| termext | Mike | Utilities for terminals (mostly color output) |
|
||||||
| cmdext | Mike | Runner for external commands/processes |
|
| confext | Mike | Parses environment configuration into structs |
|
||||||
| | | |
|
| cmdext | Mike | Runner for external commands/processes |
|
||||||
| sq | Mike | Utility functions for sql based databases |
|
| | | |
|
||||||
| tst | Mike | Utility functions for unit tests |
|
| sq | Mike | Utility functions for sql based databases |
|
||||||
| | | |
|
| tst | Mike | Utility functions for unit tests |
|
||||||
| rfctime | Mike | Classes for time seriallization, with different marshallign method for mongo and json |
|
| | | |
|
||||||
| gojson | Mike | Same interface for marshalling/unmarshalling as go/json, except with proper serialization of null arrays/maps |
|
| rfctime | Mike | Classes for time seriallization, with different marshallign method for mongo and json |
|
||||||
| | | |
|
| gojson | Mike | Same interface for marshalling/unmarshalling as go/json, except with proper serialization of null arrays/maps |
|
||||||
| bfcodegen | Mike | Various codegen tools (run via go generate) |
|
| | | |
|
||||||
| | | |
|
| bfcodegen | Mike | Various codegen tools (run via go generate) |
|
||||||
| rext | Mike | Regex Wrapper, wraps regexp with a better interface |
|
| | | |
|
||||||
| wmo | Mike | Mongo Wrapper, wraps mongodb with a better interface |
|
| rext | Mike | Regex Wrapper, wraps regexp with a better interface |
|
||||||
| | | |
|
| wmo | Mike | Mongo Wrapper, wraps mongodb with a better interface |
|
||||||
|
| | | |
|
Binary file not shown.
192
bfcodegen/csid-generate.go
Normal file
192
bfcodegen/csid-generate.go
Normal file
@@ -0,0 +1,192 @@
|
|||||||
|
package bfcodegen
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
_ "embed"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/cmdext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/cryptext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/rext"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"path/filepath"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type CSIDDef struct {
|
||||||
|
File string
|
||||||
|
FileRelative string
|
||||||
|
Name string
|
||||||
|
Prefix string
|
||||||
|
}
|
||||||
|
|
||||||
|
var rexCSIDPackage = rext.W(regexp.MustCompile(`^package\s+(?P<name>[A-Za-z0-9_]+)\s*$`))
|
||||||
|
|
||||||
|
var rexCSIDDef = rext.W(regexp.MustCompile(`^\s*type\s+(?P<name>[A-Za-z0-9_]+)\s+string\s*//\s*(@csid:type)\s+\[(?P<prefix>[A-Z0-9]{3})].*$`))
|
||||||
|
|
||||||
|
var rexCSIDChecksumConst = rext.W(regexp.MustCompile(`const ChecksumCharsetIDGenerator = "(?P<cs>[A-Za-z0-9_]*)"`))
|
||||||
|
|
||||||
|
//go:embed csid-generate.template
|
||||||
|
var templateCSIDGenerateText string
|
||||||
|
|
||||||
|
func GenerateCharsetIDSpecs(sourceDir string, destFile string) error {
|
||||||
|
|
||||||
|
files, err := os.ReadDir(sourceDir)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
oldChecksum := "N/A"
|
||||||
|
if _, err := os.Stat(destFile); !os.IsNotExist(err) {
|
||||||
|
content, err := os.ReadFile(destFile)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if m, ok := rexCSIDChecksumConst.MatchFirst(string(content)); ok {
|
||||||
|
oldChecksum = m.GroupByName("cs").Value()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
files = langext.ArrFilter(files, func(v os.DirEntry) bool { return v.Name() != path.Base(destFile) })
|
||||||
|
files = langext.ArrFilter(files, func(v os.DirEntry) bool { return strings.HasSuffix(v.Name(), ".go") })
|
||||||
|
files = langext.ArrFilter(files, func(v os.DirEntry) bool { return !strings.HasSuffix(v.Name(), "_gen.go") })
|
||||||
|
langext.SortBy(files, func(v os.DirEntry) string { return v.Name() })
|
||||||
|
|
||||||
|
newChecksumStr := goext.GoextVersion
|
||||||
|
for _, f := range files {
|
||||||
|
content, err := os.ReadFile(path.Join(sourceDir, f.Name()))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
newChecksumStr += "\n" + f.Name() + "\t" + cryptext.BytesSha256(content)
|
||||||
|
}
|
||||||
|
|
||||||
|
newChecksum := cryptext.BytesSha256([]byte(newChecksumStr))
|
||||||
|
|
||||||
|
if newChecksum != oldChecksum {
|
||||||
|
fmt.Printf("[IDGenerate] Checksum has changed ( %s -> %s ), will generate new file\n\n", oldChecksum, newChecksum)
|
||||||
|
} else {
|
||||||
|
fmt.Printf("[IDGenerate] Checksum unchanged ( %s ), nothing to do\n", oldChecksum)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
allIDs := make([]CSIDDef, 0)
|
||||||
|
|
||||||
|
pkgname := ""
|
||||||
|
|
||||||
|
for _, f := range files {
|
||||||
|
fmt.Printf("========= %s =========\n\n", f.Name())
|
||||||
|
fileIDs, pn, err := processCSIDFile(sourceDir, path.Join(sourceDir, f.Name()))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("\n")
|
||||||
|
|
||||||
|
allIDs = append(allIDs, fileIDs...)
|
||||||
|
|
||||||
|
if pn != "" {
|
||||||
|
pkgname = pn
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if pkgname == "" {
|
||||||
|
return errors.New("no package name found in any file")
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.WriteFile(destFile, []byte(fmtCSIDOutput(newChecksum, allIDs, pkgname)), 0o755)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
res, err := cmdext.RunCommand("go", []string{"fmt", destFile}, langext.Ptr(2*time.Second))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if res.CommandTimedOut {
|
||||||
|
fmt.Println(res.StdCombined)
|
||||||
|
return errors.New("go fmt timed out")
|
||||||
|
}
|
||||||
|
if res.ExitCode != 0 {
|
||||||
|
fmt.Println(res.StdCombined)
|
||||||
|
return errors.New("go fmt did not succeed")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func processCSIDFile(basedir string, fn string) ([]CSIDDef, string, error) {
|
||||||
|
file, err := os.Open(fn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() { _ = file.Close() }()
|
||||||
|
|
||||||
|
bin, err := io.ReadAll(file)
|
||||||
|
if err != nil {
|
||||||
|
return nil, "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
lines := strings.Split(string(bin), "\n")
|
||||||
|
|
||||||
|
ids := make([]CSIDDef, 0)
|
||||||
|
|
||||||
|
pkgname := ""
|
||||||
|
|
||||||
|
for i, line := range lines {
|
||||||
|
if i == 0 && strings.HasPrefix(line, "// Code generated by") {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
if match, ok := rexCSIDPackage.MatchFirst(line); i == 0 && ok {
|
||||||
|
pkgname = match.GroupByName("name").Value()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if match, ok := rexCSIDDef.MatchFirst(line); ok {
|
||||||
|
|
||||||
|
rfp, err := filepath.Rel(basedir, fn)
|
||||||
|
if err != nil {
|
||||||
|
return nil, "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
def := CSIDDef{
|
||||||
|
File: fn,
|
||||||
|
FileRelative: rfp,
|
||||||
|
Name: match.GroupByName("name").Value(),
|
||||||
|
Prefix: match.GroupByName("prefix").Value(),
|
||||||
|
}
|
||||||
|
fmt.Printf("Found ID definition { '%s' }\n", def.Name)
|
||||||
|
ids = append(ids, def)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ids, pkgname, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func fmtCSIDOutput(cs string, ids []CSIDDef, pkgname string) string {
|
||||||
|
templ := template.Must(template.New("csid-generate").Parse(templateCSIDGenerateText))
|
||||||
|
|
||||||
|
buffer := bytes.Buffer{}
|
||||||
|
|
||||||
|
err := templ.Execute(&buffer, langext.H{
|
||||||
|
"PkgName": pkgname,
|
||||||
|
"Checksum": cs,
|
||||||
|
"GoextVersion": goext.GoextVersion,
|
||||||
|
"IDs": ids,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return buffer.String()
|
||||||
|
}
|
189
bfcodegen/csid-generate.template
Normal file
189
bfcodegen/csid-generate.template
Normal file
@@ -0,0 +1,189 @@
|
|||||||
|
// Code generated by csid-generate.go DO NOT EDIT.
|
||||||
|
|
||||||
|
package {{.PkgName}}
|
||||||
|
|
||||||
|
import "crypto/rand"
|
||||||
|
import "fmt"
|
||||||
|
import "github.com/go-playground/validator/v10"
|
||||||
|
import "github.com/rs/zerolog/log"
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/rext"
|
||||||
|
import "math/big"
|
||||||
|
import "reflect"
|
||||||
|
import "regexp"
|
||||||
|
import "strings"
|
||||||
|
|
||||||
|
const ChecksumCharsetIDGenerator = "{{.Checksum}}" // GoExtVersion: {{.GoextVersion}}
|
||||||
|
|
||||||
|
const idlen = 24
|
||||||
|
|
||||||
|
const checklen = 1
|
||||||
|
|
||||||
|
const idCharset = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
||||||
|
const idCharsetLen = len(idCharset)
|
||||||
|
|
||||||
|
var charSetReverseMap = generateCharsetMap()
|
||||||
|
|
||||||
|
const ({{range .IDs}}
|
||||||
|
prefix{{.Name}} = "{{.Prefix}}" {{end}}
|
||||||
|
)
|
||||||
|
|
||||||
|
var ({{range .IDs}}
|
||||||
|
regex{{.Name}} = generateRegex(prefix{{.Name}}) {{end}}
|
||||||
|
)
|
||||||
|
|
||||||
|
func generateRegex(prefix string) rext.Regex {
|
||||||
|
return rext.W(regexp.MustCompile(fmt.Sprintf("^%s[%s]{%d}[%s]{%d}$", prefix, idCharset, idlen-len(prefix)-checklen, idCharset, checklen)))
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateCharsetMap() []int {
|
||||||
|
result := make([]int, 128)
|
||||||
|
for i := 0; i < len(result); i++ {
|
||||||
|
result[i] = -1
|
||||||
|
}
|
||||||
|
for idx, chr := range idCharset {
|
||||||
|
result[int(chr)] = idx
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateID(prefix string) string {
|
||||||
|
k := ""
|
||||||
|
csMax := big.NewInt(int64(idCharsetLen))
|
||||||
|
checksum := 0
|
||||||
|
for i := 0; i < idlen-len(prefix)-checklen; i++ {
|
||||||
|
v, err := rand.Int(rand.Reader, csMax)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
v64 := v.Int64()
|
||||||
|
k += string(idCharset[v64])
|
||||||
|
checksum = (checksum + int(v64)) % (idCharsetLen)
|
||||||
|
}
|
||||||
|
checkstr := string(idCharset[checksum%idCharsetLen])
|
||||||
|
return prefix + k + checkstr
|
||||||
|
}
|
||||||
|
|
||||||
|
func generateIDFromSeed(prefix string, seed string) string {
|
||||||
|
h := sha256.New()
|
||||||
|
|
||||||
|
iddata := ""
|
||||||
|
for len(iddata) < idlen-len(prefix)-checklen {
|
||||||
|
h.Write([]byte(seed))
|
||||||
|
bs := h.Sum(nil)
|
||||||
|
iddata += langext.NewAnyBaseConverter(idCharset).Encode(bs)
|
||||||
|
}
|
||||||
|
|
||||||
|
checksum := 0
|
||||||
|
for i := 0; i < idlen-len(prefix)-checklen; i++ {
|
||||||
|
ichr := int(iddata[i])
|
||||||
|
checksum = (checksum + charSetReverseMap[ichr]) % (idCharsetLen)
|
||||||
|
}
|
||||||
|
|
||||||
|
checkstr := string(idCharset[checksum%idCharsetLen])
|
||||||
|
|
||||||
|
return prefix + iddata[:(idlen-len(prefix)-checklen)] + checkstr
|
||||||
|
}
|
||||||
|
|
||||||
|
func validateID(prefix string, value string) error {
|
||||||
|
if len(value) != idlen {
|
||||||
|
return exerr.New(exerr.TypeInvalidCSID, "id has the wrong length").Str("value", value).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
if !strings.HasPrefix(value, prefix) {
|
||||||
|
return exerr.New(exerr.TypeInvalidCSID, "id is missing the correct prefix").Str("value", value).Str("prefix", prefix).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
checksum := 0
|
||||||
|
for i := len(prefix); i < len(value)-checklen; i++ {
|
||||||
|
ichr := int(value[i])
|
||||||
|
if ichr < 0 || ichr >= len(charSetReverseMap) || charSetReverseMap[ichr] == -1 {
|
||||||
|
return exerr.New(exerr.TypeInvalidCSID, "id contains invalid characters").Str("value", value).Build()
|
||||||
|
}
|
||||||
|
checksum = (checksum + charSetReverseMap[ichr]) % (idCharsetLen)
|
||||||
|
}
|
||||||
|
|
||||||
|
checkstr := string(idCharset[checksum%idCharsetLen])
|
||||||
|
|
||||||
|
if !strings.HasSuffix(value, checkstr) {
|
||||||
|
return exerr.New(exerr.TypeInvalidCSID, "id checkstring is invalid").Str("value", value).Str("checkstr", checkstr).Build()
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func getRawData(prefix string, value string) string {
|
||||||
|
if len(value) != idlen {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return value[len(prefix) : idlen-checklen]
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCheckString(prefix string, value string) string {
|
||||||
|
if len(value) != idlen {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return value[idlen-checklen:]
|
||||||
|
}
|
||||||
|
|
||||||
|
func ValidateEntityID(vfl validator.FieldLevel) bool {
|
||||||
|
if !vfl.Field().CanInterface() {
|
||||||
|
log.Error().Msgf("Failed to validate EntityID (cannot interface ?!?)")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
ifvalue := vfl.Field().Interface()
|
||||||
|
|
||||||
|
if value1, ok := ifvalue.(EntityID); ok {
|
||||||
|
|
||||||
|
if vfl.Field().Type().Kind() == reflect.Pointer && langext.IsNil(value1) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := value1.Valid(); err != nil {
|
||||||
|
log.Debug().Msgf("Failed to validate EntityID '%s' (%s)", value1.String(), err.Error())
|
||||||
|
return false
|
||||||
|
} else {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
log.Error().Msgf("Failed to validate EntityID (wrong type: %T)", ifvalue)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{{range .IDs}}
|
||||||
|
|
||||||
|
// ================================ {{.Name}} ({{.FileRelative}}) ================================
|
||||||
|
|
||||||
|
func New{{.Name}}() {{.Name}} {
|
||||||
|
return {{.Name}}(generateID(prefix{{.Name}}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (id {{.Name}}) Valid() error {
|
||||||
|
return validateID(prefix{{.Name}}, string(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i {{.Name}}) String() string {
|
||||||
|
return string(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i {{.Name}}) Prefix() string {
|
||||||
|
return prefix{{.Name}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (id {{.Name}}) Raw() string {
|
||||||
|
return getRawData(prefix{{.Name}}, string(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (id {{.Name}}) CheckString() string {
|
||||||
|
return getCheckString(prefix{{.Name}}, string(id))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (id {{.Name}}) Regex() rext.Regex {
|
||||||
|
return regex{{.Name}}
|
||||||
|
}
|
||||||
|
|
||||||
|
{{end}}
|
52
bfcodegen/csid-generate_test.go
Normal file
52
bfcodegen/csid-generate_test.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package bfcodegen
|
||||||
|
|
||||||
|
import (
|
||||||
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/cmdext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/tst"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed _test_example.tgz
|
||||||
|
var CSIDExampleModels []byte
|
||||||
|
|
||||||
|
func TestGenerateCSIDSpecs(t *testing.T) {
|
||||||
|
|
||||||
|
tmpFile := filepath.Join(t.TempDir(), langext.MustHexUUID()+".tgz")
|
||||||
|
|
||||||
|
tmpDir := filepath.Join(t.TempDir(), langext.MustHexUUID())
|
||||||
|
|
||||||
|
err := os.WriteFile(tmpFile, CSIDExampleModels, 0o777)
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
t.Cleanup(func() { _ = os.Remove(tmpFile) })
|
||||||
|
|
||||||
|
err = os.Mkdir(tmpDir, 0o777)
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
t.Cleanup(func() { _ = os.RemoveAll(tmpFile) })
|
||||||
|
|
||||||
|
_, err = cmdext.Runner("tar").Arg("-xvzf").Arg(tmpFile).Arg("-C").Arg(tmpDir).FailOnExitCode().FailOnTimeout().Timeout(time.Minute).Run()
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
err = GenerateCharsetIDSpecs(tmpDir, tmpDir+"/csid_gen.go")
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
err = GenerateCharsetIDSpecs(tmpDir, tmpDir+"/csid_gen.go")
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println(string(tst.Must(os.ReadFile(tmpDir + "/csid_gen.go"))(t)))
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
}
|
@@ -1,6 +1,8 @@
|
|||||||
package bfcodegen
|
package bfcodegen
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
_ "embed"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext"
|
"gogs.mikescher.com/BlackForestBytes/goext"
|
||||||
@@ -14,6 +16,7 @@ import (
|
|||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -35,10 +38,13 @@ var rexEnumPackage = rext.W(regexp.MustCompile(`^package\s+(?P<name>[A-Za-z0-9_]
|
|||||||
|
|
||||||
var rexEnumDef = rext.W(regexp.MustCompile(`^\s*type\s+(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*//\s*(@enum:type).*$`))
|
var rexEnumDef = rext.W(regexp.MustCompile(`^\s*type\s+(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*//\s*(@enum:type).*$`))
|
||||||
|
|
||||||
var rexEnumValueDef = rext.W(regexp.MustCompile(`^\s*(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*=\s*(?P<value>("[A-Za-z0-9_:]+"|[0-9]+))\s*(//(?P<descr>.*))?.*$`))
|
var rexEnumValueDef = rext.W(regexp.MustCompile(`^\s*(?P<name>[A-Za-z0-9_]+)\s+(?P<type>[A-Za-z0-9_]+)\s*=\s*(?P<value>("[A-Za-z0-9_:\s]+"|[0-9]+))\s*(//(?P<descr>.*))?.*$`))
|
||||||
|
|
||||||
var rexEnumChecksumConst = rext.W(regexp.MustCompile(`const ChecksumEnumGenerator = "(?P<cs>[A-Za-z0-9_]*)"`))
|
var rexEnumChecksumConst = rext.W(regexp.MustCompile(`const ChecksumEnumGenerator = "(?P<cs>[A-Za-z0-9_]*)"`))
|
||||||
|
|
||||||
|
//go:embed enum-generate.template
|
||||||
|
var templateEnumGenerateText string
|
||||||
|
|
||||||
func GenerateEnumSpecs(sourceDir string, destFile string) error {
|
func GenerateEnumSpecs(sourceDir string, destFile string) error {
|
||||||
|
|
||||||
files, err := os.ReadDir(sourceDir)
|
files, err := os.ReadDir(sourceDir)
|
||||||
@@ -204,133 +210,32 @@ func processEnumFile(basedir string, fn string) ([]EnumDef, string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func fmtEnumOutput(cs string, enums []EnumDef, pkgname string) string {
|
func fmtEnumOutput(cs string, enums []EnumDef, pkgname string) string {
|
||||||
str := "// Code generated by enum-generate.go DO NOT EDIT.\n"
|
|
||||||
str += "\n"
|
|
||||||
str += "package " + pkgname + "\n"
|
|
||||||
str += "\n"
|
|
||||||
|
|
||||||
str += "import \"gogs.mikescher.com/BlackForestBytes/goext/langext\"" + "\n"
|
templ := template.New("enum-generate")
|
||||||
str += "import \"gogs.mikescher.com/BlackForestBytes/goext/enums\"" + "\n"
|
|
||||||
str += "\n"
|
|
||||||
|
|
||||||
str += "const ChecksumEnumGenerator = \"" + cs + "\" // GoExtVersion: " + goext.GoextVersion + "\n"
|
templ = templ.Funcs(template.FuncMap{
|
||||||
str += "\n"
|
"boolToStr": func(b bool) string { return langext.Conditional(b, "true", "false") },
|
||||||
|
"deref": func(v *string) string { return *v },
|
||||||
|
"trimSpace": func(str string) string { return strings.TrimSpace(str) },
|
||||||
|
"hasStr": func(v EnumDef) bool { return v.Type == "string" },
|
||||||
|
"hasDescr": func(v EnumDef) bool {
|
||||||
|
return langext.ArrAll(v.Values, func(val EnumDefVal) bool { return val.Description != nil })
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
for _, enumdef := range enums {
|
templ = template.Must(templ.Parse(templateEnumGenerateText))
|
||||||
|
|
||||||
hasDescr := langext.ArrAll(enumdef.Values, func(val EnumDefVal) bool { return val.Description != nil })
|
buffer := bytes.Buffer{}
|
||||||
hasStr := enumdef.Type == "string"
|
|
||||||
|
|
||||||
str += "// ================================ " + enumdef.EnumTypeName + " ================================" + "\n"
|
|
||||||
str += "//" + "\n"
|
|
||||||
str += "// File: " + enumdef.FileRelative + "\n"
|
|
||||||
str += "// StringEnum: " + langext.Conditional(hasStr, "true", "false") + "\n"
|
|
||||||
str += "// DescrEnum: " + langext.Conditional(hasDescr, "true", "false") + "\n"
|
|
||||||
str += "//" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "var __" + enumdef.EnumTypeName + "Values = []" + enumdef.EnumTypeName + "{" + "\n"
|
|
||||||
for _, v := range enumdef.Values {
|
|
||||||
str += " " + v.VarName + "," + "\n"
|
|
||||||
}
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
if hasDescr {
|
|
||||||
str += "var __" + enumdef.EnumTypeName + "Descriptions = map[" + enumdef.EnumTypeName + "]string{" + "\n"
|
|
||||||
for _, v := range enumdef.Values {
|
|
||||||
str += " " + v.VarName + ": \"" + strings.TrimSpace(*v.Description) + "\"," + "\n"
|
|
||||||
}
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
str += "var __" + enumdef.EnumTypeName + "Varnames = map[" + enumdef.EnumTypeName + "]string{" + "\n"
|
|
||||||
for _, v := range enumdef.Values {
|
|
||||||
str += " " + v.VarName + ": \"" + v.VarName + "\"," + "\n"
|
|
||||||
}
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") Valid() bool {" + "\n"
|
|
||||||
str += " return langext.InArray(e, __" + enumdef.EnumTypeName + "Values)" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") Values() []" + enumdef.EnumTypeName + " {" + "\n"
|
|
||||||
str += " return __" + enumdef.EnumTypeName + "Values" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") ValuesAny() []any {" + "\n"
|
|
||||||
str += " return langext.ArrCastToAny(__" + enumdef.EnumTypeName + "Values)" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") ValuesMeta() []enums.EnumMetaValue {" + "\n"
|
|
||||||
str += " return " + enumdef.EnumTypeName + "ValuesMeta()"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
if hasStr {
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") String() string {" + "\n"
|
|
||||||
str += " return string(e)" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
if hasDescr {
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") Description() string {" + "\n"
|
|
||||||
str += " if d, ok := __" + enumdef.EnumTypeName + "Descriptions[e]; ok {" + "\n"
|
|
||||||
str += " return d" + "\n"
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += " return \"\"" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") VarName() string {" + "\n"
|
|
||||||
str += " if d, ok := __" + enumdef.EnumTypeName + "Varnames[e]; ok {" + "\n"
|
|
||||||
str += " return d" + "\n"
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += " return \"\"" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (e " + enumdef.EnumTypeName + ") Meta() enums.EnumMetaValue {" + "\n"
|
|
||||||
if hasDescr {
|
|
||||||
str += " return enums.EnumMetaValue{VarName: e.VarName(), Value: e, Description: langext.Ptr(e.Description())}"
|
|
||||||
} else {
|
|
||||||
str += " return enums.EnumMetaValue{VarName: e.VarName(), Value: e, Description: nil}"
|
|
||||||
}
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func Parse" + enumdef.EnumTypeName + "(vv string) (" + enumdef.EnumTypeName + ", bool) {" + "\n"
|
|
||||||
str += " for _, ev := range __" + enumdef.EnumTypeName + "Values {" + "\n"
|
|
||||||
str += " if string(ev) == vv {" + "\n"
|
|
||||||
str += " return ev, true" + "\n"
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += " return \"\", false" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func " + enumdef.EnumTypeName + "Values() []" + enumdef.EnumTypeName + " {" + "\n"
|
|
||||||
str += " return __" + enumdef.EnumTypeName + "Values" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func " + enumdef.EnumTypeName + "ValuesMeta() []enums.EnumMetaValue {" + "\n"
|
|
||||||
str += " return []enums.EnumMetaValue{" + "\n"
|
|
||||||
for _, v := range enumdef.Values {
|
|
||||||
str += " " + v.VarName + ".Meta(),\n"
|
|
||||||
}
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
|
err := templ.Execute(&buffer, langext.H{
|
||||||
|
"PkgName": pkgname,
|
||||||
|
"Checksum": cs,
|
||||||
|
"GoextVersion": goext.GoextVersion,
|
||||||
|
"Enums": enums,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return str
|
return buffer.String()
|
||||||
}
|
}
|
||||||
|
97
bfcodegen/enum-generate.template
Normal file
97
bfcodegen/enum-generate.template
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
// Code generated by enum-generate.go DO NOT EDIT.
|
||||||
|
|
||||||
|
package {{.PkgName}}
|
||||||
|
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/enums"
|
||||||
|
|
||||||
|
const ChecksumEnumGenerator = "{{.Checksum}}" // GoExtVersion: {{.GoextVersion}}
|
||||||
|
|
||||||
|
{{range .Enums}}
|
||||||
|
|
||||||
|
{{ $hasStr := ( . | hasStr ) }}
|
||||||
|
{{ $hasDescr := ( . | hasDescr ) }}
|
||||||
|
|
||||||
|
// ================================ {{.EnumTypeName}} ================================
|
||||||
|
//
|
||||||
|
// File: {{.FileRelative}}
|
||||||
|
// StringEnum: {{$hasStr | boolToStr}}
|
||||||
|
// DescrEnum: {{$hasDescr | boolToStr}}
|
||||||
|
//
|
||||||
|
|
||||||
|
var __{{.EnumTypeName}}Values = []{{.EnumTypeName}}{ {{range .Values}}
|
||||||
|
{{.VarName}}, {{end}}
|
||||||
|
}
|
||||||
|
|
||||||
|
{{if $hasDescr}}
|
||||||
|
var __{{.EnumTypeName}}Descriptions = map[{{.EnumTypeName}}]string{ {{range .Values}}
|
||||||
|
{{.VarName}}: "{{.Description | deref | trimSpace}}", {{end}}
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
var __{{.EnumTypeName}}Varnames = map[{{.EnumTypeName}}]string{ {{range .Values}}
|
||||||
|
{{.VarName}}: "{{.VarName}}", {{end}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) Valid() bool {
|
||||||
|
return langext.InArray(e, __{{.EnumTypeName}}Values)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) Values() []{{.EnumTypeName}} {
|
||||||
|
return __{{.EnumTypeName}}Values
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) ValuesAny() []any {
|
||||||
|
return langext.ArrCastToAny(__{{.EnumTypeName}}Values)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) ValuesMeta() []enums.EnumMetaValue {
|
||||||
|
return {{.EnumTypeName}}ValuesMeta()
|
||||||
|
}
|
||||||
|
|
||||||
|
{{if $hasStr}}
|
||||||
|
func (e {{.EnumTypeName}}) String() string {
|
||||||
|
return string(e)
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
{{if $hasDescr}}
|
||||||
|
func (e {{.EnumTypeName}}) Description() string {
|
||||||
|
if d, ok := __{{.EnumTypeName}}Descriptions[e]; ok {
|
||||||
|
return d
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) VarName() string {
|
||||||
|
if d, ok := __{{.EnumTypeName}}Varnames[e]; ok {
|
||||||
|
return d
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e {{.EnumTypeName}}) Meta() enums.EnumMetaValue {
|
||||||
|
{{if $hasDescr}} return enums.EnumMetaValue{VarName: e.VarName(), Value: e, Description: langext.Ptr(e.Description())} {{else}} return enums.EnumMetaValue{VarName: e.VarName(), Value: e, Description: nil} {{end}}
|
||||||
|
}
|
||||||
|
|
||||||
|
func Parse{{.EnumTypeName}}(vv string) ({{.EnumTypeName}}, bool) {
|
||||||
|
for _, ev := range __{{.EnumTypeName}}Values {
|
||||||
|
if string(ev) == vv {
|
||||||
|
return ev, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", false
|
||||||
|
}
|
||||||
|
|
||||||
|
func {{.EnumTypeName}}Values() []{{.EnumTypeName}} {
|
||||||
|
return __{{.EnumTypeName}}Values
|
||||||
|
}
|
||||||
|
|
||||||
|
func {{.EnumTypeName}}ValuesMeta() []enums.EnumMetaValue {
|
||||||
|
return []enums.EnumMetaValue{ {{range .Values}}
|
||||||
|
{{.VarName}}.Meta(), {{end}}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
{{end}}
|
@@ -2,6 +2,7 @@ package bfcodegen
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
_ "embed"
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/cmdext"
|
"gogs.mikescher.com/BlackForestBytes/goext/cmdext"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/tst"
|
"gogs.mikescher.com/BlackForestBytes/goext/tst"
|
||||||
@@ -12,7 +13,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
//go:embed _test_example.tgz
|
//go:embed _test_example.tgz
|
||||||
var ExampleModels []byte
|
var EnumExampleModels []byte
|
||||||
|
|
||||||
func TestGenerateEnumSpecs(t *testing.T) {
|
func TestGenerateEnumSpecs(t *testing.T) {
|
||||||
|
|
||||||
@@ -20,7 +21,7 @@ func TestGenerateEnumSpecs(t *testing.T) {
|
|||||||
|
|
||||||
tmpDir := filepath.Join(t.TempDir(), langext.MustHexUUID())
|
tmpDir := filepath.Join(t.TempDir(), langext.MustHexUUID())
|
||||||
|
|
||||||
err := os.WriteFile(tmpFile, ExampleModels, 0o777)
|
err := os.WriteFile(tmpFile, EnumExampleModels, 0o777)
|
||||||
tst.AssertNoErr(t, err)
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
t.Cleanup(func() { _ = os.Remove(tmpFile) })
|
t.Cleanup(func() { _ = os.Remove(tmpFile) })
|
||||||
@@ -39,4 +40,13 @@ func TestGenerateEnumSpecs(t *testing.T) {
|
|||||||
err = GenerateEnumSpecs(tmpDir, tmpDir+"/enums_gen.go")
|
err = GenerateEnumSpecs(tmpDir, tmpDir+"/enums_gen.go")
|
||||||
tst.AssertNoErr(t, err)
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println(string(tst.Must(os.ReadFile(tmpDir + "/enums_gen.go"))(t)))
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
}
|
}
|
||||||
|
@@ -1,6 +1,8 @@
|
|||||||
package bfcodegen
|
package bfcodegen
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
_ "embed"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext"
|
"gogs.mikescher.com/BlackForestBytes/goext"
|
||||||
@@ -14,6 +16,7 @@ import (
|
|||||||
"path/filepath"
|
"path/filepath"
|
||||||
"regexp"
|
"regexp"
|
||||||
"strings"
|
"strings"
|
||||||
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -29,6 +32,9 @@ var rexIDDef = rext.W(regexp.MustCompile(`^\s*type\s+(?P<name>[A-Za-z0-9_]+)\s+s
|
|||||||
|
|
||||||
var rexIDChecksumConst = rext.W(regexp.MustCompile(`const ChecksumIDGenerator = "(?P<cs>[A-Za-z0-9_]*)"`))
|
var rexIDChecksumConst = rext.W(regexp.MustCompile(`const ChecksumIDGenerator = "(?P<cs>[A-Za-z0-9_]*)"`))
|
||||||
|
|
||||||
|
//go:embed id-generate.template
|
||||||
|
var templateIDGenerateText string
|
||||||
|
|
||||||
func GenerateIDSpecs(sourceDir string, destFile string) error {
|
func GenerateIDSpecs(sourceDir string, destFile string) error {
|
||||||
|
|
||||||
files, err := os.ReadDir(sourceDir)
|
files, err := os.ReadDir(sourceDir)
|
||||||
@@ -166,71 +172,22 @@ func processIDFile(basedir string, fn string) ([]IDDef, string, error) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func fmtIDOutput(cs string, ids []IDDef, pkgname string) string {
|
func fmtIDOutput(cs string, ids []IDDef, pkgname string) string {
|
||||||
str := "// Code generated by id-generate.go DO NOT EDIT.\n"
|
templ := template.Must(template.New("id-generate").Parse(templateIDGenerateText))
|
||||||
str += "\n"
|
|
||||||
str += "package " + pkgname + "\n"
|
|
||||||
str += "\n"
|
|
||||||
|
|
||||||
str += "import \"go.mongodb.org/mongo-driver/bson\"" + "\n"
|
buffer := bytes.Buffer{}
|
||||||
str += "import \"go.mongodb.org/mongo-driver/bson/bsontype\"" + "\n"
|
|
||||||
str += "import \"go.mongodb.org/mongo-driver/bson/primitive\"" + "\n"
|
|
||||||
str += "import \"gogs.mikescher.com/BlackForestBytes/goext/exerr\"" + "\n"
|
|
||||||
str += "\n"
|
|
||||||
|
|
||||||
str += "const ChecksumIDGenerator = \"" + cs + "\" // GoExtVersion: " + goext.GoextVersion + "\n"
|
|
||||||
str += "\n"
|
|
||||||
|
|
||||||
anyDef := langext.ArrFirstOrNil(ids, func(def IDDef) bool { return def.Name == "AnyID" || def.Name == "AnyId" })
|
anyDef := langext.ArrFirstOrNil(ids, func(def IDDef) bool { return def.Name == "AnyID" || def.Name == "AnyId" })
|
||||||
|
|
||||||
for _, iddef := range ids {
|
err := templ.Execute(&buffer, langext.H{
|
||||||
|
"PkgName": pkgname,
|
||||||
str += "// ================================ " + iddef.Name + " (" + iddef.FileRelative + ") ================================" + "\n"
|
"Checksum": cs,
|
||||||
str += "" + "\n"
|
"GoextVersion": goext.GoextVersion,
|
||||||
|
"IDs": ids,
|
||||||
str += "func (i " + iddef.Name + ") MarshalBSONValue() (bsontype.Type, []byte, error) {" + "\n"
|
"AnyDef": anyDef,
|
||||||
str += " if objId, err := primitive.ObjectIDFromHex(string(i)); err == nil {" + "\n"
|
})
|
||||||
str += " return bson.MarshalValue(objId)" + "\n"
|
if err != nil {
|
||||||
str += " } else {" + "\n"
|
panic(err)
|
||||||
str += " return 0, nil, exerr.New(exerr.TypeMarshalEntityID, \"Failed to marshal " + iddef.Name + "(\"+i.String()+\") to ObjectId\").Str(\"value\", string(i)).Type(\"type\", i).Build()" + "\n"
|
|
||||||
str += " }" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (i " + iddef.Name + ") String() string {" + "\n"
|
|
||||||
str += " return string(i)" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (i " + iddef.Name + ") ObjID() (primitive.ObjectID, error) {" + "\n"
|
|
||||||
str += " return primitive.ObjectIDFromHex(string(i))" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
str += "func (i " + iddef.Name + ") Valid() bool {" + "\n"
|
|
||||||
str += " _, err := primitive.ObjectIDFromHex(string(i))" + "\n"
|
|
||||||
str += " return err == nil" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
if anyDef != nil {
|
|
||||||
str += "func (i " + iddef.Name + ") AsAny() " + anyDef.Name + " {" + "\n"
|
|
||||||
str += " return " + anyDef.Name + "(i)" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
}
|
|
||||||
|
|
||||||
str += "func New" + iddef.Name + "() " + iddef.Name + " {" + "\n"
|
|
||||||
str += " return " + iddef.Name + "(primitive.NewObjectID().Hex())" + "\n"
|
|
||||||
str += "}" + "\n"
|
|
||||||
|
|
||||||
str += "" + "\n"
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return str
|
return buffer.String()
|
||||||
}
|
}
|
||||||
|
47
bfcodegen/id-generate.template
Normal file
47
bfcodegen/id-generate.template
Normal file
@@ -0,0 +1,47 @@
|
|||||||
|
// Code generated by id-generate.go DO NOT EDIT.
|
||||||
|
|
||||||
|
package {{.PkgName}}
|
||||||
|
|
||||||
|
import "go.mongodb.org/mongo-driver/bson"
|
||||||
|
import "go.mongodb.org/mongo-driver/bson/bsontype"
|
||||||
|
import "go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
|
import "gogs.mikescher.com/BlackForestBytes/goext/exerr"
|
||||||
|
|
||||||
|
const ChecksumIDGenerator = "{{.Checksum}}" // GoExtVersion: {{.GoextVersion}}
|
||||||
|
|
||||||
|
{{range .IDs}}
|
||||||
|
|
||||||
|
// ================================ {{.Name}} ({{.FileRelative}}) ================================
|
||||||
|
|
||||||
|
func (i {{.Name}}) MarshalBSONValue() (bsontype.Type, []byte, error) {
|
||||||
|
if objId, err := primitive.ObjectIDFromHex(string(i)); err == nil {
|
||||||
|
return bson.MarshalValue(objId)
|
||||||
|
} else {
|
||||||
|
return 0, nil, exerr.New(exerr.TypeMarshalEntityID, "Failed to marshal {{.Name}}("+i.String()+") to ObjectId").Str("value", string(i)).Type("type", i).Build()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i {{.Name}}) String() string {
|
||||||
|
return string(i)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i {{.Name}}) ObjID() (primitive.ObjectID, error) {
|
||||||
|
return primitive.ObjectIDFromHex(string(i))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (i {{.Name}}) Valid() bool {
|
||||||
|
_, err := primitive.ObjectIDFromHex(string(i))
|
||||||
|
return err == nil
|
||||||
|
}
|
||||||
|
|
||||||
|
{{if ne $.AnyDef nil}}
|
||||||
|
func (i {{.Name}}) AsAny() {{$.AnyDef.Name}} {
|
||||||
|
return {{$.AnyDef.Name}}(i)
|
||||||
|
}
|
||||||
|
{{end}}
|
||||||
|
|
||||||
|
func New{{.Name}}() {{.Name}} {
|
||||||
|
return {{.Name}}(primitive.NewObjectID().Hex())
|
||||||
|
}
|
||||||
|
|
||||||
|
{{end}}
|
52
bfcodegen/id-generate_test.go
Normal file
52
bfcodegen/id-generate_test.go
Normal file
@@ -0,0 +1,52 @@
|
|||||||
|
package bfcodegen
|
||||||
|
|
||||||
|
import (
|
||||||
|
_ "embed"
|
||||||
|
"fmt"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/cmdext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/tst"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
//go:embed _test_example.tgz
|
||||||
|
var IDExampleModels []byte
|
||||||
|
|
||||||
|
func TestGenerateIDSpecs(t *testing.T) {
|
||||||
|
|
||||||
|
tmpFile := filepath.Join(t.TempDir(), langext.MustHexUUID()+".tgz")
|
||||||
|
|
||||||
|
tmpDir := filepath.Join(t.TempDir(), langext.MustHexUUID())
|
||||||
|
|
||||||
|
err := os.WriteFile(tmpFile, IDExampleModels, 0o777)
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
t.Cleanup(func() { _ = os.Remove(tmpFile) })
|
||||||
|
|
||||||
|
err = os.Mkdir(tmpDir, 0o777)
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
t.Cleanup(func() { _ = os.RemoveAll(tmpFile) })
|
||||||
|
|
||||||
|
_, err = cmdext.Runner("tar").Arg("-xvzf").Arg(tmpFile).Arg("-C").Arg(tmpDir).FailOnExitCode().FailOnTimeout().Timeout(time.Minute).Run()
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go")
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
err = GenerateIDSpecs(tmpDir, tmpDir+"/id_gen.go")
|
||||||
|
tst.AssertNoErr(t, err)
|
||||||
|
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println(string(tst.Must(os.ReadFile(tmpDir + "/id_gen.go"))(t)))
|
||||||
|
fmt.Println("=====================================================================================================")
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
fmt.Println()
|
||||||
|
}
|
@@ -41,12 +41,12 @@ func processEnvOverrides(rval reflect.Value, delim string, prefix string) error
|
|||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if rvfield.Kind() == reflect.Struct {
|
envkey, found := rsfield.Tag.Lookup("env")
|
||||||
|
if !found || envkey == "-" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
envkey, found := rsfield.Tag.Lookup("env")
|
if rvfield.Kind() == reflect.Struct && rvfield.Type() != reflect.TypeOf(time.UnixMilli(0)) {
|
||||||
if !found || envkey == "-" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
|
|
||||||
subPrefix := prefix
|
subPrefix := prefix
|
||||||
if envkey != "" {
|
if envkey != "" {
|
||||||
@@ -57,10 +57,7 @@ func processEnvOverrides(rval reflect.Value, delim string, prefix string) error
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
}
|
|
||||||
|
|
||||||
envkey := rsfield.Tag.Get("env")
|
|
||||||
if envkey == "" || envkey == "-" {
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@@ -275,7 +275,7 @@ func (b *Builder) Any(key string, val any) *Builder {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (b *Builder) Stringer(key string, val fmt.Stringer) *Builder {
|
func (b *Builder) Stringer(key string, val fmt.Stringer) *Builder {
|
||||||
if val == nil {
|
if langext.IsNil(val) {
|
||||||
return b.addMeta(key, MDTString, "(!nil)")
|
return b.addMeta(key, MDTString, "(!nil)")
|
||||||
} else {
|
} else {
|
||||||
return b.addMeta(key, MDTString, val.String())
|
return b.addMeta(key, MDTString, val.String())
|
||||||
|
@@ -56,6 +56,7 @@ var (
|
|||||||
TypeBindFailHeader = NewType("BINDFAIL_HEADER", langext.Ptr(400))
|
TypeBindFailHeader = NewType("BINDFAIL_HEADER", langext.Ptr(400))
|
||||||
|
|
||||||
TypeMarshalEntityID = NewType("MARSHAL_ENTITY_ID", langext.Ptr(400))
|
TypeMarshalEntityID = NewType("MARSHAL_ENTITY_ID", langext.Ptr(400))
|
||||||
|
TypeInvalidCSID = NewType("INVALID_CSID", langext.Ptr(400))
|
||||||
|
|
||||||
TypeUnauthorized = NewType("UNAUTHORIZED", langext.Ptr(401))
|
TypeUnauthorized = NewType("UNAUTHORIZED", langext.Ptr(401))
|
||||||
TypeAuthFailed = NewType("AUTH_FAILED", langext.Ptr(401))
|
TypeAuthFailed = NewType("AUTH_FAILED", langext.Ptr(401))
|
||||||
|
@@ -30,7 +30,7 @@ type ExErr struct {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (ee *ExErr) Error() string {
|
func (ee *ExErr) Error() string {
|
||||||
return ee.Message
|
return ee.RecursiveMessage()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Unwrap must be implemented so that some error.XXX methods work
|
// Unwrap must be implemented so that some error.XXX methods work
|
||||||
|
36
fsext/exists.go
Normal file
36
fsext/exists.go
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
package fsext
|
||||||
|
|
||||||
|
import "os"
|
||||||
|
|
||||||
|
func PathExists(fp string) (bool, error) {
|
||||||
|
_, err := os.Stat(fp)
|
||||||
|
if err == nil {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func FileExists(fp string) (bool, error) {
|
||||||
|
stat, err := os.Stat(fp)
|
||||||
|
if err == nil {
|
||||||
|
return !stat.IsDir(), nil
|
||||||
|
}
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func DirectoryExists(fp string) (bool, error) {
|
||||||
|
stat, err := os.Stat(fp)
|
||||||
|
if err == nil {
|
||||||
|
return stat.IsDir(), nil
|
||||||
|
}
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return false, err
|
||||||
|
}
|
@@ -59,7 +59,7 @@ func NewEngine(allowCors bool, ginDebug bool, bufferBody bool, timeout time.Dura
|
|||||||
// do not debug-print routes
|
// do not debug-print routes
|
||||||
gin.DebugPrintRouteFunc = func(_, _, _ string, _ int) {}
|
gin.DebugPrintRouteFunc = func(_, _, _ string, _ int) {}
|
||||||
|
|
||||||
if ginDebug {
|
if !ginDebug {
|
||||||
gin.SetMode(gin.ReleaseMode)
|
gin.SetMode(gin.ReleaseMode)
|
||||||
|
|
||||||
ginlogger := gin.Logger()
|
ginlogger := gin.Logger()
|
||||||
|
8
go.mod
8
go.mod
@@ -17,7 +17,7 @@ require (
|
|||||||
github.com/bytedance/sonic v1.10.2 // indirect
|
github.com/bytedance/sonic v1.10.2 // indirect
|
||||||
github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect
|
github.com/chenzhuoyu/base64x v0.0.0-20230717121745-296ad89f973d // indirect
|
||||||
github.com/chenzhuoyu/iasm v0.9.0 // indirect
|
github.com/chenzhuoyu/iasm v0.9.0 // indirect
|
||||||
github.com/gabriel-vasile/mimetype v1.4.2 // indirect
|
github.com/gabriel-vasile/mimetype v1.4.3 // indirect
|
||||||
github.com/gin-contrib/sse v0.1.0 // indirect
|
github.com/gin-contrib/sse v0.1.0 // indirect
|
||||||
github.com/go-playground/locales v0.14.1 // indirect
|
github.com/go-playground/locales v0.14.1 // indirect
|
||||||
github.com/go-playground/universal-translator v0.18.1 // indirect
|
github.com/go-playground/universal-translator v0.18.1 // indirect
|
||||||
@@ -25,11 +25,11 @@ require (
|
|||||||
github.com/goccy/go-json v0.10.2 // indirect
|
github.com/goccy/go-json v0.10.2 // indirect
|
||||||
github.com/golang/snappy v0.0.4 // indirect
|
github.com/golang/snappy v0.0.4 // indirect
|
||||||
github.com/json-iterator/go v1.1.12 // indirect
|
github.com/json-iterator/go v1.1.12 // indirect
|
||||||
github.com/klauspost/compress v1.17.0 // indirect
|
github.com/klauspost/compress v1.17.2 // indirect
|
||||||
github.com/klauspost/cpuid/v2 v2.2.5 // indirect
|
github.com/klauspost/cpuid/v2 v2.2.5 // indirect
|
||||||
github.com/leodido/go-urn v1.2.4 // indirect
|
github.com/leodido/go-urn v1.2.4 // indirect
|
||||||
github.com/mattn/go-colorable v0.1.13 // indirect
|
github.com/mattn/go-colorable v0.1.13 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.19 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
github.com/montanaflynn/stats v0.7.1 // indirect
|
github.com/montanaflynn/stats v0.7.1 // indirect
|
||||||
@@ -41,7 +41,7 @@ require (
|
|||||||
github.com/xdg-go/stringprep v1.0.4 // indirect
|
github.com/xdg-go/stringprep v1.0.4 // indirect
|
||||||
github.com/youmark/pkcs8 v0.0.0-20201027041543-1326539a0a0a // indirect
|
github.com/youmark/pkcs8 v0.0.0-20201027041543-1326539a0a0a // indirect
|
||||||
golang.org/x/arch v0.5.0 // indirect
|
golang.org/x/arch v0.5.0 // indirect
|
||||||
golang.org/x/net v0.16.0 // indirect
|
golang.org/x/net v0.17.0 // indirect
|
||||||
golang.org/x/sync v0.4.0 // indirect
|
golang.org/x/sync v0.4.0 // indirect
|
||||||
golang.org/x/text v0.13.0 // indirect
|
golang.org/x/text v0.13.0 // indirect
|
||||||
google.golang.org/protobuf v1.31.0 // indirect
|
google.golang.org/protobuf v1.31.0 // indirect
|
||||||
|
10
go.sum
10
go.sum
@@ -14,6 +14,8 @@ github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c
|
|||||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||||
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
github.com/gabriel-vasile/mimetype v1.4.2 h1:w5qFW6JKBz9Y393Y4q372O9A7cUSequkh1Q7OhCmWKU=
|
||||||
github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
|
github.com/gabriel-vasile/mimetype v1.4.2/go.mod h1:zApsH/mKG4w07erKIaJPFiX0Tsq9BFQgN3qGY5GnNgA=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=
|
||||||
|
github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
|
||||||
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
|
||||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
|
||||||
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
|
||||||
@@ -45,6 +47,10 @@ github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHm
|
|||||||
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
github.com/klauspost/compress v1.13.6/go.mod h1:/3/Vjq9QcHkK5uEr5lBEmyoZ1iFhe47etQ6QUkpK6sk=
|
||||||
github.com/klauspost/compress v1.17.0 h1:Rnbp4K9EjcDuVuHtd0dgA4qNuv9yKDYKK1ulpJwgrqM=
|
github.com/klauspost/compress v1.17.0 h1:Rnbp4K9EjcDuVuHtd0dgA4qNuv9yKDYKK1ulpJwgrqM=
|
||||||
github.com/klauspost/compress v1.17.0/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
github.com/klauspost/compress v1.17.0/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||||
|
github.com/klauspost/compress v1.17.1 h1:NE3C767s2ak2bweCZo3+rdP4U/HoyVXLv/X9f2gPS5g=
|
||||||
|
github.com/klauspost/compress v1.17.1/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||||
|
github.com/klauspost/compress v1.17.2 h1:RlWWUY/Dr4fL8qk9YG7DTZ7PDgME2V4csBXA8L/ixi4=
|
||||||
|
github.com/klauspost/compress v1.17.2/go.mod h1:ntbaceVETuRiXiv4DpjP66DpAtAGkEQskQzEyD//IeE=
|
||||||
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
|
||||||
github.com/klauspost/cpuid/v2 v2.2.5 h1:0E5MSMDEoAulmXNFquVs//DdoomxaoTY1kUhbc/qbZg=
|
github.com/klauspost/cpuid/v2 v2.2.5 h1:0E5MSMDEoAulmXNFquVs//DdoomxaoTY1kUhbc/qbZg=
|
||||||
github.com/klauspost/cpuid/v2 v2.2.5/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
|
github.com/klauspost/cpuid/v2 v2.2.5/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=
|
||||||
@@ -58,6 +64,8 @@ github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovk
|
|||||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
||||||
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
github.com/mattn/go-isatty v0.0.19 h1:JITubQf0MOLdlGRuRq+jtsDlekdYPia9ZFsB8h/APPA=
|
||||||
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
github.com/mattn/go-isatty v0.0.19/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
github.com/mattn/go-sqlite3 v1.14.6 h1:dNPt6NO46WmLVt2DLNpwczCmdV5boIZ6g/tlDrlRUbg=
|
github.com/mattn/go-sqlite3 v1.14.6 h1:dNPt6NO46WmLVt2DLNpwczCmdV5boIZ6g/tlDrlRUbg=
|
||||||
github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU=
|
github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU=
|
||||||
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=
|
||||||
@@ -121,6 +129,8 @@ golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qx
|
|||||||
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
|
||||||
golang.org/x/net v0.16.0 h1:7eBu7KsSvFDtSXUIDbh3aqlK4DPsZ1rByC8PFfBThos=
|
golang.org/x/net v0.16.0 h1:7eBu7KsSvFDtSXUIDbh3aqlK4DPsZ1rByC8PFfBThos=
|
||||||
golang.org/x/net v0.16.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
|
golang.org/x/net v0.16.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
|
||||||
|
golang.org/x/net v0.17.0 h1:pVaXccu2ozPjCXewfr1S7xza/zcXTity9cCdXQYSjIM=
|
||||||
|
golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE=
|
||||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||||
golang.org/x/sync v0.4.0 h1:zxkM55ReGkDlKSM+Fu41A+zmbZuaPVbGMzvvdUPznYQ=
|
golang.org/x/sync v0.4.0 h1:zxkM55ReGkDlKSM+Fu41A+zmbZuaPVbGMzvvdUPznYQ=
|
||||||
|
@@ -1,5 +1,5 @@
|
|||||||
package goext
|
package goext
|
||||||
|
|
||||||
const GoextVersion = "0.0.281"
|
const GoextVersion = "0.0.296"
|
||||||
|
|
||||||
const GoextVersionTimestamp = "2023-10-09T09:04:07+0200"
|
const GoextVersionTimestamp = "2023-11-01T00:29:58+0100"
|
||||||
|
178
langext/baseAny.go
Normal file
178
langext/baseAny.go
Normal file
@@ -0,0 +1,178 @@
|
|||||||
|
package langext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"crypto/rand"
|
||||||
|
"errors"
|
||||||
|
"math"
|
||||||
|
"math/big"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AnyBaseConverter struct {
|
||||||
|
base uint64
|
||||||
|
charset []rune
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAnyBaseConverter(cs string) AnyBaseConverter {
|
||||||
|
rcs := []rune(cs)
|
||||||
|
return AnyBaseConverter{
|
||||||
|
base: uint64(len(rcs)),
|
||||||
|
charset: rcs,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) Rand(rlen int) string {
|
||||||
|
biBase := big.NewInt(int64(bc.base))
|
||||||
|
|
||||||
|
randMax := big.NewInt(math.MaxInt64)
|
||||||
|
|
||||||
|
r := ""
|
||||||
|
|
||||||
|
for i := 0; i < rlen; i++ {
|
||||||
|
v, err := rand.Int(rand.Reader, randMax)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
r += string(bc.charset[v.Mod(v, biBase).Int64()])
|
||||||
|
}
|
||||||
|
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) EncodeUInt64(num uint64) string {
|
||||||
|
if num == 0 {
|
||||||
|
return "0"
|
||||||
|
}
|
||||||
|
|
||||||
|
b := ""
|
||||||
|
|
||||||
|
// loop as long the num is bigger than zero
|
||||||
|
for num > 0 {
|
||||||
|
r := num % bc.base
|
||||||
|
|
||||||
|
num -= r
|
||||||
|
num /= base62Base
|
||||||
|
|
||||||
|
b += string(bc.charset[int(r)])
|
||||||
|
}
|
||||||
|
|
||||||
|
return b
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) DecodeUInt64(str string) (uint64, error) {
|
||||||
|
if str == "" {
|
||||||
|
return 0, errors.New("empty string")
|
||||||
|
}
|
||||||
|
|
||||||
|
result := uint64(0)
|
||||||
|
|
||||||
|
for _, v := range str {
|
||||||
|
result *= base62Base
|
||||||
|
|
||||||
|
pos := ArrFirstIndex(bc.charset, v)
|
||||||
|
if pos == -1 {
|
||||||
|
return 0, errors.New("invalid character: " + string(v))
|
||||||
|
}
|
||||||
|
|
||||||
|
result += uint64(pos)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) Encode(src []byte) string {
|
||||||
|
value := new(big.Int)
|
||||||
|
value.SetBytes(src)
|
||||||
|
return bc.EncodeBigInt(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) EncodeBigInt(src *big.Int) string {
|
||||||
|
value := new(big.Int)
|
||||||
|
value.Set(src)
|
||||||
|
|
||||||
|
isneg := value.Sign() < 0
|
||||||
|
|
||||||
|
answer := ""
|
||||||
|
|
||||||
|
if isneg {
|
||||||
|
value.Neg(value)
|
||||||
|
}
|
||||||
|
|
||||||
|
biBase := big.NewInt(int64(bc.base))
|
||||||
|
|
||||||
|
rem := new(big.Int)
|
||||||
|
|
||||||
|
for value.Sign() > 0 {
|
||||||
|
value.QuoRem(value, biBase, rem)
|
||||||
|
answer = string(bc.charset[rem.Int64()]) + answer
|
||||||
|
}
|
||||||
|
|
||||||
|
if isneg {
|
||||||
|
return "-" + answer
|
||||||
|
} else {
|
||||||
|
return answer
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) Decode(src string) ([]byte, error) {
|
||||||
|
value, err := bc.DecodeToBigInt(src)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return value.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (bc AnyBaseConverter) DecodeToBigInt(_src string) (*big.Int, error) {
|
||||||
|
result := new(big.Int)
|
||||||
|
result.SetInt64(0)
|
||||||
|
|
||||||
|
src := []rune(_src)
|
||||||
|
|
||||||
|
if len(src) == 0 {
|
||||||
|
return nil, errors.New("string is empty")
|
||||||
|
}
|
||||||
|
if bc.base < 2 {
|
||||||
|
return nil, errors.New("not enough digits")
|
||||||
|
}
|
||||||
|
|
||||||
|
i := 0
|
||||||
|
|
||||||
|
sign := new(big.Int)
|
||||||
|
sign.SetInt64(1)
|
||||||
|
if src[i] == '+' {
|
||||||
|
i++
|
||||||
|
} else if src[i] == '-' {
|
||||||
|
i++
|
||||||
|
sign.SetInt64(-1)
|
||||||
|
}
|
||||||
|
|
||||||
|
if i >= len(src) {
|
||||||
|
return nil, errors.New("no digits in input")
|
||||||
|
}
|
||||||
|
|
||||||
|
biBase := big.NewInt(int64(bc.base))
|
||||||
|
|
||||||
|
oldResult := new(big.Int)
|
||||||
|
|
||||||
|
for ; i < len(src); i++ {
|
||||||
|
n := ArrFirstIndex(bc.charset, src[i])
|
||||||
|
if n < 0 {
|
||||||
|
return nil, errors.New("invalid characters in input")
|
||||||
|
}
|
||||||
|
|
||||||
|
oldResult.Set(result)
|
||||||
|
|
||||||
|
result.Mul(result, biBase)
|
||||||
|
result.Add(result, big.NewInt(int64(n)))
|
||||||
|
|
||||||
|
if result.Cmp(oldResult) < 0 {
|
||||||
|
return nil, errors.New("overflow")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if sign.Cmp(big.NewInt(0)) < 0 {
|
||||||
|
result.Neg(result)
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
80
langext/baseAny_test.go
Normal file
80
langext/baseAny_test.go
Normal file
@@ -0,0 +1,80 @@
|
|||||||
|
package langext
|
||||||
|
|
||||||
|
import (
|
||||||
|
"gogs.mikescher.com/BlackForestBytes/goext/tst"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func _anyEncStr(bc AnyBaseConverter, v string) string {
|
||||||
|
vr := bc.Encode([]byte(v))
|
||||||
|
return vr
|
||||||
|
}
|
||||||
|
|
||||||
|
func _anyDecStr(bc AnyBaseConverter, v string) string {
|
||||||
|
vr, err := bc.Decode(v)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
return string(vr)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAnyBase58DefaultEncoding(t *testing.T) {
|
||||||
|
tst.AssertEqual(t, _anyEncStr(NewAnyBaseConverter("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"), "Hello"), "9Ajdvzr")
|
||||||
|
tst.AssertEqual(t, _anyEncStr(NewAnyBaseConverter("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in."), "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAnyBase58DefaultDecoding(t *testing.T) {
|
||||||
|
tst.AssertEqual(t, _anyDecStr(NewAnyBaseConverter("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"), "9Ajdvzr"), "Hello")
|
||||||
|
tst.AssertEqual(t, _anyDecStr(NewAnyBaseConverter("123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"), "48638SMcJuah5okqPx4kCVf5d8QAdgbdNf28g7ReY13prUENNbMyssjq5GjsrJHF5zeZfqs4uJMUJHr7VbrU4XBUZ2Fw9DVtqtn9N1eXucEWSEZahXV6w4ysGSWqGdpeYTJf1MdDzTg8vfcQViifJjZX"), "If debugging is the process of removing software bugs, then programming must be the process of putting them in.")
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAnyBaseDecode(t *testing.T) {
|
||||||
|
|
||||||
|
const (
|
||||||
|
Binary = "01"
|
||||||
|
Decimal = "0123456789"
|
||||||
|
Hex = "0123456789ABCDEF"
|
||||||
|
DNA = "ACGT"
|
||||||
|
Base32 = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567"
|
||||||
|
Base58 = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"
|
||||||
|
Base62 = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
||||||
|
Base64 = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||||||
|
Base256 = "🚀🪐☄🛰🌌🌑🌒🌓🌔🌕🌖🌗🌘🌍🌏🌎🐉☀💻🖥💾💿😂❤😍🤣😊🙏💕😭😘👍😅👏😁🔥🥰💔💖💙😢🤔😆🙄💪😉☺👌🤗💜😔😎😇🌹🤦🎉💞✌✨🤷😱😌🌸🙌😋💗💚😏💛🙂💓🤩😄😀🖤😃💯🙈👇🎶😒🤭❣😜💋👀😪😑💥🙋😞😩😡🤪👊🥳😥🤤👉💃😳✋😚😝😴🌟😬🙃🍀🌷😻😓⭐✅🥺🌈😈🤘💦✔😣🏃💐☹🎊💘😠☝😕🌺🎂🌻😐🖕💝🙊😹🗣💫💀👑🎵🤞😛🔴😤🌼😫⚽🤙☕🏆🤫👈😮🙆🍻🍃🐶💁😲🌿🧡🎁⚡🌞🎈❌✊👋😰🤨😶🤝🚶💰🍓💢🤟🙁🚨💨🤬✈🎀🍺🤓😙💟🌱😖👶🥴▶➡❓💎💸⬇😨🌚🦋😷🕺⚠🙅😟😵👎🤲🤠🤧📌🔵💅🧐🐾🍒😗🤑🌊🤯🐷☎💧😯💆👆🎤🙇🍑❄🌴💣🐸💌📍🥀🤢👅💡💩👐📸👻🤐🤮🎼🥵🚩🍎🍊👼💍📣🥂"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TestDef struct {
|
||||||
|
FromCS string
|
||||||
|
FromVal string
|
||||||
|
ToCS string
|
||||||
|
ToVal string
|
||||||
|
}
|
||||||
|
|
||||||
|
defs := []TestDef{
|
||||||
|
{Binary, "10100101011100000101010", Decimal, "5421098"},
|
||||||
|
{Decimal, "5421098", DNA, "CCAGGTGAAGGG"},
|
||||||
|
{Decimal, "5421098", DNA, "CCAGGTGAAGGG"},
|
||||||
|
{Decimal, "80085", Base256, "🪐💞🔵"},
|
||||||
|
{Hex, "48656C6C6C20576F526C5421", Base64, "SGVsbGwgV29SbFQh"},
|
||||||
|
{Base64, "SGVsbGw/gV29SbF+Qh", Base32, "CIMVWGY3B7QFO32SNRPZBB"},
|
||||||
|
{Base64, "SGVsbGw/gV29SbF+Qh", Base58, "2fUsGKQUcgQcwSqpvy6"},
|
||||||
|
{Base64, "SGVsbGw/gV29SbF+Qh", Base62, "V34nvybdQ3m3RHk9Sr"},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, def := range defs {
|
||||||
|
|
||||||
|
d1 := NewAnyBaseConverter(def.FromCS)
|
||||||
|
d2 := NewAnyBaseConverter(def.ToCS)
|
||||||
|
|
||||||
|
v1 := tst.Must(d1.Decode(def.FromVal))(t)
|
||||||
|
v2 := tst.Must(d2.Decode(def.ToVal))(t)
|
||||||
|
|
||||||
|
tst.AssertArrayEqual(t, v1, v2)
|
||||||
|
|
||||||
|
str2 := d2.Encode(v1)
|
||||||
|
tst.AssertEqual(t, str2, def.ToVal)
|
||||||
|
|
||||||
|
str1 := d1.Encode(v2)
|
||||||
|
tst.AssertEqual(t, str1, def.FromVal)
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
@@ -3,6 +3,8 @@ package mongoext
|
|||||||
import (
|
import (
|
||||||
"go.mongodb.org/mongo-driver/bson"
|
"go.mongodb.org/mongo-driver/bson"
|
||||||
"go.mongodb.org/mongo-driver/bson/bsoncodec"
|
"go.mongodb.org/mongo-driver/bson/bsoncodec"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/bsontype"
|
||||||
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
||||||
"gogs.mikescher.com/BlackForestBytes/goext/rfctime"
|
"gogs.mikescher.com/BlackForestBytes/goext/rfctime"
|
||||||
"reflect"
|
"reflect"
|
||||||
)
|
)
|
||||||
@@ -24,5 +26,9 @@ func CreateGoExtBsonRegistry() *bsoncodec.Registry {
|
|||||||
|
|
||||||
bson.PrimitiveCodecs{}.RegisterPrimitiveCodecs(rb)
|
bson.PrimitiveCodecs{}.RegisterPrimitiveCodecs(rb)
|
||||||
|
|
||||||
|
// otherwise we get []primitve.E when unmarshalling into any
|
||||||
|
// which will result in {'key': .., 'value': ...}[] json when json-marshalling
|
||||||
|
rb.RegisterTypeMapEntry(bsontype.EmbeddedDocument, reflect.TypeOf(primitive.M{}))
|
||||||
|
|
||||||
return rb.Build()
|
return rb.Build()
|
||||||
}
|
}
|
||||||
|
@@ -14,6 +14,20 @@ func AssertEqual[T comparable](t *testing.T, actual T, expected T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func AssertArrayEqual[T comparable](t *testing.T, actual []T, expected []T) {
|
||||||
|
t.Helper()
|
||||||
|
if len(actual) != len(expected) {
|
||||||
|
t.Errorf("values differ: Actual: '%v', Expected: '%v' (len %d <> %d)", actual, expected, len(actual), len(expected))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
for i := 0; i < len(actual); i++ {
|
||||||
|
if actual[i] != expected[i] {
|
||||||
|
t.Errorf("values differ: Actual: '%v', Expected: '%v' (at index %d)", actual, expected, i)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func AssertNotEqual[T comparable](t *testing.T, actual T, expected T) {
|
func AssertNotEqual[T comparable](t *testing.T, actual T, expected T) {
|
||||||
t.Helper()
|
t.Helper()
|
||||||
if actual == expected {
|
if actual == expected {
|
||||||
|
21
tst/must.go
Normal file
21
tst/must.go
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
package tst
|
||||||
|
|
||||||
|
import (
|
||||||
|
"runtime/debug"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Must can b used to AssertNoErr of an (T, err) function
|
||||||
|
//
|
||||||
|
// Usage:
|
||||||
|
//
|
||||||
|
// input := "123.8"
|
||||||
|
// value := tst.Must(strconv.Atoi(input))(t)
|
||||||
|
func Must[T any](v T, anerr error) func(t *testing.T) T {
|
||||||
|
return func(t *testing.T) T {
|
||||||
|
if anerr != nil {
|
||||||
|
t.Error("Function returned an error: " + anerr.Error() + "\n" + string(debug.Stack()))
|
||||||
|
}
|
||||||
|
return v
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user