Compare commits
30 Commits
Author | SHA1 | Date | |
---|---|---|---|
36ed474bfe
|
|||
fdc590c8c3
|
|||
1990e5d32d
|
|||
72883cf6bd
|
|||
ff08d5f180
|
|||
72d6b538f7
|
|||
48dd30fb94
|
|||
b7c5756f11
|
|||
2070a432a5
|
|||
34e6d1819d
|
|||
87fa6021e4
|
|||
297d6c52a8
|
|||
b9c46947d2
|
|||
412277b3e0
|
|||
e46f8019ec
|
|||
ae952b2166
|
|||
b24dba9a45
|
|||
cfbc20367d
|
|||
e25912758e
|
|||
e1ae77a9db
|
|||
9d07b3955f
|
|||
02be696c25
|
|||
ba07625b7c
|
|||
aeded3fb37
|
|||
1a1cd6d0aa
|
|||
64cc1342a0
|
|||
8431b6adf5
|
|||
24e923fe84
|
|||
10ddc7c190
|
|||
7f88a0726c
|
1
.idea/inspectionProfiles/Project_Default.xml
generated
1
.idea/inspectionProfiles/Project_Default.xml
generated
@@ -1,6 +1,7 @@
|
||||
<component name="InspectionProjectProfileManager">
|
||||
<profile version="1.0">
|
||||
<option name="myName" value="Project Default" />
|
||||
<inspection_tool class="GoMixedReceiverTypes" enabled="false" level="WEAK WARNING" enabled_by_default="false" />
|
||||
<inspection_tool class="LanguageDetectionInspection" enabled="false" level="WARNING" enabled_by_default="false" />
|
||||
<inspection_tool class="SpellCheckingInspection" enabled="false" level="TYPO" enabled_by_default="false">
|
||||
<option name="processCode" value="true" />
|
||||
|
73
cmdext/builder.go
Normal file
73
cmdext/builder.go
Normal file
@@ -0,0 +1,73 @@
|
||||
package cmdext
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
type CommandRunner struct {
|
||||
program string
|
||||
args []string
|
||||
timeout *time.Duration
|
||||
env []string
|
||||
listener []CommandListener
|
||||
}
|
||||
|
||||
func Runner(program string) *CommandRunner {
|
||||
return &CommandRunner{
|
||||
program: program,
|
||||
args: make([]string, 0),
|
||||
timeout: nil,
|
||||
env: make([]string, 0),
|
||||
listener: make([]CommandListener, 0),
|
||||
}
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Arg(arg string) *CommandRunner {
|
||||
r.args = append(r.args, arg)
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Args(arg []string) *CommandRunner {
|
||||
r.args = append(r.args, arg...)
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Timeout(timeout time.Duration) *CommandRunner {
|
||||
r.timeout = &timeout
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Env(key, value string) *CommandRunner {
|
||||
r.env = append(r.env, fmt.Sprintf("%s=%s", key, value))
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) RawEnv(env string) *CommandRunner {
|
||||
r.env = append(r.env, env)
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Envs(env []string) *CommandRunner {
|
||||
r.env = append(r.env, env...)
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Listen(lstr CommandListener) *CommandRunner {
|
||||
r.listener = append(r.listener, lstr)
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) ListenStdout(lstr func(string)) *CommandRunner {
|
||||
r.listener = append(r.listener, genericCommandListener{_readStdoutLine: &lstr})
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) ListenStderr(lstr func(string)) *CommandRunner {
|
||||
r.listener = append(r.listener, genericCommandListener{_readStderrLine: &lstr})
|
||||
return r
|
||||
}
|
||||
|
||||
func (r *CommandRunner) Run() (CommandResult, error) {
|
||||
return run(*r)
|
||||
}
|
129
cmdext/cmdrunner.go
Normal file
129
cmdext/cmdrunner.go
Normal file
@@ -0,0 +1,129 @@
|
||||
package cmdext
|
||||
|
||||
import (
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/mathext"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/syncext"
|
||||
"os/exec"
|
||||
"time"
|
||||
)
|
||||
|
||||
type CommandResult struct {
|
||||
StdOut string
|
||||
StdErr string
|
||||
StdCombined string
|
||||
ExitCode int
|
||||
CommandTimedOut bool
|
||||
}
|
||||
|
||||
func run(opt CommandRunner) (CommandResult, error) {
|
||||
cmd := exec.Command(opt.program, opt.args...)
|
||||
|
||||
cmd.Env = append(cmd.Env, opt.env...)
|
||||
|
||||
stdoutPipe, err := cmd.StdoutPipe()
|
||||
if err != nil {
|
||||
return CommandResult{}, err
|
||||
}
|
||||
|
||||
stderrPipe, err := cmd.StderrPipe()
|
||||
if err != nil {
|
||||
return CommandResult{}, err
|
||||
}
|
||||
|
||||
preader := pipeReader{
|
||||
stdout: stdoutPipe,
|
||||
stderr: stderrPipe,
|
||||
}
|
||||
|
||||
err = cmd.Start()
|
||||
if err != nil {
|
||||
return CommandResult{}, err
|
||||
}
|
||||
|
||||
type resultObj struct {
|
||||
stdout string
|
||||
stderr string
|
||||
stdcombined string
|
||||
err error
|
||||
}
|
||||
|
||||
outputChan := make(chan resultObj)
|
||||
go func() {
|
||||
// we need to first fully read the pipes and then call Wait
|
||||
// see https://pkg.go.dev/os/exec#Cmd.StdoutPipe
|
||||
|
||||
stdout, stderr, stdcombined, err := preader.Read(opt.listener)
|
||||
if err != nil {
|
||||
outputChan <- resultObj{stdout, stderr, stdcombined, err}
|
||||
}
|
||||
|
||||
err = cmd.Wait()
|
||||
if err != nil {
|
||||
outputChan <- resultObj{stdout, stderr, stdcombined, err}
|
||||
}
|
||||
|
||||
outputChan <- resultObj{stdout, stderr, stdcombined, nil}
|
||||
}()
|
||||
|
||||
var timeoutChan <-chan time.Time = make(chan time.Time, 1)
|
||||
if opt.timeout != nil {
|
||||
timeoutChan = time.After(*opt.timeout)
|
||||
}
|
||||
|
||||
select {
|
||||
|
||||
case <-timeoutChan:
|
||||
_ = cmd.Process.Kill()
|
||||
for _, lstr := range opt.listener {
|
||||
lstr.Timeout()
|
||||
}
|
||||
|
||||
if fallback, ok := syncext.ReadChannelWithTimeout(outputChan, mathext.Min(32*time.Millisecond, *opt.timeout)); ok {
|
||||
// most of the time the cmd.Process.Kill() should also ahve finished the pipereader
|
||||
// and we can at least return the already collected stdout, stderr, etc
|
||||
return CommandResult{
|
||||
StdOut: fallback.stdout,
|
||||
StdErr: fallback.stderr,
|
||||
StdCombined: fallback.stdcombined,
|
||||
ExitCode: -1,
|
||||
CommandTimedOut: true,
|
||||
}, nil
|
||||
} else {
|
||||
return CommandResult{
|
||||
StdOut: "",
|
||||
StdErr: "",
|
||||
StdCombined: "",
|
||||
ExitCode: -1,
|
||||
CommandTimedOut: true,
|
||||
}, nil
|
||||
}
|
||||
|
||||
case outobj := <-outputChan:
|
||||
if exiterr, ok := outobj.err.(*exec.ExitError); ok {
|
||||
excode := exiterr.ExitCode()
|
||||
for _, lstr := range opt.listener {
|
||||
lstr.Finished(excode)
|
||||
}
|
||||
return CommandResult{
|
||||
StdOut: outobj.stdout,
|
||||
StdErr: outobj.stderr,
|
||||
StdCombined: outobj.stdcombined,
|
||||
ExitCode: excode,
|
||||
CommandTimedOut: false,
|
||||
}, nil
|
||||
} else if err != nil {
|
||||
return CommandResult{}, err
|
||||
} else {
|
||||
for _, lstr := range opt.listener {
|
||||
lstr.Finished(0)
|
||||
}
|
||||
return CommandResult{
|
||||
StdOut: outobj.stdout,
|
||||
StdErr: outobj.stderr,
|
||||
StdCombined: outobj.stdcombined,
|
||||
ExitCode: 0,
|
||||
CommandTimedOut: false,
|
||||
}, nil
|
||||
}
|
||||
}
|
||||
}
|
198
cmdext/cmdrunner_test.go
Normal file
198
cmdext/cmdrunner_test.go
Normal file
@@ -0,0 +1,198 @@
|
||||
package cmdext
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestStdout(t *testing.T) {
|
||||
|
||||
res1, err := Runner("printf").Arg("hello").Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if res1.StdErr != "" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "hello" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "hello\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestStderr(t *testing.T) {
|
||||
|
||||
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"error\", file=sys.stderr, end='')").Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if res1.StdErr != "error" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "error\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestStdcombined(t *testing.T) {
|
||||
res1, err := Runner("python").
|
||||
Arg("-c").
|
||||
Arg("import sys; import time; print(\"1\", file=sys.stderr, flush=True); time.sleep(0.1); print(\"2\", file=sys.stdout, flush=True); time.sleep(0.1); print(\"3\", file=sys.stderr, flush=True)").
|
||||
Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if res1.StdErr != "1\n3\n" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "2\n" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "1\n2\n3\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestPartialRead(t *testing.T) {
|
||||
res1, err := Runner("python").
|
||||
Arg("-c").
|
||||
Arg("import sys; import time; print(\"first message\", flush=True); time.sleep(5); print(\"cant see me\", flush=True);").
|
||||
Timeout(100 * time.Millisecond).
|
||||
Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if !res1.CommandTimedOut {
|
||||
t.Errorf("!CommandTimedOut")
|
||||
}
|
||||
if res1.StdErr != "" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "first message\n" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "first message\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestPartialReadStderr(t *testing.T) {
|
||||
res1, err := Runner("python").
|
||||
Arg("-c").
|
||||
Arg("import sys; import time; print(\"first message\", file=sys.stderr, flush=True); time.sleep(5); print(\"cant see me\", file=sys.stderr, flush=True);").
|
||||
Timeout(100 * time.Millisecond).
|
||||
Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if !res1.CommandTimedOut {
|
||||
t.Errorf("!CommandTimedOut")
|
||||
}
|
||||
if res1.StdErr != "first message\n" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "first message\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestReadUnflushedStdout(t *testing.T) {
|
||||
|
||||
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"message101\", file=sys.stdout, end='')").Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if res1.StdErr != "" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "message101" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "message101\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestReadUnflushedStderr(t *testing.T) {
|
||||
|
||||
res1, err := Runner("python").Arg("-c").Arg("import sys; print(\"message101\", file=sys.stderr, end='')").Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if res1.StdErr != "message101" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "message101\n" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestPartialReadUnflushed(t *testing.T) {
|
||||
t.SkipNow()
|
||||
|
||||
res1, err := Runner("python").
|
||||
Arg("-c").
|
||||
Arg("import sys; import time; print(\"first message\", end=''); time.sleep(5); print(\"cant see me\", end='');").
|
||||
Timeout(100 * time.Millisecond).
|
||||
Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if !res1.CommandTimedOut {
|
||||
t.Errorf("!CommandTimedOut")
|
||||
}
|
||||
if res1.StdErr != "" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "first message" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "first message" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func TestPartialReadUnflushedStderr(t *testing.T) {
|
||||
t.SkipNow()
|
||||
|
||||
res1, err := Runner("python").
|
||||
Arg("-c").
|
||||
Arg("import sys; import time; print(\"first message\", file=sys.stderr, end=''); time.sleep(5); print(\"cant see me\", file=sys.stderr, end='');").
|
||||
Timeout(100 * time.Millisecond).
|
||||
Run()
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
}
|
||||
if !res1.CommandTimedOut {
|
||||
t.Errorf("!CommandTimedOut")
|
||||
}
|
||||
if res1.StdErr != "first message" {
|
||||
t.Errorf("res1.StdErr == '%v'", res1.StdErr)
|
||||
}
|
||||
if res1.StdOut != "" {
|
||||
t.Errorf("res1.StdOut == '%v'", res1.StdOut)
|
||||
}
|
||||
if res1.StdCombined != "first message" {
|
||||
t.Errorf("res1.StdCombined == '%v'", res1.StdCombined)
|
||||
}
|
||||
|
||||
}
|
12
cmdext/helper.go
Normal file
12
cmdext/helper.go
Normal file
@@ -0,0 +1,12 @@
|
||||
package cmdext
|
||||
|
||||
import "time"
|
||||
|
||||
func RunCommand(program string, args []string, timeout *time.Duration) (CommandResult, error) {
|
||||
b := Runner(program)
|
||||
b = b.Args(args)
|
||||
if timeout != nil {
|
||||
b = b.Timeout(*timeout)
|
||||
}
|
||||
return b.Run()
|
||||
}
|
57
cmdext/listener.go
Normal file
57
cmdext/listener.go
Normal file
@@ -0,0 +1,57 @@
|
||||
package cmdext
|
||||
|
||||
type CommandListener interface {
|
||||
ReadRawStdout([]byte)
|
||||
ReadRawStderr([]byte)
|
||||
|
||||
ReadStdoutLine(string)
|
||||
ReadStderrLine(string)
|
||||
|
||||
Finished(int)
|
||||
Timeout()
|
||||
}
|
||||
|
||||
type genericCommandListener struct {
|
||||
_readRawStdout *func([]byte)
|
||||
_readRawStderr *func([]byte)
|
||||
_readStdoutLine *func(string)
|
||||
_readStderrLine *func(string)
|
||||
_finished *func(int)
|
||||
_timeout *func()
|
||||
}
|
||||
|
||||
func (g genericCommandListener) ReadRawStdout(v []byte) {
|
||||
if g._readRawStdout != nil {
|
||||
(*g._readRawStdout)(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (g genericCommandListener) ReadRawStderr(v []byte) {
|
||||
if g._readRawStderr != nil {
|
||||
(*g._readRawStderr)(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (g genericCommandListener) ReadStdoutLine(v string) {
|
||||
if g._readStdoutLine != nil {
|
||||
(*g._readStdoutLine)(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (g genericCommandListener) ReadStderrLine(v string) {
|
||||
if g._readStderrLine != nil {
|
||||
(*g._readStderrLine)(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (g genericCommandListener) Finished(v int) {
|
||||
if g._finished != nil {
|
||||
(*g._finished)(v)
|
||||
}
|
||||
}
|
||||
|
||||
func (g genericCommandListener) Timeout() {
|
||||
if g._timeout != nil {
|
||||
(*g._timeout)()
|
||||
}
|
||||
}
|
146
cmdext/pipereader.go
Normal file
146
cmdext/pipereader.go
Normal file
@@ -0,0 +1,146 @@
|
||||
package cmdext
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/syncext"
|
||||
"io"
|
||||
"sync"
|
||||
)
|
||||
|
||||
type pipeReader struct {
|
||||
stdout io.ReadCloser
|
||||
stderr io.ReadCloser
|
||||
}
|
||||
|
||||
// Read ready stdout and stdin until finished
|
||||
// also splits both pipes into lines and calld the listener
|
||||
func (pr *pipeReader) Read(listener []CommandListener) (string, string, string, error) {
|
||||
type combevt struct {
|
||||
line string
|
||||
stop bool
|
||||
}
|
||||
|
||||
errch := make(chan error, 8)
|
||||
|
||||
wg := sync.WaitGroup{}
|
||||
|
||||
// [1] read raw stdout
|
||||
|
||||
wg.Add(1)
|
||||
stdoutBufferReader, stdoutBufferWriter := io.Pipe()
|
||||
stdout := ""
|
||||
go func() {
|
||||
buf := make([]byte, 128)
|
||||
for true {
|
||||
n, out := pr.stdout.Read(buf)
|
||||
|
||||
if n > 0 {
|
||||
txt := string(buf[:n])
|
||||
stdout += txt
|
||||
_, _ = stdoutBufferWriter.Write(buf[:n])
|
||||
for _, lstr := range listener {
|
||||
lstr.ReadRawStdout(buf[:n])
|
||||
}
|
||||
}
|
||||
if out == io.EOF {
|
||||
break
|
||||
}
|
||||
if out != nil {
|
||||
errch <- out
|
||||
break
|
||||
}
|
||||
}
|
||||
_ = stdoutBufferWriter.Close()
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
// [2] read raw stderr
|
||||
|
||||
wg.Add(1)
|
||||
stderrBufferReader, stderrBufferWriter := io.Pipe()
|
||||
stderr := ""
|
||||
go func() {
|
||||
buf := make([]byte, 128)
|
||||
for true {
|
||||
n, err := pr.stderr.Read(buf)
|
||||
|
||||
if n > 0 {
|
||||
txt := string(buf[:n])
|
||||
stderr += txt
|
||||
_, _ = stderrBufferWriter.Write(buf[:n])
|
||||
for _, lstr := range listener {
|
||||
lstr.ReadRawStderr(buf[:n])
|
||||
}
|
||||
}
|
||||
if err == io.EOF {
|
||||
break
|
||||
}
|
||||
if err != nil {
|
||||
errch <- err
|
||||
break
|
||||
}
|
||||
}
|
||||
_ = stderrBufferWriter.Close()
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
combch := make(chan combevt, 32)
|
||||
|
||||
// [3] collect stdout line-by-line
|
||||
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
scanner := bufio.NewScanner(stdoutBufferReader)
|
||||
for scanner.Scan() {
|
||||
txt := scanner.Text()
|
||||
for _, lstr := range listener {
|
||||
lstr.ReadStdoutLine(txt)
|
||||
}
|
||||
combch <- combevt{txt, false}
|
||||
}
|
||||
combch <- combevt{"", true}
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
// [4] collect stderr line-by-line
|
||||
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
scanner := bufio.NewScanner(stderrBufferReader)
|
||||
for scanner.Scan() {
|
||||
txt := scanner.Text()
|
||||
for _, lstr := range listener {
|
||||
lstr.ReadStderrLine(txt)
|
||||
}
|
||||
combch <- combevt{txt, false}
|
||||
}
|
||||
combch <- combevt{"", true}
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
// [5] combine stdcombined
|
||||
|
||||
wg.Add(1)
|
||||
stdcombined := ""
|
||||
go func() {
|
||||
stopctr := 0
|
||||
for stopctr < 2 {
|
||||
vvv := <-combch
|
||||
if vvv.stop {
|
||||
stopctr++
|
||||
} else {
|
||||
stdcombined += vvv.line + "\n" // this comes from bufio.Scanner and has no newlines...
|
||||
}
|
||||
}
|
||||
wg.Done()
|
||||
}()
|
||||
|
||||
// wait for all (5) goroutines to finish
|
||||
wg.Wait()
|
||||
|
||||
if err, ok := syncext.ReadNonBlocking(errch); ok {
|
||||
return "", "", "", err
|
||||
}
|
||||
|
||||
return stdout, stderr, stdcombined, nil
|
||||
}
|
@@ -14,9 +14,21 @@ import (
|
||||
// ApplyEnvOverrides overrides field values from environment variables
|
||||
//
|
||||
// fields must be tagged with `env:"env_key"`
|
||||
func ApplyEnvOverrides[T any](c *T) error {
|
||||
|
||||
//
|
||||
// only works on exported fields
|
||||
//
|
||||
// fields without an env tag are ignored
|
||||
// fields with an `env:"-"` tag are ignore
|
||||
//
|
||||
// sub-structs are recursively parsed (if they have an env tag) and the env-variable keys are delimited by the delim parameter
|
||||
// sub-structs with `env:""` are also parsed, but the delimited is skipped (they are handled as if they were one level higher)
|
||||
func ApplyEnvOverrides[T any](c *T, delim string) error {
|
||||
rval := reflect.ValueOf(c).Elem()
|
||||
|
||||
return processEnvOverrides(rval, delim, "")
|
||||
}
|
||||
|
||||
func processEnvOverrides(rval reflect.Value, delim string, prefix string) error {
|
||||
rtyp := rval.Type()
|
||||
|
||||
for i := 0; i < rtyp.NumField(); i++ {
|
||||
@@ -24,12 +36,36 @@ func ApplyEnvOverrides[T any](c *T) error {
|
||||
rsfield := rtyp.Field(i)
|
||||
rvfield := rval.Field(i)
|
||||
|
||||
envkey := rsfield.Tag.Get("env")
|
||||
if envkey == "" {
|
||||
if !rsfield.IsExported() {
|
||||
continue
|
||||
}
|
||||
|
||||
envval, efound := os.LookupEnv(envkey)
|
||||
if rvfield.Kind() == reflect.Struct {
|
||||
|
||||
envkey, found := rsfield.Tag.Lookup("env")
|
||||
if !found || envkey == "-" {
|
||||
continue
|
||||
}
|
||||
|
||||
subPrefix := prefix
|
||||
if envkey != "" {
|
||||
subPrefix = subPrefix + envkey + delim
|
||||
}
|
||||
|
||||
err := processEnvOverrides(rvfield, delim, subPrefix)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
envkey := rsfield.Tag.Get("env")
|
||||
if envkey == "" || envkey == "-" {
|
||||
continue
|
||||
}
|
||||
|
||||
fullEnvKey := prefix + envkey
|
||||
|
||||
envval, efound := os.LookupEnv(fullEnvKey)
|
||||
if !efound {
|
||||
continue
|
||||
}
|
||||
@@ -38,86 +74,86 @@ func ApplyEnvOverrides[T any](c *T) error {
|
||||
|
||||
rvfield.Set(reflect.ValueOf(envval))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval)
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(int(0)) {
|
||||
|
||||
envint, err := strconv.ParseInt(envval, 10, bits.UintSize)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(int(envint)))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval)
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(int64(0)) {
|
||||
|
||||
envint, err := strconv.ParseInt(envval, 10, 64)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int64 (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int64 (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(int64(envint)))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval)
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(int32(0)) {
|
||||
|
||||
envint, err := strconv.ParseInt(envval, 10, 32)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(int32(envint)))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval)
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(int8(0)) {
|
||||
|
||||
envint, err := strconv.ParseInt(envval, 10, 8)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to int32 (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(int8(envint)))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, envval)
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, envval)
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(time.Duration(0)) {
|
||||
|
||||
dur, err := timeext.ParseDurationShortString(envval)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to duration (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to duration (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(dur))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, dur.String())
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, dur.String())
|
||||
|
||||
} else if rvfield.Type() == reflect.TypeOf(time.UnixMilli(0)) {
|
||||
|
||||
tim, err := time.Parse(time.RFC3339Nano, envval)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to time.time (value := '%s')", envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to time.time (value := '%s')", fullEnvKey, envval))
|
||||
}
|
||||
|
||||
rvfield.Set(reflect.ValueOf(tim))
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", envkey, tim.String())
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%s'\n", fullEnvKey, tim.String())
|
||||
|
||||
} else if rvfield.Type().ConvertibleTo(reflect.TypeOf(int(0))) {
|
||||
|
||||
envint, err := strconv.ParseInt(envval, 10, 8)
|
||||
if err != nil {
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to <%s, ,int> (value := '%s')", rvfield.Type().Name(), envkey, envval))
|
||||
return errors.New(fmt.Sprintf("Failed to parse env-config variable '%s' to <%s, ,int> (value := '%s')", rvfield.Type().Name(), fullEnvKey, envval))
|
||||
}
|
||||
|
||||
envcvl := reflect.ValueOf(envint).Convert(rvfield.Type())
|
||||
|
||||
rvfield.Set(envcvl)
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", envkey, envcvl.Interface())
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", fullEnvKey, envcvl.Interface())
|
||||
|
||||
} else if rvfield.Type().ConvertibleTo(reflect.TypeOf("")) {
|
||||
|
||||
@@ -125,7 +161,7 @@ func ApplyEnvOverrides[T any](c *T) error {
|
||||
|
||||
rvfield.Set(envcvl)
|
||||
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", envkey, envcvl.Interface())
|
||||
fmt.Printf("[CONF] Overwrite config '%s' with '%v'\n", fullEnvKey, envcvl.Interface())
|
||||
|
||||
} else {
|
||||
return errors.New(fmt.Sprintf("Unknown kind/type in config: [ %s | %s ]", rvfield.Kind().String(), rvfield.Type().String()))
|
||||
|
@@ -1,6 +1,7 @@
|
||||
package confext
|
||||
|
||||
import (
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/timeext"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
@@ -40,7 +41,7 @@ func TestApplyEnvOverridesNoop(t *testing.T) {
|
||||
|
||||
output := input
|
||||
|
||||
err := ApplyEnvOverrides(&output)
|
||||
err := ApplyEnvOverrides(&output, ".")
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
t.FailNow()
|
||||
@@ -92,7 +93,7 @@ func TestApplyEnvOverridesSimple(t *testing.T) {
|
||||
t.Setenv("TEST_V8", "1min4s")
|
||||
t.Setenv("TEST_V9", "2009-11-10T23:00:00Z")
|
||||
|
||||
err := ApplyEnvOverrides(&data)
|
||||
err := ApplyEnvOverrides(&data, ".")
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
t.FailNow()
|
||||
@@ -109,6 +110,109 @@ func TestApplyEnvOverridesSimple(t *testing.T) {
|
||||
assertEqual(t, data.V9, time.Unix(1257894000, 0).UTC())
|
||||
}
|
||||
|
||||
func TestApplyEnvOverridesRecursive(t *testing.T) {
|
||||
|
||||
type subdata struct {
|
||||
V1 int `env:"SUB_V1"`
|
||||
VX string ``
|
||||
V2 string `env:"SUB_V2"`
|
||||
V8 time.Duration `env:"SUB_V3"`
|
||||
V9 time.Time `env:"SUB_V4"`
|
||||
}
|
||||
|
||||
type testdata struct {
|
||||
V1 int `env:"TEST_V1"`
|
||||
VX string ``
|
||||
Sub1 subdata ``
|
||||
Sub2 subdata `env:"TEST_V2"`
|
||||
Sub3 subdata `env:"TEST_V3"`
|
||||
Sub4 subdata `env:""`
|
||||
V5 string `env:"-"`
|
||||
}
|
||||
|
||||
data := testdata{
|
||||
V1: 1,
|
||||
VX: "2",
|
||||
V5: "no",
|
||||
Sub1: subdata{
|
||||
V1: 3,
|
||||
VX: "4",
|
||||
V2: "5",
|
||||
V8: 6 * time.Second,
|
||||
V9: time.Date(2000, 1, 7, 1, 1, 1, 0, time.UTC),
|
||||
},
|
||||
Sub2: subdata{
|
||||
V1: 8,
|
||||
VX: "9",
|
||||
V2: "10",
|
||||
V8: 11 * time.Second,
|
||||
V9: time.Date(2000, 1, 12, 1, 1, 1, 0, timeext.TimezoneBerlin),
|
||||
},
|
||||
Sub3: subdata{
|
||||
V1: 13,
|
||||
VX: "14",
|
||||
V2: "15",
|
||||
V8: 16 * time.Second,
|
||||
V9: time.Date(2000, 1, 17, 1, 1, 1, 0, timeext.TimezoneBerlin),
|
||||
},
|
||||
Sub4: subdata{
|
||||
V1: 18,
|
||||
VX: "19",
|
||||
V2: "20",
|
||||
V8: 21 * time.Second,
|
||||
V9: time.Date(2000, 1, 22, 1, 1, 1, 0, timeext.TimezoneBerlin),
|
||||
},
|
||||
}
|
||||
|
||||
t.Setenv("TEST_V1", "999")
|
||||
t.Setenv("-", "yes")
|
||||
|
||||
t.Setenv("TEST_V2_SUB_V1", "846")
|
||||
t.Setenv("TEST_V2_SUB_V2", "222_hello_world")
|
||||
t.Setenv("TEST_V2_SUB_V3", "1min4s")
|
||||
t.Setenv("TEST_V2_SUB_V4", "2009-11-10T23:00:00Z")
|
||||
|
||||
t.Setenv("TEST_V3_SUB_V1", "33846")
|
||||
t.Setenv("TEST_V3_SUB_V2", "33_hello_world")
|
||||
t.Setenv("TEST_V3_SUB_V3", "33min4s")
|
||||
t.Setenv("TEST_V3_SUB_V4", "2033-11-10T23:00:00Z")
|
||||
|
||||
t.Setenv("SUB_V1", "11")
|
||||
t.Setenv("SUB_V2", "22")
|
||||
t.Setenv("SUB_V3", "33min")
|
||||
t.Setenv("SUB_V4", "2044-01-01T00:00:00Z")
|
||||
|
||||
err := ApplyEnvOverrides(&data, "_")
|
||||
if err != nil {
|
||||
t.Errorf("%v", err)
|
||||
t.FailNow()
|
||||
}
|
||||
|
||||
assertEqual(t, data.V1, 999)
|
||||
assertEqual(t, data.VX, "2")
|
||||
assertEqual(t, data.V5, "no")
|
||||
assertEqual(t, data.Sub1.V1, 3)
|
||||
assertEqual(t, data.Sub1.VX, "4")
|
||||
assertEqual(t, data.Sub1.V2, "5")
|
||||
assertEqual(t, data.Sub1.V8, time.Second*6)
|
||||
assertEqual(t, data.Sub1.V9, time.Unix(947206861, 0).UTC())
|
||||
assertEqual(t, data.Sub2.V1, 846)
|
||||
assertEqual(t, data.Sub2.VX, "9")
|
||||
assertEqual(t, data.Sub2.V2, "222_hello_world")
|
||||
assertEqual(t, data.Sub2.V8, time.Second*64)
|
||||
assertEqual(t, data.Sub2.V9, time.Unix(1257894000, 0).UTC())
|
||||
assertEqual(t, data.Sub3.V1, 33846)
|
||||
assertEqual(t, data.Sub3.VX, "14")
|
||||
assertEqual(t, data.Sub3.V2, "33_hello_world")
|
||||
assertEqual(t, data.Sub3.V8, time.Second*1984)
|
||||
assertEqual(t, data.Sub3.V9, time.Unix(2015276400, 0).UTC())
|
||||
assertEqual(t, data.Sub4.V1, 11)
|
||||
assertEqual(t, data.Sub4.VX, "19")
|
||||
assertEqual(t, data.Sub4.V2, "22")
|
||||
assertEqual(t, data.Sub4.V8, time.Second*1980)
|
||||
assertEqual(t, data.Sub4.V9, time.Unix(2335219200, 0).UTC())
|
||||
}
|
||||
|
||||
func assertEqual[T comparable](t *testing.T, actual T, expected T) {
|
||||
if actual != expected {
|
||||
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||
|
68
cryptext/aes.go
Normal file
68
cryptext/aes.go
Normal file
@@ -0,0 +1,68 @@
|
||||
package cryptext
|
||||
|
||||
import (
|
||||
"crypto/aes"
|
||||
"crypto/cipher"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
"golang.org/x/crypto/scrypt"
|
||||
"io"
|
||||
)
|
||||
|
||||
// https://stackoverflow.com/a/18819040/1761622
|
||||
|
||||
func EncryptAESSimple(password, text []byte) ([]byte, error) {
|
||||
|
||||
key, err := scrypt.Key(password, nil, 32768, 8, 1, 32) // this is not 100% correct, rounds too low and salt is missing
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
block, err := aes.NewCipher(key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
b := base64.StdEncoding.EncodeToString(text)
|
||||
ciphertext := make([]byte, aes.BlockSize+len(b))
|
||||
|
||||
iv := ciphertext[:aes.BlockSize]
|
||||
if _, err := io.ReadFull(rand.Reader, iv); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
cfb := cipher.NewCFBEncrypter(block, iv)
|
||||
cfb.XORKeyStream(ciphertext[aes.BlockSize:], []byte(b))
|
||||
|
||||
return ciphertext, nil
|
||||
}
|
||||
|
||||
func DecryptAESSimple(password, text []byte) ([]byte, error) {
|
||||
|
||||
key, err := scrypt.Key(password, nil, 32768, 8, 1, 32) // this is not 100% correct, rounds too low and salt is missing
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
block, err := aes.NewCipher(key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(text) < aes.BlockSize {
|
||||
return nil, errors.New("ciphertext too short")
|
||||
}
|
||||
|
||||
iv := text[:aes.BlockSize]
|
||||
text = text[aes.BlockSize:]
|
||||
cfb := cipher.NewCFBDecrypter(block, iv)
|
||||
cfb.XORKeyStream(text, text)
|
||||
|
||||
data, err := base64.StdEncoding.DecodeString(string(text))
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
22
cryptext/aes_test.go
Normal file
22
cryptext/aes_test.go
Normal file
@@ -0,0 +1,22 @@
|
||||
package cryptext
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestEncryptAESSimple(t *testing.T) {
|
||||
|
||||
pw := []byte("hunter12")
|
||||
|
||||
str1 := []byte("Hello World")
|
||||
|
||||
str2, err := EncryptAESSimple(pw, str1)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
str3, err := DecryptAESSimple(pw, str2)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
assertEqual(t, string(str1), string(str3))
|
||||
}
|
365
cryptext/passHash.go
Normal file
365
cryptext/passHash.go
Normal file
@@ -0,0 +1,365 @@
|
||||
package cryptext
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"crypto/sha256"
|
||||
"encoding/base64"
|
||||
"encoding/hex"
|
||||
"errors"
|
||||
"fmt"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/totpext"
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const LatestPassHashVersion = 4
|
||||
|
||||
// PassHash
|
||||
// - [v0]: plaintext password ( `0|...` )
|
||||
// - [v1]: sha256(plaintext)
|
||||
// - [v2]: seed | sha256<seed>(plaintext)
|
||||
// - [v3]: seed | sha256<seed>(plaintext) | [hex(totp)]
|
||||
// - [v4]: bcrypt(plaintext) | [hex(totp)]
|
||||
type PassHash string
|
||||
|
||||
func (ph PassHash) Valid() bool {
|
||||
_, _, _, _, _, valid := ph.Data()
|
||||
return valid
|
||||
}
|
||||
|
||||
func (ph PassHash) HasTOTP() bool {
|
||||
_, _, _, otp, _, _ := ph.Data()
|
||||
return otp
|
||||
}
|
||||
|
||||
func (ph PassHash) Data() (_version int, _seed []byte, _payload []byte, _totp bool, _totpsecret []byte, _valid bool) {
|
||||
|
||||
split := strings.Split(string(ph), "|")
|
||||
if len(split) == 0 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
|
||||
version, err := strconv.ParseInt(split[0], 10, 32)
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
|
||||
if version == 0 {
|
||||
if len(split) != 2 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
return int(version), nil, []byte(split[1]), false, nil, true
|
||||
}
|
||||
|
||||
if version == 1 {
|
||||
if len(split) != 2 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
payload, err := base64.RawStdEncoding.DecodeString(split[1])
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
return int(version), nil, payload, false, nil, true
|
||||
}
|
||||
|
||||
//
|
||||
if version == 2 {
|
||||
if len(split) != 3 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
seed, err := base64.RawStdEncoding.DecodeString(split[1])
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
payload, err := base64.RawStdEncoding.DecodeString(split[2])
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
return int(version), seed, payload, false, nil, true
|
||||
}
|
||||
|
||||
if version == 3 {
|
||||
if len(split) != 4 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
seed, err := base64.RawStdEncoding.DecodeString(split[1])
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
payload, err := base64.RawStdEncoding.DecodeString(split[2])
|
||||
if err != nil {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
totp := false
|
||||
totpsecret := make([]byte, 0)
|
||||
if split[3] != "0" {
|
||||
totpsecret, err = hex.DecodeString(split[3])
|
||||
totp = true
|
||||
}
|
||||
return int(version), seed, payload, totp, totpsecret, true
|
||||
}
|
||||
|
||||
if version == 4 {
|
||||
if len(split) != 3 {
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
payload := []byte(split[1])
|
||||
totp := false
|
||||
totpsecret := make([]byte, 0)
|
||||
if split[2] != "0" {
|
||||
totpsecret, err = hex.DecodeString(split[3])
|
||||
totp = true
|
||||
}
|
||||
return int(version), nil, payload, totp, totpsecret, true
|
||||
}
|
||||
|
||||
return -1, nil, nil, false, nil, false
|
||||
}
|
||||
|
||||
func (ph PassHash) Verify(plainpass string, totp *string) bool {
|
||||
version, seed, payload, hastotp, totpsecret, valid := ph.Data()
|
||||
if !valid {
|
||||
return false
|
||||
}
|
||||
|
||||
if hastotp && totp == nil {
|
||||
return false
|
||||
}
|
||||
|
||||
if version == 0 {
|
||||
return langext.ArrEqualsExact([]byte(plainpass), payload)
|
||||
}
|
||||
|
||||
if version == 1 {
|
||||
return langext.ArrEqualsExact(hash256(plainpass), payload)
|
||||
}
|
||||
|
||||
if version == 2 {
|
||||
return langext.ArrEqualsExact(hash256Seeded(plainpass, seed), payload)
|
||||
}
|
||||
|
||||
if version == 3 {
|
||||
if !hastotp {
|
||||
return langext.ArrEqualsExact(hash256Seeded(plainpass, seed), payload)
|
||||
} else {
|
||||
return langext.ArrEqualsExact(hash256Seeded(plainpass, seed), payload) && totpext.Validate(totpsecret, *totp)
|
||||
}
|
||||
}
|
||||
|
||||
if version == 4 {
|
||||
if !hastotp {
|
||||
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) == nil
|
||||
} else {
|
||||
return bcrypt.CompareHashAndPassword(payload, []byte(plainpass)) == nil && totpext.Validate(totpsecret, *totp)
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func (ph PassHash) NeedsPasswordUpgrade() bool {
|
||||
version, _, _, _, _, valid := ph.Data()
|
||||
return valid && version < LatestPassHashVersion
|
||||
}
|
||||
|
||||
func (ph PassHash) Upgrade(plainpass string) (PassHash, error) {
|
||||
version, _, _, hastotp, totpsecret, valid := ph.Data()
|
||||
if !valid {
|
||||
return "", errors.New("invalid password")
|
||||
}
|
||||
if version == LatestPassHashVersion {
|
||||
return ph, nil
|
||||
}
|
||||
if hastotp {
|
||||
return HashPassword(plainpass, totpsecret)
|
||||
} else {
|
||||
return HashPassword(plainpass, nil)
|
||||
}
|
||||
}
|
||||
|
||||
func (ph PassHash) ClearTOTP() (PassHash, error) {
|
||||
version, _, _, _, _, valid := ph.Data()
|
||||
if !valid {
|
||||
return "", errors.New("invalid PassHash")
|
||||
}
|
||||
|
||||
if version == 0 {
|
||||
return ph, nil
|
||||
}
|
||||
|
||||
if version == 1 {
|
||||
return ph, nil
|
||||
}
|
||||
|
||||
if version == 2 {
|
||||
return ph, nil
|
||||
}
|
||||
|
||||
if version == 3 {
|
||||
split := strings.Split(string(ph), "|")
|
||||
split[3] = "0"
|
||||
return PassHash(strings.Join(split, "|")), nil
|
||||
}
|
||||
|
||||
if version == 4 {
|
||||
split := strings.Split(string(ph), "|")
|
||||
split[2] = "0"
|
||||
return PassHash(strings.Join(split, "|")), nil
|
||||
}
|
||||
|
||||
return "", errors.New("unknown version")
|
||||
}
|
||||
|
||||
func (ph PassHash) WithTOTP(totpSecret []byte) (PassHash, error) {
|
||||
version, _, _, _, _, valid := ph.Data()
|
||||
if !valid {
|
||||
return "", errors.New("invalid PassHash")
|
||||
}
|
||||
|
||||
if version == 0 {
|
||||
return "", errors.New("version does not support totp, needs upgrade")
|
||||
}
|
||||
|
||||
if version == 1 {
|
||||
return "", errors.New("version does not support totp, needs upgrade")
|
||||
}
|
||||
|
||||
if version == 2 {
|
||||
return "", errors.New("version does not support totp, needs upgrade")
|
||||
}
|
||||
|
||||
if version == 3 {
|
||||
split := strings.Split(string(ph), "|")
|
||||
split[3] = hex.EncodeToString(totpSecret)
|
||||
return PassHash(strings.Join(split, "|")), nil
|
||||
}
|
||||
|
||||
if version == 4 {
|
||||
split := strings.Split(string(ph), "|")
|
||||
split[2] = hex.EncodeToString(totpSecret)
|
||||
return PassHash(strings.Join(split, "|")), nil
|
||||
}
|
||||
|
||||
return "", errors.New("unknown version")
|
||||
}
|
||||
|
||||
func (ph PassHash) Change(newPlainPass string) (PassHash, error) {
|
||||
version, _, _, hastotp, totpsecret, valid := ph.Data()
|
||||
if !valid {
|
||||
return "", errors.New("invalid PassHash")
|
||||
}
|
||||
|
||||
if version == 0 {
|
||||
return HashPasswordV0(newPlainPass)
|
||||
}
|
||||
|
||||
if version == 1 {
|
||||
return HashPasswordV1(newPlainPass)
|
||||
}
|
||||
|
||||
if version == 2 {
|
||||
return HashPasswordV2(newPlainPass)
|
||||
}
|
||||
|
||||
if version == 3 {
|
||||
return HashPasswordV3(newPlainPass, langext.Conditional(hastotp, totpsecret, nil))
|
||||
}
|
||||
|
||||
if version == 4 {
|
||||
return HashPasswordV4(newPlainPass, langext.Conditional(hastotp, totpsecret, nil))
|
||||
}
|
||||
|
||||
return "", errors.New("unknown version")
|
||||
}
|
||||
|
||||
func (ph PassHash) String() string {
|
||||
return string(ph)
|
||||
}
|
||||
|
||||
func HashPassword(plainpass string, totpSecret []byte) (PassHash, error) {
|
||||
return HashPasswordV4(plainpass, totpSecret)
|
||||
}
|
||||
|
||||
func HashPasswordV4(plainpass string, totpSecret []byte) (PassHash, error) {
|
||||
var strtotp string
|
||||
|
||||
if totpSecret == nil {
|
||||
strtotp = "0"
|
||||
} else {
|
||||
strtotp = hex.EncodeToString(totpSecret)
|
||||
}
|
||||
|
||||
payload, err := bcrypt.GenerateFromPassword([]byte(plainpass), bcrypt.MinCost)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return PassHash(fmt.Sprintf("4|%s|%s", string(payload), strtotp)), nil
|
||||
}
|
||||
|
||||
func HashPasswordV3(plainpass string, totpSecret []byte) (PassHash, error) {
|
||||
var strtotp string
|
||||
|
||||
if totpSecret == nil {
|
||||
strtotp = "0"
|
||||
} else {
|
||||
strtotp = hex.EncodeToString(totpSecret)
|
||||
}
|
||||
|
||||
seed, err := newSeed()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
checksum := hash256Seeded(plainpass, seed)
|
||||
|
||||
return PassHash(fmt.Sprintf("3|%s|%s|%s",
|
||||
base64.RawStdEncoding.EncodeToString(seed),
|
||||
base64.RawStdEncoding.EncodeToString(checksum),
|
||||
strtotp)), nil
|
||||
}
|
||||
|
||||
func HashPasswordV2(plainpass string) (PassHash, error) {
|
||||
seed, err := newSeed()
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
checksum := hash256Seeded(plainpass, seed)
|
||||
|
||||
return PassHash(fmt.Sprintf("2|%s|%s", base64.RawStdEncoding.EncodeToString(seed), base64.RawStdEncoding.EncodeToString(checksum))), nil
|
||||
}
|
||||
|
||||
func HashPasswordV1(plainpass string) (PassHash, error) {
|
||||
return PassHash(fmt.Sprintf("1|%s", base64.RawStdEncoding.EncodeToString(hash256(plainpass)))), nil
|
||||
}
|
||||
|
||||
func HashPasswordV0(plainpass string) (PassHash, error) {
|
||||
return PassHash(fmt.Sprintf("0|%s", plainpass)), nil
|
||||
}
|
||||
|
||||
func hash256(s string) []byte {
|
||||
h := sha256.New()
|
||||
h.Write([]byte(s))
|
||||
bs := h.Sum(nil)
|
||||
return bs
|
||||
}
|
||||
|
||||
func hash256Seeded(s string, seed []byte) []byte {
|
||||
h := sha256.New()
|
||||
h.Write(seed)
|
||||
h.Write([]byte(s))
|
||||
bs := h.Sum(nil)
|
||||
return bs
|
||||
}
|
||||
|
||||
func newSeed() ([]byte, error) {
|
||||
secret := make([]byte, 32)
|
||||
_, err := rand.Read(secret)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return secret, nil
|
||||
}
|
@@ -19,38 +19,38 @@ import (
|
||||
// There are also a bunch of unit tests to ensure that the cache is always in a consistent state
|
||||
//
|
||||
|
||||
type LRUMap[TData any] struct {
|
||||
type LRUMap[TKey comparable, TData any] struct {
|
||||
maxsize int
|
||||
lock sync.Mutex
|
||||
|
||||
cache map[string]*cacheNode[TData]
|
||||
cache map[TKey]*cacheNode[TKey, TData]
|
||||
|
||||
lfuHead *cacheNode[TData]
|
||||
lfuTail *cacheNode[TData]
|
||||
lfuHead *cacheNode[TKey, TData]
|
||||
lfuTail *cacheNode[TKey, TData]
|
||||
}
|
||||
|
||||
type cacheNode[TData any] struct {
|
||||
key string
|
||||
type cacheNode[TKey comparable, TData any] struct {
|
||||
key TKey
|
||||
data TData
|
||||
parent *cacheNode[TData]
|
||||
child *cacheNode[TData]
|
||||
parent *cacheNode[TKey, TData]
|
||||
child *cacheNode[TKey, TData]
|
||||
}
|
||||
|
||||
func NewLRUMap[TData any](size int) *LRUMap[TData] {
|
||||
func NewLRUMap[TKey comparable, TData any](size int) *LRUMap[TKey, TData] {
|
||||
if size <= 2 && size != 0 {
|
||||
panic("Size must be > 2 (or 0)")
|
||||
}
|
||||
|
||||
return &LRUMap[TData]{
|
||||
return &LRUMap[TKey, TData]{
|
||||
maxsize: size,
|
||||
lock: sync.Mutex{},
|
||||
cache: make(map[string]*cacheNode[TData], size+1),
|
||||
cache: make(map[TKey]*cacheNode[TKey, TData], size+1),
|
||||
lfuHead: nil,
|
||||
lfuTail: nil,
|
||||
}
|
||||
}
|
||||
|
||||
func (c *LRUMap[TData]) Put(key string, value TData) {
|
||||
func (c *LRUMap[TKey, TData]) Put(key TKey, value TData) {
|
||||
if c.maxsize == 0 {
|
||||
return // cache disabled
|
||||
}
|
||||
@@ -68,7 +68,7 @@ func (c *LRUMap[TData]) Put(key string, value TData) {
|
||||
}
|
||||
|
||||
// key does not exist: insert into map and add to top of LFU
|
||||
node = &cacheNode[TData]{
|
||||
node = &cacheNode[TKey, TData]{
|
||||
key: key,
|
||||
data: value,
|
||||
parent: nil,
|
||||
@@ -93,7 +93,7 @@ func (c *LRUMap[TData]) Put(key string, value TData) {
|
||||
}
|
||||
}
|
||||
|
||||
func (c *LRUMap[TData]) TryGet(key string) (TData, bool) {
|
||||
func (c *LRUMap[TKey, TData]) TryGet(key TKey) (TData, bool) {
|
||||
if c.maxsize == 0 {
|
||||
return *new(TData), false // cache disabled
|
||||
}
|
||||
@@ -109,7 +109,7 @@ func (c *LRUMap[TData]) TryGet(key string) (TData, bool) {
|
||||
return val.data, ok
|
||||
}
|
||||
|
||||
func (c *LRUMap[TData]) moveNodeToTop(node *cacheNode[TData]) {
|
||||
func (c *LRUMap[TKey, TData]) moveNodeToTop(node *cacheNode[TKey, TData]) {
|
||||
// (only called in critical section !)
|
||||
|
||||
if c.lfuHead == node { // fast case
|
||||
@@ -142,7 +142,7 @@ func (c *LRUMap[TData]) moveNodeToTop(node *cacheNode[TData]) {
|
||||
}
|
||||
}
|
||||
|
||||
func (c *LRUMap[TData]) Size() int {
|
||||
func (c *LRUMap[TKey, TData]) Size() int {
|
||||
c.lock.Lock()
|
||||
defer c.lock.Unlock()
|
||||
return len(c.cache)
|
||||
|
116
dataext/stack.go
Normal file
116
dataext/stack.go
Normal file
@@ -0,0 +1,116 @@
|
||||
package dataext
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
"sync"
|
||||
)
|
||||
|
||||
var ErrEmptyStack = errors.New("stack is empty")
|
||||
|
||||
type Stack[T any] struct {
|
||||
lock *sync.Mutex
|
||||
data []T
|
||||
}
|
||||
|
||||
func NewStack[T any](threadsafe bool, initialCapacity int) *Stack[T] {
|
||||
var lck *sync.Mutex = nil
|
||||
if threadsafe {
|
||||
lck = &sync.Mutex{}
|
||||
}
|
||||
return &Stack[T]{
|
||||
lock: lck,
|
||||
data: make([]T, 0, initialCapacity),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Stack[T]) Push(v T) {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
s.data = append(s.data, v)
|
||||
}
|
||||
|
||||
func (s *Stack[T]) Pop() (T, error) {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
l := len(s.data)
|
||||
if l == 0 {
|
||||
return *new(T), ErrEmptyStack
|
||||
}
|
||||
|
||||
result := s.data[l-1]
|
||||
s.data = s.data[:l-1]
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (s *Stack[T]) OptPop() *T {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
l := len(s.data)
|
||||
if l == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
result := s.data[l-1]
|
||||
s.data = s.data[:l-1]
|
||||
|
||||
return langext.Ptr(result)
|
||||
}
|
||||
|
||||
func (s *Stack[T]) Peek() (T, error) {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
l := len(s.data)
|
||||
|
||||
if l == 0 {
|
||||
return *new(T), ErrEmptyStack
|
||||
}
|
||||
|
||||
return s.data[l-1], nil
|
||||
}
|
||||
|
||||
func (s *Stack[T]) OptPeek() *T {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
l := len(s.data)
|
||||
|
||||
if l == 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
return langext.Ptr(s.data[l-1])
|
||||
}
|
||||
|
||||
func (s *Stack[T]) Length() int {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
return len(s.data)
|
||||
}
|
||||
|
||||
func (s *Stack[T]) Empty() bool {
|
||||
if s.lock != nil {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
}
|
||||
|
||||
return len(s.data) == 0
|
||||
}
|
9
go.mod
9
go.mod
@@ -3,8 +3,11 @@ module gogs.mikescher.com/BlackForestBytes/goext
|
||||
go 1.19
|
||||
|
||||
require (
|
||||
golang.org/x/sys v0.1.0
|
||||
golang.org/x/term v0.1.0
|
||||
golang.org/x/sys v0.3.0
|
||||
golang.org/x/term v0.3.0
|
||||
)
|
||||
|
||||
require github.com/jmoiron/sqlx v1.3.5 // indirect
|
||||
require (
|
||||
github.com/jmoiron/sqlx v1.3.5 // indirect
|
||||
golang.org/x/crypto v0.4.0 // indirect
|
||||
)
|
||||
|
6
go.sum
6
go.sum
@@ -3,7 +3,13 @@ github.com/jmoiron/sqlx v1.3.5 h1:vFFPA71p1o5gAeqtEAwLU4dnX2napprKtHr7PYIcN3g=
|
||||
github.com/jmoiron/sqlx v1.3.5/go.mod h1:nRVWtLre0KfCLJvgxzCsLVMogSvQ1zNJtpYr2Ccp0mQ=
|
||||
github.com/lib/pq v1.2.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=
|
||||
github.com/mattn/go-sqlite3 v1.14.6/go.mod h1:NyWgC/yNuGj7Q9rpYnZvas74GogHl5/Z4A/KQRfk6bU=
|
||||
golang.org/x/crypto v0.4.0 h1:UVQgzMY87xqpKNgb+kDsll2Igd33HszWHFLmpaRMq/8=
|
||||
golang.org/x/crypto v0.4.0/go.mod h1:3quD/ATkf6oY+rnes5c3ExXTbLc8mueNue5/DoinL80=
|
||||
golang.org/x/sys v0.1.0 h1:kunALQeHf1/185U1i0GOB/fy1IPRDDpuoOOqRReG57U=
|
||||
golang.org/x/sys v0.1.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.3.0 h1:w8ZOecv6NaNa/zC8944JTU3vz4u6Lagfk4RPQxv92NQ=
|
||||
golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/term v0.1.0 h1:g6Z6vPFA9dYBAF7DWcH6sCcOntplXsDKcliusYijMlw=
|
||||
golang.org/x/term v0.1.0/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/term v0.3.0 h1:qoo4akIqOcDME5bhc/NgxUdovd6BSS2uMsVjB56q1xI=
|
||||
golang.org/x/term v0.3.0/go.mod h1:q750SLmJuPmVoN1blW3UFBPREJfb1KmY3vwxfr+nFDA=
|
||||
|
@@ -265,6 +265,16 @@ func ArrMap[T1 any, T2 any](arr []T1, conv func(v T1) T2) []T2 {
|
||||
return r
|
||||
}
|
||||
|
||||
func ArrFilterMap[T1 any, T2 any](arr []T1, filter func(v T1) bool, conv func(v T1) T2) []T2 {
|
||||
r := make([]T2, 0, len(arr))
|
||||
for _, v := range arr {
|
||||
if filter(v) {
|
||||
r = append(r, conv(v))
|
||||
}
|
||||
}
|
||||
return r
|
||||
}
|
||||
|
||||
func ArrSum[T NumberConstraint](arr []T) T {
|
||||
var r T = 0
|
||||
for _, v := range arr {
|
||||
|
@@ -15,3 +15,35 @@ func Conditional[T any](v bool, resTrue T, resFalse T) T {
|
||||
return resFalse
|
||||
}
|
||||
}
|
||||
|
||||
func ConditionalFn00[T any](v bool, resTrue T, resFalse T) T {
|
||||
if v {
|
||||
return resTrue
|
||||
} else {
|
||||
return resFalse
|
||||
}
|
||||
}
|
||||
|
||||
func ConditionalFn10[T any](v bool, resTrue func() T, resFalse T) T {
|
||||
if v {
|
||||
return resTrue()
|
||||
} else {
|
||||
return resFalse
|
||||
}
|
||||
}
|
||||
|
||||
func ConditionalFn01[T any](v bool, resTrue T, resFalse func() T) T {
|
||||
if v {
|
||||
return resTrue
|
||||
} else {
|
||||
return resFalse()
|
||||
}
|
||||
}
|
||||
|
||||
func ConditionalFn11[T any](v bool, resTrue func() T, resFalse func() T) T {
|
||||
if v {
|
||||
return resTrue()
|
||||
} else {
|
||||
return resFalse()
|
||||
}
|
||||
}
|
||||
|
@@ -7,3 +7,11 @@ func MapKeyArr[T comparable, V any](v map[T]V) []T {
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func ArrToMap[T comparable, V any](a []V, keyfunc func(V) T) map[T]V {
|
||||
result := make(map[T]V, len(a))
|
||||
for _, v := range a {
|
||||
result[keyfunc(v)] = v
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
@@ -34,3 +34,7 @@ func IsNil(i interface{}) bool {
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func PtrEquals[T comparable](v1 *T, v2 *T) bool {
|
||||
return (v1 == nil && v2 == nil) || (v1 != nil && v2 != nil && *v1 == *v2)
|
||||
}
|
||||
|
@@ -37,3 +37,21 @@ func IsSliceSorted[T any](arr []T, less func(v1, v2 T) bool) bool {
|
||||
return less(arr[i1], arr[i2])
|
||||
})
|
||||
}
|
||||
|
||||
func SortBy[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||
sort.Slice(arr, func(i1, i2 int) bool {
|
||||
return selector(arr[i1]) < selector(arr[i2])
|
||||
})
|
||||
}
|
||||
|
||||
func SortByStable[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||
sort.SliceStable(arr, func(i1, i2 int) bool {
|
||||
return selector(arr[i1]) < selector(arr[i2])
|
||||
})
|
||||
}
|
||||
|
||||
func IsSortedBy[TElem any, TSel OrderedConstraint](arr []TElem, selector func(v TElem) TSel) {
|
||||
sort.SliceStable(arr, func(i1, i2 int) bool {
|
||||
return selector(arr[i1]) < selector(arr[i2])
|
||||
})
|
||||
}
|
||||
|
130
rext/wrapper.go
Normal file
130
rext/wrapper.go
Normal file
@@ -0,0 +1,130 @@
|
||||
package rext
|
||||
|
||||
import "regexp"
|
||||
|
||||
type Regex interface {
|
||||
IsMatch(haystack string) bool
|
||||
MatchFirst(haystack string) (RegexMatch, bool)
|
||||
MatchAll(haystack string) []RegexMatch
|
||||
ReplaceAll(haystack string, repl string, literal bool) string
|
||||
ReplaceAllFunc(haystack string, repl func(string) string) string
|
||||
RemoveAll(haystack string) string
|
||||
GroupCount() int
|
||||
}
|
||||
|
||||
type regexWrapper struct {
|
||||
rex *regexp.Regexp
|
||||
subnames []string
|
||||
}
|
||||
|
||||
type RegexMatch struct {
|
||||
haystack string
|
||||
submatchesIndex []int
|
||||
subnames []string
|
||||
}
|
||||
|
||||
type RegexMatchGroup struct {
|
||||
haystack string
|
||||
start int
|
||||
end int
|
||||
}
|
||||
|
||||
func W(rex *regexp.Regexp) Regex {
|
||||
return ®exWrapper{rex: rex, subnames: rex.SubexpNames()}
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------------------------------------------------
|
||||
|
||||
func (w *regexWrapper) IsMatch(haystack string) bool {
|
||||
return w.rex.MatchString(haystack)
|
||||
}
|
||||
|
||||
func (w *regexWrapper) MatchFirst(haystack string) (RegexMatch, bool) {
|
||||
res := w.rex.FindStringSubmatchIndex(haystack)
|
||||
if res == nil {
|
||||
return RegexMatch{}, false
|
||||
}
|
||||
|
||||
return RegexMatch{haystack: haystack, submatchesIndex: res, subnames: w.subnames}, true
|
||||
}
|
||||
|
||||
func (w *regexWrapper) MatchAll(haystack string) []RegexMatch {
|
||||
resarr := w.rex.FindAllStringSubmatchIndex(haystack, -1)
|
||||
|
||||
matches := make([]RegexMatch, 0, len(resarr))
|
||||
for _, res := range resarr {
|
||||
matches = append(matches, RegexMatch{haystack: haystack, submatchesIndex: res, subnames: w.subnames})
|
||||
}
|
||||
|
||||
return matches
|
||||
}
|
||||
|
||||
func (w *regexWrapper) ReplaceAll(haystack string, repl string, literal bool) string {
|
||||
if literal {
|
||||
// do not expand placeholder aka $1, $2, ...
|
||||
return w.rex.ReplaceAllLiteralString(haystack, repl)
|
||||
} else {
|
||||
return w.rex.ReplaceAllString(haystack, repl)
|
||||
}
|
||||
}
|
||||
|
||||
func (w *regexWrapper) ReplaceAllFunc(haystack string, repl func(string) string) string {
|
||||
return w.rex.ReplaceAllStringFunc(haystack, repl)
|
||||
}
|
||||
|
||||
func (w *regexWrapper) RemoveAll(haystack string) string {
|
||||
return w.rex.ReplaceAllLiteralString(haystack, "")
|
||||
}
|
||||
|
||||
// GroupCount returns the amount of groups in this match, does not count group-0 (whole match)
|
||||
func (w *regexWrapper) GroupCount() int {
|
||||
return len(w.subnames) - 1
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------------------------------------------------
|
||||
|
||||
func (m RegexMatch) FullMatch() RegexMatchGroup {
|
||||
return RegexMatchGroup{haystack: m.haystack, start: m.submatchesIndex[0], end: m.submatchesIndex[1]}
|
||||
}
|
||||
|
||||
// GroupCount returns the amount of groups in this match, does not count group-0 (whole match)
|
||||
func (m RegexMatch) GroupCount() int {
|
||||
return len(m.subnames) - 1
|
||||
}
|
||||
|
||||
// GroupByIndex returns the value of a matched group (group 0 == whole match)
|
||||
func (m RegexMatch) GroupByIndex(idx int) RegexMatchGroup {
|
||||
return RegexMatchGroup{haystack: m.haystack, start: m.submatchesIndex[idx*2], end: m.submatchesIndex[idx*2+1]}
|
||||
}
|
||||
|
||||
// GroupByName returns the value of a matched group (group 0 == whole match)
|
||||
func (m RegexMatch) GroupByName(name string) RegexMatchGroup {
|
||||
for idx, subname := range m.subnames {
|
||||
if subname == name {
|
||||
return RegexMatchGroup{haystack: m.haystack, start: m.submatchesIndex[idx*2], end: m.submatchesIndex[idx*2+1]}
|
||||
}
|
||||
}
|
||||
panic("failed to find regex-group by name")
|
||||
}
|
||||
|
||||
// ---------------------------------------------------------------------------------------------------------------------
|
||||
|
||||
func (g RegexMatchGroup) Value() string {
|
||||
return g.haystack[g.start:g.end]
|
||||
}
|
||||
|
||||
func (g RegexMatchGroup) Start() int {
|
||||
return g.start
|
||||
}
|
||||
|
||||
func (g RegexMatchGroup) End() int {
|
||||
return g.end
|
||||
}
|
||||
|
||||
func (g RegexMatchGroup) Range() (int, int) {
|
||||
return g.start, g.end
|
||||
}
|
||||
|
||||
func (g RegexMatchGroup) Length() int {
|
||||
return g.end - g.start
|
||||
}
|
45
rfctime/interface.go
Normal file
45
rfctime/interface.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package rfctime
|
||||
|
||||
import "time"
|
||||
|
||||
type RFCTime interface {
|
||||
Time() time.Time
|
||||
Serialize() string
|
||||
|
||||
UnmarshalJSON(bytes []byte) error
|
||||
MarshalJSON() ([]byte, error)
|
||||
|
||||
MarshalBinary() ([]byte, error)
|
||||
UnmarshalBinary(data []byte) error
|
||||
|
||||
GobEncode() ([]byte, error)
|
||||
GobDecode(data []byte) error
|
||||
|
||||
MarshalText() ([]byte, error)
|
||||
UnmarshalText(data []byte) error
|
||||
|
||||
After(u RFCTime) bool
|
||||
Before(u RFCTime) bool
|
||||
Equal(u RFCTime) bool
|
||||
IsZero() bool
|
||||
Date() (year int, month time.Month, day int)
|
||||
Year() int
|
||||
Month() time.Month
|
||||
Day() int
|
||||
Weekday() time.Weekday
|
||||
ISOWeek() (year, week int)
|
||||
Clock() (hour, min, sec int)
|
||||
Hour() int
|
||||
Minute() int
|
||||
Second() int
|
||||
Nanosecond() int
|
||||
YearDay() int
|
||||
Sub(u RFCTime) time.Duration
|
||||
Unix() int64
|
||||
UnixMilli() int64
|
||||
UnixMicro() int64
|
||||
UnixNano() int64
|
||||
Format(layout string) string
|
||||
GoString() string
|
||||
String() string
|
||||
}
|
182
rfctime/rfc3339.go
Normal file
182
rfctime/rfc3339.go
Normal file
@@ -0,0 +1,182 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
)
|
||||
|
||||
type RFC3339Time time.Time
|
||||
|
||||
func (t RFC3339Time) Time() time.Time {
|
||||
return time.Time(t)
|
||||
}
|
||||
|
||||
func (t RFC3339Time) MarshalBinary() ([]byte, error) {
|
||||
return (time.Time)(t).MarshalBinary()
|
||||
}
|
||||
|
||||
func (t *RFC3339Time) UnmarshalBinary(data []byte) error {
|
||||
return (*time.Time)(t).UnmarshalBinary(data)
|
||||
}
|
||||
|
||||
func (t RFC3339Time) GobEncode() ([]byte, error) {
|
||||
return (time.Time)(t).GobEncode()
|
||||
}
|
||||
|
||||
func (t *RFC3339Time) GobDecode(data []byte) error {
|
||||
return (*time.Time)(t).GobDecode(data)
|
||||
}
|
||||
|
||||
func (t *RFC3339Time) UnmarshalJSON(data []byte) error {
|
||||
str := ""
|
||||
if err := json.Unmarshal(data, &str); err != nil {
|
||||
return err
|
||||
}
|
||||
t0, err := time.Parse(t.FormatStr(), str)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = RFC3339Time(t0)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t RFC3339Time) MarshalJSON() ([]byte, error) {
|
||||
str := t.Time().Format(t.FormatStr())
|
||||
return json.Marshal(str)
|
||||
}
|
||||
|
||||
func (t RFC3339Time) MarshalText() ([]byte, error) {
|
||||
b := make([]byte, 0, len(t.FormatStr()))
|
||||
return t.Time().AppendFormat(b, t.FormatStr()), nil
|
||||
}
|
||||
|
||||
func (t *RFC3339Time) UnmarshalText(data []byte) error {
|
||||
var err error
|
||||
v, err := time.Parse(t.FormatStr(), string(data))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tt := RFC3339Time(v)
|
||||
*t = tt
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Serialize() string {
|
||||
return t.Time().Format(t.FormatStr())
|
||||
}
|
||||
|
||||
func (t RFC3339Time) FormatStr() string {
|
||||
return time.RFC3339
|
||||
}
|
||||
|
||||
func (t RFC3339Time) After(u RFCTime) bool {
|
||||
return t.Time().After(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Before(u RFCTime) bool {
|
||||
return t.Time().Before(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Equal(u RFCTime) bool {
|
||||
return t.Time().Equal(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339Time) IsZero() bool {
|
||||
return t.Time().IsZero()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Date() (year int, month time.Month, day int) {
|
||||
return t.Time().Date()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Year() int {
|
||||
return t.Time().Year()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Month() time.Month {
|
||||
return t.Time().Month()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Day() int {
|
||||
return t.Time().Day()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Weekday() time.Weekday {
|
||||
return t.Time().Weekday()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) ISOWeek() (year, week int) {
|
||||
return t.Time().ISOWeek()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Clock() (hour, min, sec int) {
|
||||
return t.Time().Clock()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Hour() int {
|
||||
return t.Time().Hour()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Minute() int {
|
||||
return t.Time().Minute()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Second() int {
|
||||
return t.Time().Second()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Nanosecond() int {
|
||||
return t.Time().Nanosecond()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) YearDay() int {
|
||||
return t.Time().YearDay()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Add(d time.Duration) RFC3339Time {
|
||||
return RFC3339Time(t.Time().Add(d))
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Sub(u RFCTime) time.Duration {
|
||||
return t.Time().Sub(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339Time) AddDate(years int, months int, days int) RFC3339Time {
|
||||
return RFC3339Time(t.Time().AddDate(years, months, days))
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Unix() int64 {
|
||||
return t.Time().Unix()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) UnixMilli() int64 {
|
||||
return t.Time().UnixMilli()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) UnixMicro() int64 {
|
||||
return t.Time().UnixMicro()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) UnixNano() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) Format(layout string) string {
|
||||
return t.Time().Format(layout)
|
||||
}
|
||||
|
||||
func (t RFC3339Time) GoString() string {
|
||||
return t.Time().GoString()
|
||||
}
|
||||
|
||||
func (t RFC3339Time) String() string {
|
||||
return t.Time().String()
|
||||
}
|
||||
|
||||
func NewRFC3339(t time.Time) RFC3339Time {
|
||||
return RFC3339Time(t)
|
||||
}
|
||||
|
||||
func NowRFC3339() RFC3339Time {
|
||||
return RFC3339Time(time.Now())
|
||||
}
|
182
rfctime/rfc3339Nano.go
Normal file
182
rfctime/rfc3339Nano.go
Normal file
@@ -0,0 +1,182 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"time"
|
||||
)
|
||||
|
||||
type RFC3339NanoTime time.Time
|
||||
|
||||
func (t RFC3339NanoTime) Time() time.Time {
|
||||
return time.Time(t)
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) MarshalBinary() ([]byte, error) {
|
||||
return (time.Time)(t).MarshalBinary()
|
||||
}
|
||||
|
||||
func (t *RFC3339NanoTime) UnmarshalBinary(data []byte) error {
|
||||
return (*time.Time)(t).UnmarshalBinary(data)
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) GobEncode() ([]byte, error) {
|
||||
return (time.Time)(t).GobEncode()
|
||||
}
|
||||
|
||||
func (t *RFC3339NanoTime) GobDecode(data []byte) error {
|
||||
return (*time.Time)(t).GobDecode(data)
|
||||
}
|
||||
|
||||
func (t *RFC3339NanoTime) UnmarshalJSON(data []byte) error {
|
||||
str := ""
|
||||
if err := json.Unmarshal(data, &str); err != nil {
|
||||
return err
|
||||
}
|
||||
t0, err := time.Parse(t.FormatStr(), str)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = RFC3339NanoTime(t0)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) MarshalJSON() ([]byte, error) {
|
||||
str := t.Time().Format(t.FormatStr())
|
||||
return json.Marshal(str)
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) MarshalText() ([]byte, error) {
|
||||
b := make([]byte, 0, len(t.FormatStr()))
|
||||
return t.Time().AppendFormat(b, t.FormatStr()), nil
|
||||
}
|
||||
|
||||
func (t *RFC3339NanoTime) UnmarshalText(data []byte) error {
|
||||
var err error
|
||||
v, err := time.Parse(t.FormatStr(), string(data))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
tt := RFC3339NanoTime(v)
|
||||
*t = tt
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Serialize() string {
|
||||
return t.Time().Format(t.FormatStr())
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) FormatStr() string {
|
||||
return time.RFC3339Nano
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) After(u RFCTime) bool {
|
||||
return t.Time().After(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Before(u RFCTime) bool {
|
||||
return t.Time().Before(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Equal(u RFCTime) bool {
|
||||
return t.Time().Equal(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) IsZero() bool {
|
||||
return t.Time().IsZero()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Date() (year int, month time.Month, day int) {
|
||||
return t.Time().Date()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Year() int {
|
||||
return t.Time().Year()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Month() time.Month {
|
||||
return t.Time().Month()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Day() int {
|
||||
return t.Time().Day()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Weekday() time.Weekday {
|
||||
return t.Time().Weekday()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) ISOWeek() (year, week int) {
|
||||
return t.Time().ISOWeek()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Clock() (hour, min, sec int) {
|
||||
return t.Time().Clock()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Hour() int {
|
||||
return t.Time().Hour()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Minute() int {
|
||||
return t.Time().Minute()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Second() int {
|
||||
return t.Time().Second()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Nanosecond() int {
|
||||
return t.Time().Nanosecond()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) YearDay() int {
|
||||
return t.Time().YearDay()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Add(d time.Duration) RFC3339NanoTime {
|
||||
return RFC3339NanoTime(t.Time().Add(d))
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Sub(u RFCTime) time.Duration {
|
||||
return t.Time().Sub(u.Time())
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) AddDate(years int, months int, days int) RFC3339NanoTime {
|
||||
return RFC3339NanoTime(t.Time().AddDate(years, months, days))
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Unix() int64 {
|
||||
return t.Time().Unix()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) UnixMilli() int64 {
|
||||
return t.Time().UnixMilli()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) UnixMicro() int64 {
|
||||
return t.Time().UnixMicro()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) UnixNano() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) Format(layout string) string {
|
||||
return t.Time().Format(layout)
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) GoString() string {
|
||||
return t.Time().GoString()
|
||||
}
|
||||
|
||||
func (t RFC3339NanoTime) String() string {
|
||||
return t.Time().String()
|
||||
}
|
||||
|
||||
func NewRFC3339Nano(t time.Time) RFC3339NanoTime {
|
||||
return RFC3339NanoTime(t)
|
||||
}
|
||||
|
||||
func NowRFC3339Nano() RFC3339NanoTime {
|
||||
return RFC3339NanoTime(time.Now())
|
||||
}
|
51
rfctime/rfc3339Nano_test.go
Normal file
51
rfctime/rfc3339Nano_test.go
Normal file
@@ -0,0 +1,51 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestRoundtrip(t *testing.T) {
|
||||
|
||||
type Wrap struct {
|
||||
Value RFC3339NanoTime `json:"v"`
|
||||
}
|
||||
|
||||
val1 := NewRFC3339Nano(time.Now())
|
||||
w1 := Wrap{val1}
|
||||
|
||||
jstr1, err := json.Marshal(w1)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
if string(jstr1) != "{\"v\":\"2023-01-29T20:32:36.149692117+01:00\"}" {
|
||||
t.Errorf("repr differs")
|
||||
}
|
||||
|
||||
w2 := Wrap{}
|
||||
|
||||
err = json.Unmarshal(jstr1, &w2)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
jstr2, err := json.Marshal(w2)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
assertEqual(t, string(jstr1), string(jstr2))
|
||||
|
||||
if !w1.Value.Equal(&w2.Value) {
|
||||
t.Errorf("time differs")
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func assertEqual(t *testing.T, actual string, expected string) {
|
||||
if actual != expected {
|
||||
t.Errorf("values differ: Actual: '%v', Expected: '%v'", actual, expected)
|
||||
}
|
||||
}
|
176
rfctime/unix.go
Normal file
176
rfctime/unix.go
Normal file
@@ -0,0 +1,176 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type UnixTime time.Time
|
||||
|
||||
func (t UnixTime) Time() time.Time {
|
||||
return time.Time(t)
|
||||
}
|
||||
|
||||
func (t UnixTime) MarshalBinary() ([]byte, error) {
|
||||
return (time.Time)(t).MarshalBinary()
|
||||
}
|
||||
|
||||
func (t *UnixTime) UnmarshalBinary(data []byte) error {
|
||||
return (*time.Time)(t).UnmarshalBinary(data)
|
||||
}
|
||||
|
||||
func (t UnixTime) GobEncode() ([]byte, error) {
|
||||
return (time.Time)(t).GobEncode()
|
||||
}
|
||||
|
||||
func (t *UnixTime) GobDecode(data []byte) error {
|
||||
return (*time.Time)(t).GobDecode(data)
|
||||
}
|
||||
|
||||
func (t *UnixTime) UnmarshalJSON(data []byte) error {
|
||||
str := ""
|
||||
if err := json.Unmarshal(data, &str); err != nil {
|
||||
return err
|
||||
}
|
||||
t0, err := strconv.ParseInt(str, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixTime(time.Unix(t0, 0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixTime) MarshalJSON() ([]byte, error) {
|
||||
str := strconv.FormatInt(t.Time().Unix(), 10)
|
||||
return json.Marshal(str)
|
||||
}
|
||||
|
||||
func (t UnixTime) MarshalText() ([]byte, error) {
|
||||
return []byte(strconv.FormatInt(t.Time().Unix(), 10)), nil
|
||||
}
|
||||
|
||||
func (t *UnixTime) UnmarshalText(data []byte) error {
|
||||
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixTime(time.Unix(t0, 0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixTime) Serialize() string {
|
||||
return strconv.FormatInt(t.Time().Unix(), 10)
|
||||
}
|
||||
|
||||
func (t UnixTime) After(u RFCTime) bool {
|
||||
return t.Time().After(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixTime) Before(u RFCTime) bool {
|
||||
return t.Time().Before(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixTime) Equal(u RFCTime) bool {
|
||||
return t.Time().Equal(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixTime) IsZero() bool {
|
||||
return t.Time().IsZero()
|
||||
}
|
||||
|
||||
func (t UnixTime) Date() (year int, month time.Month, day int) {
|
||||
return t.Time().Date()
|
||||
}
|
||||
|
||||
func (t UnixTime) Year() int {
|
||||
return t.Time().Year()
|
||||
}
|
||||
|
||||
func (t UnixTime) Month() time.Month {
|
||||
return t.Time().Month()
|
||||
}
|
||||
|
||||
func (t UnixTime) Day() int {
|
||||
return t.Time().Day()
|
||||
}
|
||||
|
||||
func (t UnixTime) Weekday() time.Weekday {
|
||||
return t.Time().Weekday()
|
||||
}
|
||||
|
||||
func (t UnixTime) ISOWeek() (year, week int) {
|
||||
return t.Time().ISOWeek()
|
||||
}
|
||||
|
||||
func (t UnixTime) Clock() (hour, min, sec int) {
|
||||
return t.Time().Clock()
|
||||
}
|
||||
|
||||
func (t UnixTime) Hour() int {
|
||||
return t.Time().Hour()
|
||||
}
|
||||
|
||||
func (t UnixTime) Minute() int {
|
||||
return t.Time().Minute()
|
||||
}
|
||||
|
||||
func (t UnixTime) Second() int {
|
||||
return t.Time().Second()
|
||||
}
|
||||
|
||||
func (t UnixTime) Nanosecond() int {
|
||||
return t.Time().Nanosecond()
|
||||
}
|
||||
|
||||
func (t UnixTime) YearDay() int {
|
||||
return t.Time().YearDay()
|
||||
}
|
||||
|
||||
func (t UnixTime) Add(d time.Duration) UnixTime {
|
||||
return UnixTime(t.Time().Add(d))
|
||||
}
|
||||
|
||||
func (t UnixTime) Sub(u RFCTime) time.Duration {
|
||||
return t.Time().Sub(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixTime) AddDate(years int, months int, days int) UnixTime {
|
||||
return UnixTime(t.Time().AddDate(years, months, days))
|
||||
}
|
||||
|
||||
func (t UnixTime) Unix() int64 {
|
||||
return t.Time().Unix()
|
||||
}
|
||||
|
||||
func (t UnixTime) UnixMilli() int64 {
|
||||
return t.Time().UnixMilli()
|
||||
}
|
||||
|
||||
func (t UnixTime) UnixMicro() int64 {
|
||||
return t.Time().UnixMicro()
|
||||
}
|
||||
|
||||
func (t UnixTime) UnixNano() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t UnixTime) Format(layout string) string {
|
||||
return t.Time().Format(layout)
|
||||
}
|
||||
|
||||
func (t UnixTime) GoString() string {
|
||||
return t.Time().GoString()
|
||||
}
|
||||
|
||||
func (t UnixTime) String() string {
|
||||
return t.Time().String()
|
||||
}
|
||||
|
||||
func NewUnix(t time.Time) UnixTime {
|
||||
return UnixTime(t)
|
||||
}
|
||||
|
||||
func NowUnix() UnixTime {
|
||||
return UnixTime(time.Now())
|
||||
}
|
176
rfctime/unixMilli.go
Normal file
176
rfctime/unixMilli.go
Normal file
@@ -0,0 +1,176 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type UnixMilliTime time.Time
|
||||
|
||||
func (t UnixMilliTime) Time() time.Time {
|
||||
return time.Time(t)
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) MarshalBinary() ([]byte, error) {
|
||||
return (time.Time)(t).MarshalBinary()
|
||||
}
|
||||
|
||||
func (t *UnixMilliTime) UnmarshalBinary(data []byte) error {
|
||||
return (*time.Time)(t).UnmarshalBinary(data)
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) GobEncode() ([]byte, error) {
|
||||
return (time.Time)(t).GobEncode()
|
||||
}
|
||||
|
||||
func (t *UnixMilliTime) GobDecode(data []byte) error {
|
||||
return (*time.Time)(t).GobDecode(data)
|
||||
}
|
||||
|
||||
func (t *UnixMilliTime) UnmarshalJSON(data []byte) error {
|
||||
str := ""
|
||||
if err := json.Unmarshal(data, &str); err != nil {
|
||||
return err
|
||||
}
|
||||
t0, err := strconv.ParseInt(str, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixMilliTime(time.UnixMilli(t0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) MarshalJSON() ([]byte, error) {
|
||||
str := strconv.FormatInt(t.Time().UnixMilli(), 10)
|
||||
return json.Marshal(str)
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) MarshalText() ([]byte, error) {
|
||||
return []byte(strconv.FormatInt(t.Time().UnixMilli(), 10)), nil
|
||||
}
|
||||
|
||||
func (t *UnixMilliTime) UnmarshalText(data []byte) error {
|
||||
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixMilliTime(time.UnixMilli(t0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Serialize() string {
|
||||
return strconv.FormatInt(t.Time().UnixMilli(), 10)
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) After(u RFCTime) bool {
|
||||
return t.Time().After(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Before(u RFCTime) bool {
|
||||
return t.Time().Before(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Equal(u RFCTime) bool {
|
||||
return t.Time().Equal(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) IsZero() bool {
|
||||
return t.Time().IsZero()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Date() (year int, month time.Month, day int) {
|
||||
return t.Time().Date()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Year() int {
|
||||
return t.Time().Year()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Month() time.Month {
|
||||
return t.Time().Month()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Day() int {
|
||||
return t.Time().Day()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Weekday() time.Weekday {
|
||||
return t.Time().Weekday()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) ISOWeek() (year, week int) {
|
||||
return t.Time().ISOWeek()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Clock() (hour, min, sec int) {
|
||||
return t.Time().Clock()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Hour() int {
|
||||
return t.Time().Hour()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Minute() int {
|
||||
return t.Time().Minute()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Second() int {
|
||||
return t.Time().Second()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Nanosecond() int {
|
||||
return t.Time().Nanosecond()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) YearDay() int {
|
||||
return t.Time().YearDay()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Add(d time.Duration) UnixMilliTime {
|
||||
return UnixMilliTime(t.Time().Add(d))
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Sub(u RFCTime) time.Duration {
|
||||
return t.Time().Sub(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) AddDate(years int, months int, days int) UnixMilliTime {
|
||||
return UnixMilliTime(t.Time().AddDate(years, months, days))
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Unix() int64 {
|
||||
return t.Time().Unix()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) UnixMilli() int64 {
|
||||
return t.Time().UnixMilli()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) UnixMicro() int64 {
|
||||
return t.Time().UnixMicro()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) UnixNano() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) Format(layout string) string {
|
||||
return t.Time().Format(layout)
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) GoString() string {
|
||||
return t.Time().GoString()
|
||||
}
|
||||
|
||||
func (t UnixMilliTime) String() string {
|
||||
return t.Time().String()
|
||||
}
|
||||
|
||||
func NewUnixMilli(t time.Time) UnixMilliTime {
|
||||
return UnixMilliTime(t)
|
||||
}
|
||||
|
||||
func NowUnixMilli() UnixMilliTime {
|
||||
return UnixMilliTime(time.Now())
|
||||
}
|
176
rfctime/unixNano.go
Normal file
176
rfctime/unixNano.go
Normal file
@@ -0,0 +1,176 @@
|
||||
package rfctime
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"strconv"
|
||||
"time"
|
||||
)
|
||||
|
||||
type UnixNanoTime time.Time
|
||||
|
||||
func (t UnixNanoTime) Time() time.Time {
|
||||
return time.Time(t)
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) MarshalBinary() ([]byte, error) {
|
||||
return (time.Time)(t).MarshalBinary()
|
||||
}
|
||||
|
||||
func (t *UnixNanoTime) UnmarshalBinary(data []byte) error {
|
||||
return (*time.Time)(t).UnmarshalBinary(data)
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) GobEncode() ([]byte, error) {
|
||||
return (time.Time)(t).GobEncode()
|
||||
}
|
||||
|
||||
func (t *UnixNanoTime) GobDecode(data []byte) error {
|
||||
return (*time.Time)(t).GobDecode(data)
|
||||
}
|
||||
|
||||
func (t *UnixNanoTime) UnmarshalJSON(data []byte) error {
|
||||
str := ""
|
||||
if err := json.Unmarshal(data, &str); err != nil {
|
||||
return err
|
||||
}
|
||||
t0, err := strconv.ParseInt(str, 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixNanoTime(time.Unix(0, t0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) MarshalJSON() ([]byte, error) {
|
||||
str := strconv.FormatInt(t.Time().UnixNano(), 10)
|
||||
return json.Marshal(str)
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) MarshalText() ([]byte, error) {
|
||||
return []byte(strconv.FormatInt(t.Time().UnixNano(), 10)), nil
|
||||
}
|
||||
|
||||
func (t *UnixNanoTime) UnmarshalText(data []byte) error {
|
||||
t0, err := strconv.ParseInt(string(data), 10, 64)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
*t = UnixNanoTime(time.Unix(0, t0))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Serialize() string {
|
||||
return strconv.FormatInt(t.Time().UnixNano(), 10)
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) After(u RFCTime) bool {
|
||||
return t.Time().After(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Before(u RFCTime) bool {
|
||||
return t.Time().Before(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Equal(u RFCTime) bool {
|
||||
return t.Time().Equal(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) IsZero() bool {
|
||||
return t.Time().IsZero()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Date() (year int, month time.Month, day int) {
|
||||
return t.Time().Date()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Year() int {
|
||||
return t.Time().Year()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Month() time.Month {
|
||||
return t.Time().Month()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Day() int {
|
||||
return t.Time().Day()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Weekday() time.Weekday {
|
||||
return t.Time().Weekday()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) ISOWeek() (year, week int) {
|
||||
return t.Time().ISOWeek()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Clock() (hour, min, sec int) {
|
||||
return t.Time().Clock()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Hour() int {
|
||||
return t.Time().Hour()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Minute() int {
|
||||
return t.Time().Minute()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Second() int {
|
||||
return t.Time().Second()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Nanosecond() int {
|
||||
return t.Time().Nanosecond()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) YearDay() int {
|
||||
return t.Time().YearDay()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Add(d time.Duration) UnixNanoTime {
|
||||
return UnixNanoTime(t.Time().Add(d))
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Sub(u RFCTime) time.Duration {
|
||||
return t.Time().Sub(u.Time())
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) AddDate(years int, months int, days int) UnixNanoTime {
|
||||
return UnixNanoTime(t.Time().AddDate(years, months, days))
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Unix() int64 {
|
||||
return t.Time().Unix()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) UnixMilli() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) UnixMicro() int64 {
|
||||
return t.Time().UnixMicro()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) UnixNano() int64 {
|
||||
return t.Time().UnixNano()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) Format(layout string) string {
|
||||
return t.Time().Format(layout)
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) GoString() string {
|
||||
return t.Time().GoString()
|
||||
}
|
||||
|
||||
func (t UnixNanoTime) String() string {
|
||||
return t.Time().String()
|
||||
}
|
||||
|
||||
func NewUnixNano(t time.Time) UnixNanoTime {
|
||||
return UnixNanoTime(t)
|
||||
}
|
||||
|
||||
func NowUnixNano() UnixNanoTime {
|
||||
return UnixNanoTime(time.Now())
|
||||
}
|
10
sq/params.go
10
sq/params.go
@@ -1,3 +1,13 @@
|
||||
package sq
|
||||
|
||||
type PP map[string]any
|
||||
|
||||
func Join(pps ...PP) PP {
|
||||
r := PP{}
|
||||
for _, add := range pps {
|
||||
for k, v := range add {
|
||||
r[k] = v
|
||||
}
|
||||
}
|
||||
return r
|
||||
}
|
||||
|
@@ -50,7 +50,6 @@ func ScanSingle[TData any](rows *sqlx.Rows, mode StructScanMode, sec StructScanS
|
||||
return *new(TData), err
|
||||
}
|
||||
} else if mode == SModeExtended {
|
||||
var data TData
|
||||
err := strscan.StructScanExt(&data)
|
||||
if err != nil {
|
||||
return *new(TData), err
|
||||
|
@@ -51,7 +51,7 @@ func (r *StructScanner) Start(dest any) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
// StructScan forked from github.com/jmoiron/sqlx@v1.3.5/sqlx.go
|
||||
// StructScanExt forked from github.com/jmoiron/sqlx@v1.3.5/sqlx.go
|
||||
// does also wok with nullabel structs (from LEFT JOIN's)
|
||||
func (r *StructScanner) StructScanExt(dest any) error {
|
||||
v := reflect.ValueOf(dest)
|
||||
@@ -113,7 +113,7 @@ func (r *StructScanner) StructScanExt(dest any) error {
|
||||
|
||||
if _, ok := forcenulled[k]; !ok {
|
||||
f := reflectx.FieldByIndexes(v, traversal[0:i])
|
||||
f.Set(reflect.New(f.Type().Elem())) // set to nil
|
||||
f.Set(reflect.Zero(f.Type())) // set to nil
|
||||
forcenulled[k] = true
|
||||
}
|
||||
|
||||
@@ -138,7 +138,7 @@ func (r *StructScanner) StructScanExt(dest any) error {
|
||||
return errors.New(fmt.Sprintf("Cannot set field %v to NULL value from column '%s' (type: %s)", traversal, r.columns[i], f.Type().String()))
|
||||
}
|
||||
|
||||
f.Set(reflect.New(f.Type().Elem())) // set to nil
|
||||
f.Set(reflect.Zero(f.Type())) // set to nil
|
||||
} else {
|
||||
f.Set(val3)
|
||||
}
|
||||
@@ -148,7 +148,7 @@ func (r *StructScanner) StructScanExt(dest any) error {
|
||||
return r.rows.Err()
|
||||
}
|
||||
|
||||
// StructScan forked from github.com/jmoiron/sqlx@v1.3.5/sqlx.go
|
||||
// StructScanBase forked from github.com/jmoiron/sqlx@v1.3.5/sqlx.go
|
||||
// without (relevant) changes
|
||||
func (r *StructScanner) StructScanBase(dest any) error {
|
||||
v := reflect.ValueOf(dest)
|
||||
|
@@ -2,58 +2,50 @@ package syncext
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync/atomic"
|
||||
"gogs.mikescher.com/BlackForestBytes/goext/langext"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type AtomicBool struct {
|
||||
v int32
|
||||
waiter chan bool // unbuffered
|
||||
v bool
|
||||
listener map[string]chan bool
|
||||
lock sync.Mutex
|
||||
}
|
||||
|
||||
func NewAtomicBool(value bool) *AtomicBool {
|
||||
if value {
|
||||
return &AtomicBool{v: 1, waiter: make(chan bool)}
|
||||
} else {
|
||||
return &AtomicBool{v: 0, waiter: make(chan bool)}
|
||||
return &AtomicBool{
|
||||
v: value,
|
||||
listener: make(map[string]chan bool),
|
||||
lock: sync.Mutex{},
|
||||
}
|
||||
}
|
||||
|
||||
func (a *AtomicBool) Get() bool {
|
||||
return atomic.LoadInt32(&a.v) == 1
|
||||
a.lock.Lock()
|
||||
defer a.lock.Unlock()
|
||||
return a.v
|
||||
}
|
||||
|
||||
func (a *AtomicBool) Set(value bool) {
|
||||
if value {
|
||||
atomic.StoreInt32(&a.v, 1)
|
||||
} else {
|
||||
atomic.StoreInt32(&a.v, 0)
|
||||
}
|
||||
a.lock.Lock()
|
||||
defer a.lock.Unlock()
|
||||
|
||||
a.v = value
|
||||
|
||||
for k, v := range a.listener {
|
||||
select {
|
||||
case a.waiter <- value:
|
||||
case v <- value:
|
||||
// message sent
|
||||
default:
|
||||
// no receiver on channel
|
||||
delete(a.listener, k)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (a *AtomicBool) Wait(waitFor bool) {
|
||||
if a.Get() == waitFor {
|
||||
return
|
||||
}
|
||||
|
||||
for {
|
||||
if v, ok := ReadChannelWithTimeout(a.waiter, 128*time.Millisecond); ok {
|
||||
if v == waitFor {
|
||||
return
|
||||
}
|
||||
} else {
|
||||
if a.Get() == waitFor {
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
_ = a.WaitWithContext(context.Background(), waitFor)
|
||||
}
|
||||
|
||||
func (a *AtomicBool) WaitWithTimeout(timeout time.Duration, waitFor bool) error {
|
||||
@@ -71,12 +63,25 @@ func (a *AtomicBool) WaitWithContext(ctx context.Context, waitFor bool) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
uuid, _ := langext.NewHexUUID()
|
||||
|
||||
waitchan := make(chan bool)
|
||||
|
||||
a.lock.Lock()
|
||||
a.listener[uuid] = waitchan
|
||||
a.lock.Unlock()
|
||||
defer func() {
|
||||
a.lock.Lock()
|
||||
delete(a.listener, uuid)
|
||||
a.lock.Unlock()
|
||||
}()
|
||||
|
||||
for {
|
||||
if err := ctx.Err(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
timeOut := 128 * time.Millisecond
|
||||
timeOut := 1024 * time.Millisecond
|
||||
|
||||
if dl, ok := ctx.Deadline(); ok {
|
||||
timeOutMax := dl.Sub(time.Now())
|
||||
@@ -87,7 +92,7 @@ func (a *AtomicBool) WaitWithContext(ctx context.Context, waitFor bool) error {
|
||||
}
|
||||
}
|
||||
|
||||
if v, ok := ReadChannelWithTimeout(a.waiter, timeOut); ok {
|
||||
if v, ok := ReadChannelWithTimeout(waitchan, timeOut); ok {
|
||||
if v == waitFor {
|
||||
return nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user