Gisle Aune
2 years ago
commit
8fc0220e51
29 changed files with 1315 additions and 0 deletions
-
2.gitignore
-
12auth/provider.go
-
64auth/service.go
-
61auth/user.go
-
37character/character.go
-
9error.go
-
17go.mod
-
26go.sum
-
86internal/generate/id.go
-
29internal/genutils/array.go
-
5internal/genutils/ptr.go
-
29internal/genutils/set.go
-
13internal/genutils/sort.go
-
70internal/genutils/update.go
-
47ports/mysql/database.go
-
19ports/mysql/migrations/20230312124831_create_table_tag.sql
-
128ports/mysql/mysqlgen/db.go
-
20ports/mysql/mysqlgen/models.go
-
147ports/mysql/mysqlgen/tag.sql.go
-
22ports/mysql/null.go
-
20ports/mysql/queries/tag.sql
-
20ports/mysql/sqlc.yaml
-
125ports/mysql/tags.go
-
11tag/repository.go
-
26tag/service.go
-
46tag/tag.go
-
110tag/tree.go
-
95tag/tree_test.go
-
19tag/update.go
@ -0,0 +1,2 @@ |
|||
.idea |
|||
.vscode |
@ -0,0 +1,12 @@ |
|||
package auth |
|||
|
|||
import ( |
|||
"context" |
|||
) |
|||
|
|||
type Provider interface { |
|||
ListUsers(ctx context.Context) ([]UserInfo, error) |
|||
LoginUser(ctx context.Context, username, password string) (*Result, error) |
|||
SetupUser(ctx context.Context, session, username, preferredUsername, newPassword string) (*UserInfo, error) |
|||
ValidateToken(ctx context.Context, token string) *UserInfo |
|||
} |
@ -0,0 +1,64 @@ |
|||
package auth |
|||
|
|||
import ( |
|||
"context" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
type Service struct { |
|||
Provider Provider |
|||
|
|||
userListMutex sync.Mutex |
|||
userList []UserInfo |
|||
userMap map[string]string |
|||
userListTime time.Time |
|||
|
|||
key struct{ Stuff uint64 } |
|||
key2 struct{ Stuff2 string } |
|||
} |
|||
|
|||
func (s *Service) ValidateUser(ctx context.Context, token string) *UserInfo { |
|||
return s.Provider.ValidateToken(ctx, token) |
|||
} |
|||
|
|||
func (s *Service) AddContextUser(ctx context.Context, user UserInfo) context.Context { |
|||
return context.WithValue(ctx, &s.key, &user) |
|||
} |
|||
|
|||
func (s *Service) Users(ctx context.Context) ([]UserInfo, map[string]string, error) { |
|||
s.userListMutex.Lock() |
|||
if time.Since(s.userListTime) < time.Minute { |
|||
m := s.userMap |
|||
l := s.userList |
|||
s.userListMutex.Unlock() |
|||
return l, m, nil |
|||
} |
|||
s.userListMutex.Unlock() |
|||
|
|||
users, err := s.Provider.ListUsers(ctx) |
|||
if err != nil { |
|||
return nil, nil, err |
|||
} |
|||
|
|||
s.userListMutex.Lock() |
|||
s.userList = users |
|||
s.userMap = make(map[string]string, len(users)) |
|||
for _, user := range users { |
|||
s.userMap[user.ID] = user.Name |
|||
} |
|||
m := s.userMap |
|||
s.userListTime = time.Now() |
|||
s.userListMutex.Unlock() |
|||
|
|||
return users, m, nil |
|||
} |
|||
|
|||
func (s *Service) GetUser(ctx context.Context) *UserInfo { |
|||
v := ctx.Value(&s.key) |
|||
if v == nil { |
|||
return nil |
|||
} |
|||
|
|||
return v.(*UserInfo) |
|||
} |
@ -0,0 +1,61 @@ |
|||
package auth |
|||
|
|||
import ( |
|||
"fmt" |
|||
) |
|||
|
|||
type User struct { |
|||
ID string `json:"id"` |
|||
Name string `json:"name"` |
|||
} |
|||
|
|||
type UserInfo struct { |
|||
User |
|||
Permissions []string `json:"permissions"` |
|||
} |
|||
|
|||
type Result struct { |
|||
User *UserInfo `json:"user"` |
|||
Token string `json:"token,omitempty"` |
|||
Session string `json:"session,omitempty"` |
|||
PasswordChangeRequired bool `json:"passwordChangeRequired"` |
|||
} |
|||
|
|||
func (user *UserInfo) HasIDOrPermission(userID, subject, action string) bool { |
|||
return user.HasID(userID) || user.HasPermission(subject, action) |
|||
} |
|||
|
|||
func (user *UserInfo) HasID(id string) bool { |
|||
return user != nil && user.ID == id |
|||
} |
|||
|
|||
func (user *UserInfo) HasOpPermission(userID, subject, action string) bool { |
|||
if !user.HasID(userID) && !user.HasPermission(subject, "admin") { |
|||
return false |
|||
} |
|||
|
|||
return user.HasPermission(subject, action) |
|||
} |
|||
|
|||
func (user *UserInfo) HasPermission(subject, action string) bool { |
|||
if user == nil { |
|||
return false |
|||
} |
|||
|
|||
anyAll := "*.*" |
|||
anyAction := fmt.Sprintf("%s.*", subject) |
|||
anySubject := fmt.Sprintf("*.%s", action) |
|||
specific := fmt.Sprintf("%s.%s", subject, action) |
|||
|
|||
if action == "admin" { |
|||
anyAction = specific |
|||
} |
|||
|
|||
for _, perm := range user.Permissions { |
|||
if perm == anyAll || perm == anyAction || perm == anySubject || perm == specific { |
|||
return true |
|||
} |
|||
} |
|||
|
|||
return false |
|||
} |
@ -0,0 +1,37 @@ |
|||
package character |
|||
|
|||
import ( |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/auth" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/tag" |
|||
) |
|||
|
|||
type Character struct { |
|||
ID string `json:"id"` |
|||
Name string `json:"name"` |
|||
ShortName string `json:"shortName"` |
|||
DisplayName string `json:"displayName,omitempty"` |
|||
Description string `json:"description"` |
|||
FavoriteColor string `json:"favoriteColor"` |
|||
WikiPage string `json:"wikiPage"` |
|||
|
|||
Author auth.User `json:"author"` |
|||
Attributes []Attribute `json:"attributes"` |
|||
Relations []Relation `json:"relations"` |
|||
Tags []Tag `json:"tags"` |
|||
} |
|||
|
|||
type Attribute struct { |
|||
Label string `json:"label,omitempty"` |
|||
Value string `json:"value,omitempty"` |
|||
} |
|||
|
|||
type Relation struct { |
|||
Label string `json:"label,omitempty"` |
|||
Character |
|||
} |
|||
|
|||
type Tag struct { |
|||
Label string `json:"label,omitempty"` |
|||
Restricted bool `json:"restricted,omitempty"` |
|||
tag.Tag |
|||
} |
@ -0,0 +1,9 @@ |
|||
package rpdata2 |
|||
|
|||
import "fmt" |
|||
|
|||
type NotFound string |
|||
|
|||
func (e NotFound) Error() string { |
|||
return fmt.Sprintf("%s not found", string(e)) |
|||
} |
@ -0,0 +1,17 @@ |
|||
module git.aiterp.net/rpdata2-take2/rpdata2 |
|||
|
|||
go 1.19 |
|||
|
|||
require ( |
|||
github.com/Masterminds/squirrel v1.5.3 |
|||
github.com/google/uuid v1.3.0 |
|||
github.com/stretchr/testify v1.8.2 |
|||
) |
|||
|
|||
require ( |
|||
github.com/davecgh/go-spew v1.1.1 // indirect |
|||
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 // indirect |
|||
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 // indirect |
|||
github.com/pmezard/go-difflib v1.0.0 // indirect |
|||
gopkg.in/yaml.v3 v3.0.1 // indirect |
|||
) |
@ -0,0 +1,26 @@ |
|||
github.com/Masterminds/squirrel v1.5.3 h1:YPpoceAcxuzIljlr5iWpNKaql7hLeG1KLSrhvdHpkZc= |
|||
github.com/Masterminds/squirrel v1.5.3/go.mod h1:NNaOrjSoIDfDA40n7sr2tPNZRfjzjA400rg+riTZj10= |
|||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= |
|||
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/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I= |
|||
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= |
|||
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 h1:SOEGU9fKiNWd/HOJuq6+3iTQz8KNCLtVX6idSoTLdUw= |
|||
github.com/lann/builder v0.0.0-20180802200727-47ae307949d0/go.mod h1:dXGbAdH5GtBTC4WfIxhKZfyBF/HBFgRZSWwZ9g/He9o= |
|||
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 h1:P6pPBnrTSX3DEVR4fDembhRWSsG5rVo6hYhAB/ADZrk= |
|||
github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0/go.mod h1:vmVJ0l/dxyfGW6FmdpVm2joNMFikkuWg0EoCKLGUMNw= |
|||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM= |
|||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= |
|||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= |
|||
github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= |
|||
github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= |
|||
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= |
|||
github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= |
|||
github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= |
|||
github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8= |
|||
github.com/stretchr/testify v1.8.2/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= |
|||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM= |
|||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= |
|||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
|||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= |
|||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= |
@ -0,0 +1,86 @@ |
|||
package generate |
|||
|
|||
import ( |
|||
"crypto/rand" |
|||
"encoding/binary" |
|||
mathRand "math/rand" |
|||
"strconv" |
|||
"strings" |
|||
) |
|||
|
|||
// ID generates an ID using crypto-random, falling back to math random when that fails
|
|||
// to avoid disrupting operation because of a faulty RNG.
|
|||
func ID(prefix string, length int) string { |
|||
var data [32]byte |
|||
|
|||
result := strings.Builder{} |
|||
result.Grow(length + 32) |
|||
result.WriteString(prefix) |
|||
|
|||
pos := 0 |
|||
for result.Len() < length { |
|||
if pos == 0 { |
|||
randRead(data[:]) |
|||
} |
|||
|
|||
result.WriteString(strconv.FormatUint(binary.BigEndian.Uint64(data[pos:pos+8]), 36)) |
|||
|
|||
pos = (pos + 8) % 32 |
|||
} |
|||
|
|||
return result.String()[:length] |
|||
} |
|||
|
|||
func randRead(data []byte) { |
|||
n, err := rand.Read(data) |
|||
if err != nil { |
|||
mathRand.Read(data[n:]) |
|||
} |
|||
} |
|||
|
|||
// InternalErrorID generates a long string
|
|||
func InternalErrorID() string { |
|||
return ID("ISE", 32) |
|||
} |
|||
|
|||
// TagID generates a location ID. len=8
|
|||
func TagID(longName string) string { |
|||
return friendlyID('T', longName, 8) |
|||
} |
|||
|
|||
// CharacterID generates a character ID. len=8
|
|||
func CharacterID(longName string) string { |
|||
return friendlyID('C', longName, 8) |
|||
} |
|||
|
|||
func friendlyID(prefix byte, name string, length int) string { |
|||
b := strings.Builder{} |
|||
b.Grow(4) |
|||
b.WriteByte(prefix) |
|||
for _, ch := range strings.ToLower(name) { |
|||
if ch >= 'a' && ch <= 'z' { |
|||
b.WriteRune(ch) |
|||
|
|||
if b.Len() > 3 { |
|||
break |
|||
} |
|||
} |
|||
} |
|||
|
|||
return ID(b.String(), length) |
|||
} |
|||
|
|||
// StoryID generates a story ID: len=8
|
|||
func StoryID() string { |
|||
return ID("S", 12) |
|||
} |
|||
|
|||
// PostID generates a post ID. len=12
|
|||
func PostID() string { |
|||
return ID("P", 16) |
|||
} |
|||
|
|||
// AnnotationID generates an annotation.sql ID. len=12
|
|||
func AnnotationID() string { |
|||
return ID("A", 16) |
|||
} |
@ -0,0 +1,29 @@ |
|||
package genutils |
|||
|
|||
func UpsertIntoArray[T comparable](arr []T, values ...T) []T { |
|||
outer: |
|||
for _, value := range values { |
|||
for _, value2 := range arr { |
|||
if value2 == value { |
|||
continue outer |
|||
} |
|||
} |
|||
|
|||
arr = append(arr, value) |
|||
} |
|||
|
|||
return arr |
|||
} |
|||
|
|||
func RemoveFromArray[T comparable](arr []T, values ...T) []T { |
|||
for _, value := range values { |
|||
for i, value2 := range arr { |
|||
if value2 == value { |
|||
arr = append(arr[:i], arr[i+1:]...) |
|||
break |
|||
} |
|||
} |
|||
} |
|||
|
|||
return arr |
|||
} |
@ -0,0 +1,5 @@ |
|||
package genutils |
|||
|
|||
func Ptr[T any](t T) *T { |
|||
return &t |
|||
} |
@ -0,0 +1,29 @@ |
|||
package genutils |
|||
|
|||
type Set[T comparable] struct { |
|||
m map[T]bool |
|||
a []T |
|||
} |
|||
|
|||
func (set *Set[T]) Add(values ...T) { |
|||
if set.m == nil { |
|||
set.m = make(map[T]bool, len(values)*4) |
|||
} |
|||
|
|||
for _, value := range values { |
|||
if set.m[value] { |
|||
continue |
|||
} |
|||
|
|||
set.m[value] = true |
|||
set.a = append(set.a, value) |
|||
} |
|||
} |
|||
|
|||
func (set *Set[T]) Len() int { |
|||
return len(set.a) |
|||
} |
|||
|
|||
func (set *Set[T]) Values() []T { |
|||
return set.a[:len(set.a):len(set.a)] |
|||
} |
@ -0,0 +1,13 @@ |
|||
package genutils |
|||
|
|||
import "sort" |
|||
|
|||
type Lesser[T any] interface { |
|||
Less(T) bool |
|||
} |
|||
|
|||
func SortSlice[T Lesser[T]](slice []T) { |
|||
sort.Slice(slice, func(i, j int) bool { |
|||
return slice[i].Less(slice[j]) |
|||
}) |
|||
} |
@ -0,0 +1,70 @@ |
|||
package genutils |
|||
|
|||
type hasValid interface { |
|||
Valid() bool |
|||
} |
|||
|
|||
func ApplyUpdate[T any](dst *T, value *T) { |
|||
if value != nil { |
|||
*dst = *value |
|||
} |
|||
} |
|||
|
|||
func ApplyUpdateValid[T hasValid](dst *T, value *T) { |
|||
if value != nil && (*value).Valid() { |
|||
*dst = *value |
|||
} |
|||
} |
|||
|
|||
func ApplyUpdateNonZero[T comparable](dst *T, value *T) { |
|||
var zero T |
|||
if value != nil && *value != zero { |
|||
*dst = *value |
|||
} |
|||
} |
|||
|
|||
func ApplyUpdateNilZero[T comparable](dst **T, value *T) { |
|||
if value != nil { |
|||
var zero T |
|||
|
|||
if *value == zero { |
|||
*dst = nil |
|||
} else { |
|||
valueCopy := *value |
|||
*dst = &valueCopy |
|||
} |
|||
} |
|||
} |
|||
|
|||
func ApplyMapUpdate[K comparable, V any](dst *map[K]V, src map[K]*V) { |
|||
if *dst == nil { |
|||
dst = &map[K]V{} |
|||
} |
|||
for key, value := range src { |
|||
if value != nil { |
|||
(*dst)[key] = *value |
|||
} else { |
|||
delete(*dst, key) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func ApplyUpdateMapNilZero[K comparable, V comparable](dst *map[K]V, src map[K]V) { |
|||
var zero V |
|||
if *dst == nil { |
|||
dst = &map[K]V{} |
|||
} |
|||
for key, value := range src { |
|||
if value != zero { |
|||
(*dst)[key] = value |
|||
} else { |
|||
delete(*dst, key) |
|||
} |
|||
} |
|||
} |
|||
|
|||
func ApplyArrayUpdate[T comparable](arr *[]T, upsert []T, remove []T) { |
|||
*arr = make([]T, 0, len(*arr)+len(upsert)) |
|||
*arr = UpsertIntoArray(*arr, upsert...) |
|||
*arr = RemoveFromArray(*arr, remove...) |
|||
} |
@ -0,0 +1,47 @@ |
|||
package mysql |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"fmt" |
|||
"git.aiterp.net/rpdata2/rpdata2-take2/ports/mysql/mysqlgen" |
|||
"git.aiterp.net/rpdata2/rpdata2-take2/tag" |
|||
"time" |
|||
) |
|||
|
|||
type Database struct { |
|||
db *sql.DB |
|||
q *mysqlgen.Queries |
|||
} |
|||
|
|||
func (db *Database) Tags() tag.Repository { |
|||
return &tagRepository{ |
|||
db: db.db, |
|||
q: db.q, |
|||
} |
|||
} |
|||
|
|||
func Connect(host string, port int, username, password, database string) (*Database, error) { |
|||
db, err := sql.Open("mysql", fmt.Sprintf( |
|||
"%s:%s@(%s:%d)/%s?parseTime=true", username, password, host, port, database, |
|||
)) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
db.SetMaxOpenConns(10) |
|||
db.SetMaxIdleConns(10) |
|||
db.SetConnMaxIdleTime(time.Minute) |
|||
|
|||
err = db.Ping() |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
q, err := mysqlgen.Prepare(context.Background(), db) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return &Database{db: db, q: q}, nil |
|||
} |
@ -0,0 +1,19 @@ |
|||
-- +goose Up |
|||
-- +goose StatementBegin |
|||
CREATE TABLE tag |
|||
( |
|||
`id` CHAR(8) NOT NULL PRIMARY KEY, |
|||
`parent_id` CHAR(8) NULL, |
|||
`owner_id` VARCHAR(255) NOT NULL, |
|||
`name` VARCHAR(255) NOT NULL, |
|||
`kind` INT NOT NULL, |
|||
`description` TEXT NOT NULL, |
|||
`listed` BOOL NOT NULL, |
|||
`secret` BOOL NOT NULL |
|||
); |
|||
-- +goose StatementEnd |
|||
|
|||
-- +goose Down |
|||
-- +goose StatementBegin |
|||
DROP TABLE tag; |
|||
-- +goose StatementEnd |
@ -0,0 +1,128 @@ |
|||
// Code generated by sqlc. DO NOT EDIT.
|
|||
// versions:
|
|||
// sqlc v1.13.0
|
|||
|
|||
package mysqlgen |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"fmt" |
|||
) |
|||
|
|||
type DBTX interface { |
|||
ExecContext(context.Context, string, ...interface{}) (sql.Result, error) |
|||
PrepareContext(context.Context, string) (*sql.Stmt, error) |
|||
QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error) |
|||
QueryRowContext(context.Context, string, ...interface{}) *sql.Row |
|||
} |
|||
|
|||
func New(db DBTX) *Queries { |
|||
return &Queries{db: db} |
|||
} |
|||
|
|||
func Prepare(ctx context.Context, db DBTX) (*Queries, error) { |
|||
q := Queries{db: db} |
|||
var err error |
|||
if q.deleteTagStmt, err = db.PrepareContext(ctx, deleteTag); err != nil { |
|||
return nil, fmt.Errorf("error preparing query DeleteTag: %w", err) |
|||
} |
|||
if q.findTagStmt, err = db.PrepareContext(ctx, findTag); err != nil { |
|||
return nil, fmt.Errorf("error preparing query FindTag: %w", err) |
|||
} |
|||
if q.listListedTagsStmt, err = db.PrepareContext(ctx, listListedTags); err != nil { |
|||
return nil, fmt.Errorf("error preparing query ListListedTags: %w", err) |
|||
} |
|||
if q.listTagsStmt, err = db.PrepareContext(ctx, listTags); err != nil { |
|||
return nil, fmt.Errorf("error preparing query ListTags: %w", err) |
|||
} |
|||
if q.replaceTagStmt, err = db.PrepareContext(ctx, replaceTag); err != nil { |
|||
return nil, fmt.Errorf("error preparing query ReplaceTag: %w", err) |
|||
} |
|||
return &q, nil |
|||
} |
|||
|
|||
func (q *Queries) Close() error { |
|||
var err error |
|||
if q.deleteTagStmt != nil { |
|||
if cerr := q.deleteTagStmt.Close(); cerr != nil { |
|||
err = fmt.Errorf("error closing deleteTagStmt: %w", cerr) |
|||
} |
|||
} |
|||
if q.findTagStmt != nil { |
|||
if cerr := q.findTagStmt.Close(); cerr != nil { |
|||
err = fmt.Errorf("error closing findTagStmt: %w", cerr) |
|||
} |
|||
} |
|||
if q.listListedTagsStmt != nil { |
|||
if cerr := q.listListedTagsStmt.Close(); cerr != nil { |
|||
err = fmt.Errorf("error closing listListedTagsStmt: %w", cerr) |
|||
} |
|||
} |
|||
if q.listTagsStmt != nil { |
|||
if cerr := q.listTagsStmt.Close(); cerr != nil { |
|||
err = fmt.Errorf("error closing listTagsStmt: %w", cerr) |
|||
} |
|||
} |
|||
if q.replaceTagStmt != nil { |
|||
if cerr := q.replaceTagStmt.Close(); cerr != nil { |
|||
err = fmt.Errorf("error closing replaceTagStmt: %w", cerr) |
|||
} |
|||
} |
|||
return err |
|||
} |
|||
|
|||
func (q *Queries) exec(ctx context.Context, stmt *sql.Stmt, query string, args ...interface{}) (sql.Result, error) { |
|||
switch { |
|||
case stmt != nil && q.tx != nil: |
|||
return q.tx.StmtContext(ctx, stmt).ExecContext(ctx, args...) |
|||
case stmt != nil: |
|||
return stmt.ExecContext(ctx, args...) |
|||
default: |
|||
return q.db.ExecContext(ctx, query, args...) |
|||
} |
|||
} |
|||
|
|||
func (q *Queries) query(ctx context.Context, stmt *sql.Stmt, query string, args ...interface{}) (*sql.Rows, error) { |
|||
switch { |
|||
case stmt != nil && q.tx != nil: |
|||
return q.tx.StmtContext(ctx, stmt).QueryContext(ctx, args...) |
|||
case stmt != nil: |
|||
return stmt.QueryContext(ctx, args...) |
|||
default: |
|||
return q.db.QueryContext(ctx, query, args...) |
|||
} |
|||
} |
|||
|
|||
func (q *Queries) queryRow(ctx context.Context, stmt *sql.Stmt, query string, args ...interface{}) *sql.Row { |
|||
switch { |
|||
case stmt != nil && q.tx != nil: |
|||
return q.tx.StmtContext(ctx, stmt).QueryRowContext(ctx, args...) |
|||
case stmt != nil: |
|||
return stmt.QueryRowContext(ctx, args...) |
|||
default: |
|||
return q.db.QueryRowContext(ctx, query, args...) |
|||
} |
|||
} |
|||
|
|||
type Queries struct { |
|||
db DBTX |
|||
tx *sql.Tx |
|||
deleteTagStmt *sql.Stmt |
|||
findTagStmt *sql.Stmt |
|||
listListedTagsStmt *sql.Stmt |
|||
listTagsStmt *sql.Stmt |
|||
replaceTagStmt *sql.Stmt |
|||
} |
|||
|
|||
func (q *Queries) WithTx(tx *sql.Tx) *Queries { |
|||
return &Queries{ |
|||
db: tx, |
|||
tx: tx, |
|||
deleteTagStmt: q.deleteTagStmt, |
|||
findTagStmt: q.findTagStmt, |
|||
listListedTagsStmt: q.listListedTagsStmt, |
|||
listTagsStmt: q.listTagsStmt, |
|||
replaceTagStmt: q.replaceTagStmt, |
|||
} |
|||
} |
@ -0,0 +1,20 @@ |
|||
// Code generated by sqlc. DO NOT EDIT.
|
|||
// versions:
|
|||
// sqlc v1.13.0
|
|||
|
|||
package mysqlgen |
|||
|
|||
import ( |
|||
"database/sql" |
|||
) |
|||
|
|||
type Tag struct { |
|||
ID string |
|||
ParentID sql.NullString |
|||
OwnerID string |
|||
Name string |
|||
Kind int |
|||
Description string |
|||
Listed bool |
|||
Secret bool |
|||
} |
@ -0,0 +1,147 @@ |
|||
// Code generated by sqlc. DO NOT EDIT.
|
|||
// versions:
|
|||
// sqlc v1.13.0
|
|||
// source: tag.sql
|
|||
|
|||
package mysqlgen |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
) |
|||
|
|||
const deleteTag = `-- name: DeleteTag :exec |
|||
DELETE FROM tag WHERE id = ? |
|||
` |
|||
|
|||
func (q *Queries) DeleteTag(ctx context.Context, id string) error { |
|||
_, err := q.exec(ctx, q.deleteTagStmt, deleteTag, id) |
|||
return err |
|||
} |
|||
|
|||
const findTag = `-- name: FindTag :one |
|||
SELECT id, parent_id, owner_id, name, kind, description, listed, secret |
|||
FROM tag |
|||
WHERE id = ? |
|||
` |
|||
|
|||
func (q *Queries) FindTag(ctx context.Context, id string) (Tag, error) { |
|||
row := q.queryRow(ctx, q.findTagStmt, findTag, id) |
|||
var i Tag |
|||
err := row.Scan( |
|||
&i.ID, |
|||
&i.ParentID, |
|||
&i.OwnerID, |
|||
&i.Name, |
|||
&i.Kind, |
|||
&i.Description, |
|||
&i.Listed, |
|||
&i.Secret, |
|||
) |
|||
return i, err |
|||
} |
|||
|
|||
const listListedTags = `-- name: ListListedTags :many |
|||
SELECT id, parent_id, owner_id, name, kind, description, listed, secret |
|||
FROM tag |
|||
WHERE listed = 1 |
|||
` |
|||
|
|||
func (q *Queries) ListListedTags(ctx context.Context) ([]Tag, error) { |
|||
rows, err := q.query(ctx, q.listListedTagsStmt, listListedTags) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
defer rows.Close() |
|||
items := []Tag{} |
|||
for rows.Next() { |
|||
var i Tag |
|||
if err := rows.Scan( |
|||
&i.ID, |
|||
&i.ParentID, |
|||
&i.OwnerID, |
|||
&i.Name, |
|||
&i.Kind, |
|||
&i.Description, |
|||
&i.Listed, |
|||
&i.Secret, |
|||
); err != nil { |
|||
return nil, err |
|||
} |
|||
items = append(items, i) |
|||
} |
|||
if err := rows.Close(); err != nil { |
|||
return nil, err |
|||
} |
|||
if err := rows.Err(); err != nil { |
|||
return nil, err |
|||
} |
|||
return items, nil |
|||
} |
|||
|
|||
const listTags = `-- name: ListTags :many |
|||
SELECT id, parent_id, owner_id, name, kind, description, listed, secret |
|||
FROM tag |
|||
` |
|||
|
|||
func (q *Queries) ListTags(ctx context.Context) ([]Tag, error) { |
|||
rows, err := q.query(ctx, q.listTagsStmt, listTags) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
defer rows.Close() |
|||
items := []Tag{} |
|||
for rows.Next() { |
|||
var i Tag |
|||
if err := rows.Scan( |
|||
&i.ID, |
|||
&i.ParentID, |
|||
&i.OwnerID, |
|||
&i.Name, |
|||
&i.Kind, |
|||
&i.Description, |
|||
&i.Listed, |
|||
&i.Secret, |
|||
); err != nil { |
|||
return nil, err |
|||
} |
|||
items = append(items, i) |
|||
} |
|||
if err := rows.Close(); err != nil { |
|||
return nil, err |
|||
} |
|||
if err := rows.Err(); err != nil { |
|||
return nil, err |
|||
} |
|||
return items, nil |
|||
} |
|||
|
|||
const replaceTag = `-- name: ReplaceTag :exec |
|||
REPLACE INTO tag (id, parent_id, owner_id, name, kind, description, listed, secret) |
|||
VALUES (?, ?, ?, ?, ?, ?, ?, ?) |
|||
` |
|||
|
|||
type ReplaceTagParams struct { |
|||
ID string |
|||
ParentID sql.NullString |
|||
OwnerID string |
|||
Name string |
|||
Kind int |
|||
Description string |
|||
Listed bool |
|||
Secret bool |
|||
} |
|||
|
|||
func (q *Queries) ReplaceTag(ctx context.Context, arg ReplaceTagParams) error { |
|||
_, err := q.exec(ctx, q.replaceTagStmt, replaceTag, |
|||
arg.ID, |
|||
arg.ParentID, |
|||
arg.OwnerID, |
|||
arg.Name, |
|||
arg.Kind, |
|||
arg.Description, |
|||
arg.Listed, |
|||
arg.Secret, |
|||
) |
|||
return err |
|||
} |
@ -0,0 +1,22 @@ |
|||
package mysql |
|||
|
|||
import ( |
|||
"database/sql" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/internal/genutils" |
|||
) |
|||
|
|||
func toNullString(s *string) sql.NullString { |
|||
if s != nil { |
|||
return sql.NullString{Valid: true, String: *s} |
|||
} else { |
|||
return sql.NullString{Valid: false} |
|||
} |
|||
} |
|||
|
|||
func fromNullString(ns sql.NullString) *string { |
|||
if ns.Valid { |
|||
return genutils.Ptr(ns.String) |
|||
} else { |
|||
return nil |
|||
} |
|||
} |
@ -0,0 +1,20 @@ |
|||
-- name: FindTag :one |
|||
SELECT * |
|||
FROM tag |
|||
WHERE id = ?; |
|||
|
|||
-- name: ListTags :many |
|||
SELECT * |
|||
FROM tag; |
|||
|
|||
-- name: ListListedTags :many |
|||
SELECT * |
|||
FROM tag |
|||
WHERE listed = 1; |
|||
|
|||
-- name: ReplaceTag :exec |
|||
REPLACE INTO tag (id, parent_id, owner_id, name, kind, description, listed, secret) |
|||
VALUES (?, ?, ?, ?, ?, ?, ?, ?); |
|||
|
|||
-- name: DeleteTag :exec |
|||
DELETE FROM tag WHERE id = ?; |
@ -0,0 +1,20 @@ |
|||
version: "1" |
|||
packages: |
|||
- name: "mysqlgen" |
|||
path: "./mysqlgen" |
|||
queries: "./queries" |
|||
schema: "./migrations" |
|||
engine: "mysql" |
|||
emit_prepared_queries: true |
|||
emit_interface: false |
|||
emit_exact_table_names: false |
|||
emit_empty_slices: true |
|||
emit_json_tags: false |
|||
overrides: |
|||
- go_type: "float64" |
|||
db_type: "float" |
|||
- go_type: "float64" |
|||
db_type: "float" |
|||
nullable: true |
|||
- go_type: "int" |
|||
db_type: "int" |
@ -0,0 +1,125 @@ |
|||
package mysql |
|||
|
|||
import ( |
|||
"context" |
|||
"database/sql" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/ports/mysql/mysqlgen" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/tag" |
|||
"github.com/Masterminds/squirrel" |
|||
) |
|||
|
|||
type tagRepository struct { |
|||
db *sql.DB |
|||
q *mysqlgen.Queries |
|||
} |
|||
|
|||
func (r *tagRepository) FindOne(ctx context.Context, id string) (*tag.Tag, error) { |
|||
row, err := r.q.FindTag(ctx, id) |
|||
if err != sql.ErrNoRows { |
|||
return nil, rpdata2.NotFound("Tag") |
|||
} else if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
return &tag.Tag{ |
|||
ID: row.ID, |
|||
ParentID: fromNullString(row.ParentID), |
|||
OwnerID: row.OwnerID, |
|||
Name: row.Name, |
|||
Kind: tag.Kind(row.Kind), |
|||
Description: row.Description, |
|||
Listed: row.Listed, |
|||
Secret: row.Secret, |
|||
}, nil |
|||
} |
|||
|
|||
func (r *tagRepository) FindRecursive(ctx context.Context, id string) ([]tag.Tag, error) { |
|||
return r.findRecursive(ctx, make([]tag.Tag, 0, 32), []string{id}) |
|||
} |
|||
|
|||
func (r *tagRepository) findRecursive(ctx context.Context, tags []tag.Tag, parentIDs []string) ([]tag.Tag, error) { |
|||
query, args, err := squirrel.Select("id,parent_id,owner_id,name,kind,description,listed,secret"). |
|||
From("tag"). |
|||
Where(squirrel.Eq{"parent_id": parentIDs}). |
|||
ToSql() |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
rows, err := r.db.QueryContext(ctx, query, args...) |
|||
if err != nil && err != sql.ErrNoRows { |
|||
return nil, err |
|||
} |
|||
defer rows.Close() |
|||
|
|||
nextParentIDs := make([]string, 0, 8) |
|||
for rows.Next() { |
|||
var t tag.Tag |
|||
var ns sql.NullString |
|||
|
|||
err = rows.Scan(&t.ID, &ns, &t.OwnerID, &t.Name, &t.Kind, &t.Description, &t.Listed, &t.Secret) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
t.ParentID = fromNullString(ns) |
|||
nextParentIDs = append(nextParentIDs, t.ID) |
|||
|
|||
tags = append(tags, t) |
|||
} |
|||
|
|||
if len(nextParentIDs) > 0 { |
|||
return r.findRecursive(ctx, tags, nextParentIDs) |
|||
} else { |
|||
return tags, nil |
|||
} |
|||
} |
|||
|
|||
func (r *tagRepository) List(ctx context.Context, full bool) ([]tag.Tag, error) { |
|||
var rows []mysqlgen.Tag |
|||
var err error |
|||
if full { |
|||
rows, err = r.q.ListTags(ctx) |
|||
} else { |
|||
rows, err = r.q.ListListedTags(ctx) |
|||
} |
|||
if err != nil { |
|||
if err == sql.ErrNoRows { |
|||
return []tag.Tag{}, nil |
|||
} |
|||
return nil, err |
|||
} |
|||
|
|||
res := make([]tag.Tag, 0, len(rows)) |
|||
for _, row := range rows { |
|||
res = append(res, tag.Tag{ |
|||
ID: row.ID, |
|||
ParentID: fromNullString(row.ParentID), |
|||
OwnerID: row.OwnerID, |
|||
Name: row.Name, |
|||
Kind: tag.Kind(row.Kind), |
|||
Description: row.Description, |
|||
Listed: row.Listed, |
|||
Secret: row.Secret, |
|||
}) |
|||
} |
|||
|
|||
return res, nil |
|||
} |
|||
|
|||
func (r *tagRepository) Save(ctx context.Context, tag tag.Tag) error { |
|||
return r.q.ReplaceTag(ctx, mysqlgen.ReplaceTagParams{ |
|||
ID: tag.ID, |
|||
ParentID: toNullString(tag.ParentID), |
|||
OwnerID: tag.OwnerID, |
|||
Name: tag.Name, |
|||
Kind: int(tag.Kind), |
|||
Description: tag.Description, |
|||
Listed: tag.Listed, |
|||
Secret: tag.Secret, |
|||
}) |
|||
} |
|||
|
|||
func (r *tagRepository) Delete(ctx context.Context, tag tag.Tag) error { |
|||
return r.q.DeleteTag(ctx, tag.ID) |
|||
} |
@ -0,0 +1,11 @@ |
|||
package tag |
|||
|
|||
import "context" |
|||
|
|||
type Repository interface { |
|||
FindOne(ctx context.Context, id string) (*Tag, error) |
|||
FindRecursive(ctx context.Context, id string) ([]Tag, error) |
|||
List(ctx context.Context, full bool) ([]Tag, error) |
|||
Save(ctx context.Context, tag Tag) error |
|||
Delete(ctx context.Context, tag Tag) error |
|||
} |
@ -0,0 +1,26 @@ |
|||
package tag |
|||
|
|||
import ( |
|||
"context" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/auth" |
|||
) |
|||
|
|||
type Service struct { |
|||
Repository Repository |
|||
Auth auth.Service |
|||
} |
|||
|
|||
func (s *Service) Find(ctx context.Context, id string) (*Node, error) { |
|||
tags, err := s.Repository.FindRecursive(ctx, id) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
|
|||
root := BuildForest(tags)[0].WithoutSecret(s.Auth.GetUser(ctx)) |
|||
if root == nil { |
|||
return nil, rpdata2.NotFound("Tag") |
|||
} |
|||
|
|||
return root, nil |
|||
} |
@ -0,0 +1,46 @@ |
|||
package tag |
|||
|
|||
type Tag struct { |
|||
ID string `json:"id"` |
|||
ParentID *string `json:"parentId"` |
|||
OwnerID string `json:"ownerId"` // The user who can change the tag
|
|||
Name string `json:"name"` |
|||
Kind Kind `json:"kind"` |
|||
Description string `json:"description"` |
|||
Listed bool `json:"listed"` // Whether to list it for search
|
|||
Secret bool `json:"secret"` // Only owners can see it
|
|||
} |
|||
|
|||
type Kind int |
|||
|
|||
const ( |
|||
// TKOrganization is an organization, faction or group
|
|||
TKOrganization = 1 << iota |
|||
// TKLocation is a settlement, building, planet, etc.... Icon: Pin
|
|||
TKLocation |
|||
// TKEvent is a specific event
|
|||
TKEvent |
|||
// TKPlot is a larger plot
|
|||
TKPlot |
|||
// TKSeries is a thread of content that's not exactly a plot/event
|
|||
TKSeries |
|||
) |
|||
|
|||
/* |
|||
L Aite |
|||
L Freedom Falls |
|||
LO Miner's Respite |
|||
L Derrai Union |
|||
L Derrai |
|||
L Aroste |
|||
L Aroste |
|||
P "The Collectors" |
|||
|
|||
P "Rellis Stuff" |
|||
E Rellis' Contract |
|||
E Familial Ties |
|||
E Skipping Work |
|||
E Sala's Backlog |
|||
S Anywhere but Here |
|||
|
|||
*/ |
@ -0,0 +1,110 @@ |
|||
package tag |
|||
|
|||
import ( |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/auth" |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/internal/genutils" |
|||
) |
|||
|
|||
type Node struct { |
|||
Tag |
|||
|
|||
Children []Node `json:"children"` |
|||
} |
|||
|
|||
func (n Node) WithoutSecret(reqUser *auth.UserInfo) *Node { |
|||
return n.filter(func(n Node) bool { |
|||
if !n.Secret { |
|||
return false |
|||
} |
|||
|
|||
return reqUser != nil && reqUser.HasIDOrPermission(n.OwnerID, "tag", "view_secret") |
|||
}) |
|||
} |
|||
|
|||
func (n Node) WithoutUnlisted() *Node { |
|||
return n.filter(func(n Node) bool { return n.Listed }) |
|||
} |
|||
|
|||
func (n Node) filter(cb func(n Node) bool) *Node { |
|||
if !cb(n) { |
|||
return nil |
|||
} |
|||
|
|||
filtered := make([]Node, 0, len(n.Children)) |
|||
for _, child := range n.Children { |
|||
pruned := child.filter(cb) |
|||
if pruned == nil { |
|||
continue |
|||
} |
|||
|
|||
filtered = append(filtered, *pruned) |
|||
} |
|||
|
|||
n.Children = filtered |
|||
return &n |
|||
} |
|||
|
|||
func (n Node) Less(n2 Node) bool { |
|||
if n.Kind < n2.Kind { |
|||
return true |
|||
} |
|||
|
|||
return n.Name < n2.Name |
|||
} |
|||
|
|||
func BuildForest(tags []Tag) []Node { |
|||
nodes := make(map[string]*Node) |
|||
for _, tag := range tags { |
|||
nodes[tag.ID] = &Node{Tag: tag, Children: []Node{}} |
|||
} |
|||
|
|||
children := make(map[string][]*Node, len(tags)) |
|||
for _, node := range nodes { |
|||
if node.ParentID != nil { |
|||
children[*node.ParentID] = append(children[*node.ParentID], node) |
|||
} |
|||
} |
|||
|
|||
cleared := make(map[string]bool) |
|||
for len(cleared) < len(nodes) { |
|||
anyCleared := false |
|||
for _, node := range nodes { |
|||
if cleared[node.ID] { |
|||
continue |
|||
} |
|||
|
|||
good := true |
|||
for _, child := range children[node.ID] { |
|||
if !cleared[child.ID] { |
|||
good = false |
|||
break |
|||
} |
|||
} |
|||
|
|||
if good { |
|||
genutils.SortSlice(node.Children) |
|||
|
|||
cleared[node.ID] = true |
|||
anyCleared = true |
|||
if node.ParentID != nil { |
|||
nodes[*node.ParentID].Children = append(nodes[*node.ParentID].Children, *node) |
|||
} |
|||
} |
|||
} |
|||
|
|||
if !anyCleared { |
|||
panic("deadlock in tag.BuildForest") |
|||
} |
|||
} |
|||
|
|||
res := make([]Node, 0, 8) |
|||
for _, node := range nodes { |
|||
if node.ParentID == nil { |
|||
res = append(res, *node) |
|||
} |
|||
} |
|||
|
|||
genutils.SortSlice(res) |
|||
|
|||
return res |
|||
} |
@ -0,0 +1,95 @@ |
|||
package tag |
|||
|
|||
import ( |
|||
"git.aiterp.net/rpdata2-take2/rpdata2/internal/genutils" |
|||
"github.com/stretchr/testify/assert" |
|||
"testing" |
|||
) |
|||
|
|||
func printNode(t *testing.T, prefix string, node Node) { |
|||
t.Log(prefix + "[" + node.ID + "] " + node.Name) |
|||
for _, child := range node.Children { |
|||
printNode(t, prefix+" ", child) |
|||
} |
|||
} |
|||
|
|||
func TestNode_Less(t *testing.T) { |
|||
n1 := Node{Tag: Tag{Kind: TKLocation | TKOrganization, Name: "Miner's Respite"}} |
|||
n2 := Node{Tag: Tag{Kind: TKLocation | TKOrganization, Name: "Litae's Grace"}} |
|||
n3 := Node{Tag: Tag{Kind: TKOrganization, Name: "Redrock Agency"}} |
|||
n4 := Node{Tag: Tag{Kind: TKLocation, Name: "Redrock HQ"}} |
|||
|
|||
assert.True(t, n2.Less(n1)) |
|||
assert.True(t, !n1.Less(n2)) |
|||
assert.True(t, n3.Less(n4)) |
|||
} |
|||
|
|||
func TestNode_WithoutUnlisted(t *testing.T) { |
|||
before := Node{Tag{Name: "A", Listed: true}, []Node{ |
|||
{Tag{Name: "B", Listed: true}, []Node{ |
|||
{Tag{Name: "D", Listed: false}, []Node{ |
|||
{Tag{Name: "F", Listed: true}, []Node{}}, |
|||
}}, |
|||
{Tag{Name: "E", Secret: true, Listed: true}, []Node{ |
|||
{Tag{Name: "G", Listed: true}, []Node{}}, |
|||
}}, |
|||
}}, |
|||
{Tag{Name: "C", Listed: true}, []Node{}}, |
|||
}} |
|||
|
|||
after := Node{Tag{Name: "A", Listed: true}, []Node{ |
|||
{Tag{Name: "B", Listed: true}, []Node{ |
|||
{Tag{Name: "E", Secret: true, Listed: true}, []Node{ |
|||
{Tag{Name: "G", Listed: true}, []Node{}}, |
|||
}}, |
|||
}}, |
|||
{Tag{Name: "C", Listed: true}, []Node{}}, |
|||
}} |
|||
|
|||
assert.Equal(t, &after, before.WithoutUnlisted()) |
|||
} |
|||
|
|||
func TestBuildForest(t *testing.T) { |
|||
tags := []Tag{ |
|||
{ID: "T0", ParentID: genutils.Ptr("T7"), Name: "Root"}, |
|||
{ID: "T1", ParentID: genutils.Ptr("T0"), Name: "Stuff"}, |
|||
{ID: "T2", ParentID: genutils.Ptr("T0"), Name: "Things"}, |
|||
{ID: "T3", Name: "Second Root"}, |
|||
{ID: "T4", ParentID: genutils.Ptr("T2"), Name: "Items"}, |
|||
{ID: "T5", ParentID: genutils.Ptr("T1"), Name: "Objects"}, |
|||
{ID: "T6", ParentID: genutils.Ptr("T4"), Name: "Thingies"}, |
|||
{ID: "T7", Name: "Real Root"}, |
|||
{ID: "T8", ParentID: genutils.Ptr("T3"), Name: "Another Leaf"}, |
|||
{ID: "T9", ParentID: genutils.Ptr("T5"), Name: "Branch"}, |
|||
{ID: "TA", ParentID: genutils.Ptr("T9"), Name: "Leafy Leaf"}, |
|||
} |
|||
|
|||
nodes := BuildForest(tags) |
|||
for _, node := range nodes { |
|||
printNode(t, "", node) |
|||
} |
|||
|
|||
expected := []Node{ |
|||
{tags[7], []Node{ |
|||
{tags[0], []Node{ |
|||
{tags[1], []Node{ |
|||
{tags[5], []Node{ |
|||
{tags[9], []Node{ |
|||
{tags[10], []Node{}}, |
|||
}}, |
|||
}}, |
|||
}}, |
|||
{tags[2], []Node{ |
|||
{tags[4], []Node{ |
|||
{tags[6], []Node{}}, |
|||
}}, |
|||
}}, |
|||
}}, |
|||
}}, |
|||
{tags[3], []Node{ |
|||
{tags[8], []Node{}}, |
|||
}}, |
|||
} |
|||
|
|||
assert.Equal(t, expected, nodes) |
|||
} |
@ -0,0 +1,19 @@ |
|||
package tag |
|||
|
|||
import "git.aiterp.net/rpdata2-take2/rpdata2/internal/genutils" |
|||
|
|||
type Update struct { |
|||
ParentID *string `json:"parentId"` |
|||
Name *string `json:"name"` |
|||
Kind *Kind `json:"kind"` |
|||
Listed *bool `json:"listed"` |
|||
Secret *bool `json:"secret"` // Only owners can see it
|
|||
} |
|||
|
|||
func (t *Tag) ApplyUpdate(update *Update) { |
|||
genutils.ApplyUpdateNilZero(&t.ParentID, update.ParentID) |
|||
genutils.ApplyUpdate(&t.Name, update.Name) |
|||
genutils.ApplyUpdate(&t.Kind, update.Kind) |
|||
genutils.ApplyUpdate(&t.Listed, update.Listed) |
|||
genutils.ApplyUpdate(&t.Secret, update.Secret) |
|||
} |
Write
Preview
Loading…
Cancel
Save
Reference in new issue