From 78a38f508e7b9d2389c9c422c2d164b9059b03f8 Mon Sep 17 00:00:00 2001
From: Kyren223 <Kyren223@proton.me>
Date: Sun, 1 Dec 2024 10:09:21 +0200
Subject: Implemented frequency creation, BROKEN right now

---
 internal/data/models.go             | 14 +++++++-------
 internal/data/users_networks.sql.go | 31 +++++++++++++++++++++++++++++--
 2 files changed, 36 insertions(+), 9 deletions(-)

(limited to 'internal/data')

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,
-- 
cgit v1.3.1