idk how to plan this out
This commit is contained in:
gitGnome
@@ -72,86 +72,6 @@ func Init(ctx context.Context) {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hubs (
|
||||
id UUID PRIMARY KEY,
|
||||
creator_id UUID NOT NULL REFERENCES users(id),
|
||||
name TEXT NOT NULL,
|
||||
allow_user_color BOOLEAN NOT NULL DEFAULT TRUE,
|
||||
rgba BIGINT NOT NULL DEFAULT 0 CHECK (rgba BETWEEN 0 AND 4294967295),
|
||||
created_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hub_roles (
|
||||
hub_id UUID NOT NULL REFERENCES hubs(id) ON DELETE CASCADE,
|
||||
role_id SMALLINT NOT NULL,
|
||||
name TEXT NOT NULL,
|
||||
permissions INTEGER NOT NULL DEFAULT 0,
|
||||
rgba BIGINT NOT NULL DEFAULT 0 CHECK (rgba BETWEEN 0 AND 4294967295),
|
||||
PRIMARY KEY (hub_id, role_id)
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hub_channel_groups (
|
||||
channel_group_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||
hub_id UUID NOT NULL REFERENCES hubs(id) ON DELETE CASCADE,
|
||||
name TEXT NOT NULL,
|
||||
rgba BIGINT NOT NULL DEFAULT 0 CHECK (rgba BETWEEN 0 AND 4294967295),
|
||||
position SMALLINT NOT NULL DEFAULT 0
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hub_channels (
|
||||
channel_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||||
hub_id UUID NOT NULL REFERENCES hubs(id) ON DELETE CASCADE,
|
||||
name TEXT NOT NULL,
|
||||
parent_group_id UUID REFERENCES hub_channel_groups(channel_group_id) ON DELETE CASCADE,
|
||||
position SMALLINT NOT NULL DEFAULT 0
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hub_users (
|
||||
hub_id UUID NOT NULL REFERENCES hubs(id) ON DELETE CASCADE,
|
||||
user_id UUID NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||||
name TEXT NOT NULL,
|
||||
created_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||||
PRIMARY KEY (hub_id, user_id)
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = dbConn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS hub_user_roles (
|
||||
hub_id UUID NOT NULL,
|
||||
user_id UUID NOT NULL,
|
||||
role_id SMALLINT NOT NULL,
|
||||
PRIMARY KEY (hub_id, user_id, role_id),
|
||||
FOREIGN KEY (hub_id, user_id) REFERENCES hub_users(hub_id, user_id) ON DELETE CASCADE,
|
||||
FOREIGN KEY (hub_id, role_id) REFERENCES hub_roles(hub_id, role_id) ON DELETE CASCADE
|
||||
)
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
func UserSave(ctx context.Context, user *types.User) error {
|
||||
@@ -277,7 +197,7 @@ func ConnectionsGetBelongingToUser(ctx context.Context, user *types.User) error
|
||||
}
|
||||
|
||||
for rows.Next() {
|
||||
conn := types.NewConnection()
|
||||
conn := types.CreateConn()
|
||||
if err = rows.Scan(&conn.Id, &conn.RequestorId, &conn.RecipientId, &conn.State, &conn.CreatedAt); err != nil {
|
||||
return fmt.Errorf("scanning connection row: %w", err)
|
||||
}
|
||||
@@ -335,182 +255,3 @@ func ConnectionGetMessagesBefore(ctx context.Context, before time.Time, connecti
|
||||
return messages, rows.Err()
|
||||
}
|
||||
|
||||
func HubSave(ctx context.Context, hub *types.Hub) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
INSERT INTO hubs (id, creator_id, name, allow_user_color, rgba, created_at) VALUES ($1, $2, $3, $4, $5, $6)
|
||||
`, hub.Id, hub.Creator, hub.Name, hub.AllowUserColor, convertions.RgbaToUint32(hub.Color), hub.CreatedAt)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubDelete(ctx context.Context, hub *types.Hub) error {
|
||||
_, err := dbConn.Exec(ctx, `DELETE FROM hubs WHERE id = $1`, hub.Id)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubGet(ctx context.Context, hub *types.Hub) error {
|
||||
var rgba uint32
|
||||
err := dbConn.QueryRow(ctx, `
|
||||
SELECT creator_id, name, allow_user_color, rgba, created_at FROM hubs WHERE id = $1
|
||||
`, hub.Id).Scan(&hub.Creator, &hub.Name, &hub.AllowUserColor, &rgba, &hub.CreatedAt)
|
||||
if err == nil {
|
||||
hub.Color = convertions.Uint32ToRgba(rgba)
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
func HubsGetBelongingToUser(ctx context.Context, userId uuid.UUID) ([]*types.Hub, error) {
|
||||
rows, err := dbConn.Query(ctx, `
|
||||
SELECT h.id, h.creator_id, h.name, h.allow_user_color, h.rgba, h.created_at
|
||||
FROM hubs h
|
||||
JOIN hub_users hu ON h.id = hu.hub_id
|
||||
WHERE hu.user_id = $1
|
||||
`, userId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var hubs []*types.Hub
|
||||
for rows.Next() {
|
||||
hub := types.NewHub()
|
||||
var rgba uint32
|
||||
if err = rows.Scan(&hub.Id, &hub.Creator, &hub.Name, &hub.AllowUserColor, &rgba, &hub.CreatedAt); err != nil {
|
||||
return nil, fmt.Errorf("scanning hub row: %w", err)
|
||||
}
|
||||
hub.Color = convertions.Uint32ToRgba(rgba)
|
||||
hubs = append(hubs, hub)
|
||||
}
|
||||
return hubs, rows.Err()
|
||||
}
|
||||
|
||||
func HubGetUsers(ctx context.Context, hub *types.Hub) error {
|
||||
rows, err := dbConn.Query(ctx, `
|
||||
SELECT hu.user_id, hu.name, hu.created_at, COALESCE(array_agg(hur.role_id) FILTER (WHERE hur.role_id IS NOT NULL), '{}')
|
||||
FROM hub_users hu
|
||||
LEFT JOIN hub_user_roles hur ON hu.hub_id = hur.hub_id AND hu.user_id = hur.user_id
|
||||
WHERE hu.hub_id = $1
|
||||
GROUP BY hu.user_id, hu.name, hu.created_at
|
||||
`, hub.Id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
hu := &types.HubUser{}
|
||||
var roles []int16
|
||||
if err = rows.Scan(&hu.Id, &hu.Username, &hu.CreatedAt, &roles); err != nil {
|
||||
return fmt.Errorf("scanning hub user row: %w", err)
|
||||
}
|
||||
hu.Roles = make([]uint8, len(roles))
|
||||
for i, r := range roles {
|
||||
hu.Roles[i] = uint8(r)
|
||||
}
|
||||
hub.Users[hu.Id] = hu
|
||||
}
|
||||
return rows.Err()
|
||||
}
|
||||
|
||||
func HubGetRoles(ctx context.Context, hub *types.Hub) error {
|
||||
rows, err := dbConn.Query(ctx, `
|
||||
SELECT role_id, name, permissions, rgba FROM hub_roles WHERE hub_id = $1
|
||||
`, hub.Id)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
for rows.Next() {
|
||||
hr := &types.HubRole{}
|
||||
var rgba uint32
|
||||
if err = rows.Scan(&hr.Id, &hr.Name, &hr.RolePermission, &rgba); err != nil {
|
||||
return fmt.Errorf("scanning hub role row: %w", err)
|
||||
}
|
||||
hr.Color = convertions.Uint32ToRgba(rgba)
|
||||
hub.Roles[hr.Id] = hr
|
||||
}
|
||||
return rows.Err()
|
||||
}
|
||||
|
||||
func GetWholeHub(ctx context.Context, hub *types.Hub) error {
|
||||
if err := HubGet(ctx, hub); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := HubGetUsers(ctx, hub); err != nil {
|
||||
return err
|
||||
}
|
||||
return HubGetRoles(ctx, hub)
|
||||
}
|
||||
|
||||
func HubUserAdd(ctx context.Context, hubId uuid.UUID, hu *types.HubUser) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
INSERT INTO hub_users (hub_id, user_id, name, created_at) VALUES ($1, $2, $3, $4)
|
||||
`, hubId, hu.Id, hu.Username, hu.CreatedAt)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubUserRemove(ctx context.Context, hubId uuid.UUID, userId uuid.UUID) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
DELETE FROM hub_users WHERE hub_id = $1 AND user_id = $2
|
||||
`, hubId, userId)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubUserRoleAdd(ctx context.Context, hubId uuid.UUID, userId uuid.UUID, roleId uint8) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
INSERT INTO hub_user_roles (hub_id, user_id, role_id) VALUES ($1, $2, $3)
|
||||
`, hubId, userId, roleId)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubUserRoleRemove(ctx context.Context, hubId uuid.UUID, userId uuid.UUID, roleId uint8) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
DELETE FROM hub_user_roles WHERE hub_id = $1 AND user_id = $2 AND role_id = $3
|
||||
`, hubId, userId, roleId)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubRoleSave(ctx context.Context, hubId uuid.UUID, role *types.HubRole) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
INSERT INTO hub_roles (hub_id, role_id, name, permissions, rgba) VALUES ($1, $2, $3, $4, $5)
|
||||
`, hubId, role.Id, role.Name, role.RolePermission, convertions.RgbaToUint32(role.Color))
|
||||
return err
|
||||
}
|
||||
|
||||
func HubRoleDelete(ctx context.Context, hubId uuid.UUID, roleId uint8) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
DELETE FROM hub_roles WHERE hub_id = $1 AND role_id = $2
|
||||
`, hubId, roleId)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubRoleUpdate(ctx context.Context, hubId uuid.UUID, role *types.HubRole) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
UPDATE hub_roles SET name = $1, permissions = $2, rgba = $3 WHERE hub_id = $4 AND role_id = $5
|
||||
`, role.Name, role.RolePermission, convertions.RgbaToUint32(role.Color), hubId, role.Id)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubChannelGroupSave(ctx context.Context, hubId uuid.UUID, cg *types.HubChannelGroup) error {
|
||||
_, err := dbConn.Exec(ctx, `
|
||||
INSERT INTO hub_channel_groups (channel_group_id, hub_id, name, rgba, position) VALUES ($1, $2, $3, $4, $5)
|
||||
`, cg.Id, hubId, cg.Name, convertions.RgbaToUint32(cg.Color), cg.Position)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubChannelGroupDelete(ctx context.Context, cgId uuid.UUID) error {
|
||||
_, err := dbConn.Exec(ctx, `DELETE FROM hub_channel_groups WHERE channel_group_id = $1`, cgId)
|
||||
return err
|
||||
}
|
||||
|
||||
func HubChannelSave(ctx context.Context, hubId uuid.UUID, ch *types.HubChannel) error {
|
||||
return dbConn.QueryRow(ctx, `
|
||||
INSERT INTO hub_channels (hub_id, name, parent_group_id, position) VALUES ($1, $2, $3, $4)
|
||||
RETURNING channel_id
|
||||
`, hubId, ch.Name, ch.ParentGroupId, ch.Position).Scan(&ch.Id)
|
||||
}
|
||||
|
||||
func HubChannelDelete(ctx context.Context, chId uuid.UUID) error {
|
||||
_, err := dbConn.Exec(ctx, `DELETE FROM hub_channels WHERE channel_id = $1`, chId)
|
||||
return err
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user