database/permissions.go at master · TicketsBot-cloud/database · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
package database

import (
	"context"
	"github.com/jackc/pgx/v4"
	"github.com/jackc/pgx/v4/pgxpool"
)

type Permissions struct {
	*pgxpool.Pool
}

func newPermissions(db *pgxpool.Pool) *Permissions {
	return &Permissions{
		db,
	}
}

func (p Permissions) Schema() string {
	return `
CREATE TABLE IF NOT EXISTS permissions(
	"guild_id" int8 NOT NULL,
	"user_id" int8 NOT NULL,
	"support" bool NOT NULL,
	"admin" bool NOT NULL,
	PRIMARY KEY("guild_id", "user_id")
);
CREATE INDEX IF NOT EXISTS permissions_guild_id ON permissions("guild_id");
`
}

func (p *Permissions) IsSupport(ctx context.Context, guildId, userId uint64) (support bool, e error) {
	var admin bool

	if err := p.QueryRow(ctx, `SELECT "support", "admin" from permissions WHERE "guild_id" = $1 AND "user_id" = $2;`, guildId, userId).Scan(&support, &admin); err != nil && err != pgx.ErrNoRows {
		e = err
	}

	if admin {
		support = true
	}

	return
}

func (p *Permissions) IsAdmin(ctx context.Context, guildId, userId uint64) (admin bool, e error) {
	if err := p.QueryRow(ctx, `SELECT "admin" from permissions WHERE "guild_id" = $1 AND "user_id" = $2;`, guildId, userId).Scan(&admin); err != nil && err != pgx.ErrNoRows {
		e = err
	}

	return
}

func (p *Permissions) GetAdmins(ctx context.Context, guildId uint64) (admins []uint64, e error) {
	rows, err := p.Query(ctx, `SELECT "user_id" from permissions WHERE "guild_id" = $1 AND "admin" = true;`, guildId)
	defer rows.Close()
	if err != nil && err != pgx.ErrNoRows {
		e = err
	}

	for rows.Next() {
		var userId uint64
		if err := rows.Scan(&userId); err != nil {
			e = err
			continue
		}

		admins = append(admins, userId)
	}

	return
}

func (p *Permissions) GetSupport(ctx context.Context, guildId uint64) (support []uint64, e error) {
	rows, err := p.Query(ctx, `SELECT "user_id" from permissions WHERE "guild_id" = $1 AND ("admin" = true OR "support" = true);`, guildId)
	defer rows.Close()
	if err != nil && err != pgx.ErrNoRows {
		e = err
	}

	for rows.Next() {
		var userId uint64
		if err := rows.Scan(&userId); err != nil {
			e = err
			continue
		}

		support = append(support, userId)
	}

	return
}

func (p *Permissions) GetSupportOnly(ctx context.Context, guildId uint64) (support []uint64, e error) {
	query := `SELECT "user_id" from permissions WHERE "guild_id" = $1 AND "admin" = false AND "support" = true;`
	rows, err := p.Query(ctx, query, guildId)
	defer rows.Close()
	if err != nil && err != pgx.ErrNoRows {
		e = err
	}

	for rows.Next() {
		var userId uint64
		if err := rows.Scan(&userId); err != nil {
			e = err
			continue
		}

		support = append(support, userId)
	}

	return
}

func (p *Permissions) AddAdmin(ctx context.Context, guildId, userId uint64) (err error) {
	query := `INSERT INTO permissions("guild_id", "user_id", "support", "admin") VALUES($1, $2, true, true) ON CONFLICT("guild_id", "user_id") DO UPDATE SET "admin" = true, "support" = true;`
	_, err = p.Exec(ctx, query, guildId, userId)
	return
}

func (p *Permissions) AddSupport(ctx context.Context, guildId, userId uint64) (err error) {
	query := `INSERT INTO permissions("guild_id", "user_id", "support", "admin") VALUES($1, $2, true, false) ON CONFLICT("guild_id", "user_id") DO UPDATE SET "admin" = false, "support" = true;`
	_, err = p.Exec(ctx, query, guildId, userId)
	return
}

func (p *Permissions) RemoveAdmin(ctx context.Context, guildId, userId uint64) (err error) {
	query := `UPDATE permissions SET "admin" = false WHERE "guild_id" = $1 AND "user_id" = $2;`
	_, err = p.Exec(ctx, query, guildId, userId)
	return
}

func (p *Permissions) RemoveSupport(ctx context.Context, guildId, userId uint64) (err error) {
	query := `UPDATE permissions SET "admin" = false, "support" = false WHERE "guild_id" = $1 AND "user_id" = $2;`
	_, err = p.Exec(ctx, query, guildId, userId)
	return
}