diff options
| author | Kyren223 <Kyren223@proton.me> | 2024-12-01 10:09:21 +0200 |
|---|---|---|
| committer | Kyren223 <Kyren223@proton.me> | 2024-12-01 10:09:21 +0200 |
| commit | 78a38f508e7b9d2389c9c422c2d164b9059b03f8 (patch) | |
| tree | 67550791d99597eaa6ab50334ccc880f06a8de17 /internal/data | |
| parent | 94166da109cf19ce48311d07aa67875fcb85f178 (diff) | |
Implemented frequency creation, BROKEN right now
Diffstat (limited to 'internal/data')
| -rw-r--r-- | internal/data/models.go | 14 | ||||
| -rw-r--r-- | internal/data/users_networks.sql.go | 31 |
2 files changed, 36 insertions, 9 deletions
diff --git a/internal/data/models.go b/internal/data/models.go index 566d2ef..f53d726 100644 --- a/internal/data/models.go +++ b/internal/data/models.go @@ -51,13 +51,7 @@ type UserBlockedUser struct { BlockedUserID snowflake.ID } -type UserTrustedUser struct { - TrusterUserID snowflake.ID - TrustedUserID snowflake.ID - TrustedPublicKey []byte -} - -type UsersNetwork struct { +type UserNetwork struct { UserID snowflake.ID NetworkID snowflake.ID JoinedAt string @@ -68,3 +62,9 @@ type UsersNetwork struct { BanReason *string Position *int64 } + +type UserTrustedUser struct { + TrusterUserID snowflake.ID + TrustedUserID snowflake.ID + TrustedPublicKey []byte +} diff --git a/internal/data/users_networks.sql.go b/internal/data/users_networks.sql.go index 4c1d84f..42733da 100644 --- a/internal/data/users_networks.sql.go +++ b/internal/data/users_networks.sql.go @@ -107,6 +107,33 @@ func (q *Queries) GetNetworkMembers(ctx context.Context, networkID snowflake.ID) return items, nil } +const getUserNetwork = `-- name: GetUserNetwork :one +SELECT user_id, network_id, joined_at, is_member, is_admin, is_muted, is_banned, ban_reason, position FROM users_networks +WHERE user_id = ? AND network_id = ? +` + +type GetUserNetworkParams struct { + UserID snowflake.ID + NetworkID snowflake.ID +} + +func (q *Queries) GetUserNetwork(ctx context.Context, arg GetUserNetworkParams) (UserNetwork, error) { + row := q.db.QueryRowContext(ctx, getUserNetwork, arg.UserID, arg.NetworkID) + var i UserNetwork + err := row.Scan( + &i.UserID, + &i.NetworkID, + &i.JoinedAt, + &i.IsMember, + &i.IsAdmin, + &i.IsMuted, + &i.IsBanned, + &i.BanReason, + &i.Position, + ) + return i, err +} + const getUserNetworks = `-- name: GetUserNetworks :many SELECT networks.id, networks.owner_id, networks.name, networks.icon, networks.bg_hex_color, networks.fg_hex_color, networks.is_public, users_networks.position FROM networks JOIN users_networks ON networks.id = users_networks.network_id @@ -183,7 +210,7 @@ type SetNetworkUserParams struct { BanReason *string } -func (q *Queries) SetNetworkUser(ctx context.Context, arg SetNetworkUserParams) (UsersNetwork, error) { +func (q *Queries) SetNetworkUser(ctx context.Context, arg SetNetworkUserParams) (UserNetwork, error) { row := q.db.QueryRowContext(ctx, setNetworkUser, arg.UserID, arg.NetworkID, @@ -193,7 +220,7 @@ func (q *Queries) SetNetworkUser(ctx context.Context, arg SetNetworkUserParams) arg.IsBanned, arg.BanReason, ) - var i UsersNetwork + var i UserNetwork err := row.Scan( &i.UserID, &i.NetworkID, |