76 righe
1.3 KiB
Go
76 righe
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"strings"
|
|
|
|
"git.rootprojects.org/root/telebit/internal/mgmt/authstore"
|
|
)
|
|
|
|
func main() {
|
|
connStr := "postgres://postgres:postgres@localhost:5432/postgres"
|
|
|
|
if !strings.Contains(connStr, "sslmode=") {
|
|
sep := "?"
|
|
if strings.Contains(connStr, sep) {
|
|
sep = "&"
|
|
}
|
|
if strings.Contains(connStr, "@localhost/") ||
|
|
strings.Contains(connStr, "@localhost:") {
|
|
connStr += sep + "sslmode=disable"
|
|
} else {
|
|
connStr += sep + "sslmode=required"
|
|
}
|
|
}
|
|
|
|
store, err := authstore.NewStore(connStr, initSQL)
|
|
if nil != err {
|
|
log.Fatal("connection error", err)
|
|
return
|
|
}
|
|
|
|
num := "8"
|
|
slug := num + "-xxx-client"
|
|
pubkey := num + "-somehash"
|
|
auth1 := &authstore.Authorization{
|
|
Slug: slug,
|
|
SharedKey: "3-xxxx-zzzz-yyyy",
|
|
PublicKey: pubkey,
|
|
}
|
|
err = store.Add(auth1)
|
|
if nil != err {
|
|
log.Fatal("add error", err)
|
|
return
|
|
}
|
|
|
|
auth, err := store.Get(slug)
|
|
if nil != err {
|
|
log.Fatal("get by slug error", err)
|
|
return
|
|
}
|
|
|
|
auth, err = store.Get(pubkey)
|
|
if nil != err {
|
|
log.Fatal("get by pub error", err)
|
|
return
|
|
}
|
|
|
|
auth1.MachinePPID = "a-secretish-id"
|
|
err = store.Set(auth1)
|
|
if nil != err {
|
|
log.Fatal("set machine id", err)
|
|
return
|
|
}
|
|
|
|
err = store.Delete(auth1)
|
|
if nil != err {
|
|
log.Fatal("set machine id", err)
|
|
return
|
|
}
|
|
|
|
store.Close()
|
|
|
|
fmt.Printf("%#v\n", auth)
|
|
}
|