summaryrefslogtreecommitdiff
path: root/internal/db
diff options
context:
space:
mode:
Diffstat (limited to 'internal/db')
-rw-r--r--internal/db/bundb/bundb.go4
-rw-r--r--internal/db/bundb/emoji.go46
-rw-r--r--internal/db/bundb/emoji_test.go42
-rw-r--r--internal/db/db.go1
-rw-r--r--internal/db/emoji.go31
5 files changed, 124 insertions, 0 deletions
diff --git a/internal/db/bundb/bundb.go b/internal/db/bundb/bundb.go
index a79e7e4f4..f458132a1 100644
--- a/internal/db/bundb/bundb.go
+++ b/internal/db/bundb/bundb.go
@@ -77,6 +77,7 @@ type bunDBService struct {
db.Admin
db.Basic
db.Domain
+ db.Emoji
db.Instance
db.Media
db.Mention
@@ -168,6 +169,9 @@ func NewBunDBService(ctx context.Context) (db.DB, error) {
Domain: &domainDB{
conn: conn,
},
+ Emoji: &emojiDB{
+ conn: conn,
+ },
Instance: &instanceDB{
conn: conn,
},
diff --git a/internal/db/bundb/emoji.go b/internal/db/bundb/emoji.go
new file mode 100644
index 000000000..55bc71e1e
--- /dev/null
+++ b/internal/db/bundb/emoji.go
@@ -0,0 +1,46 @@
+/*
+ GoToSocial
+ Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org
+
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU Affero General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Affero General Public License for more details.
+
+ You should have received a copy of the GNU Affero General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package bundb
+
+import (
+ "context"
+
+ "github.com/superseriousbusiness/gotosocial/internal/db"
+ "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+)
+
+type emojiDB struct {
+ conn *DBConn
+}
+
+func (e emojiDB) GetCustomEmojis(ctx context.Context) ([]*gtsmodel.Emoji, db.Error) {
+ emojis := []*gtsmodel.Emoji{}
+
+ q := e.conn.
+ NewSelect().
+ Model(&emojis).
+ Where("visible_in_picker = true").
+ Where("disabled = false").
+ Order("shortcode ASC")
+
+ if err := q.Scan(ctx); err != nil {
+ return nil, e.conn.ProcessError(err)
+ }
+ return emojis, nil
+}
diff --git a/internal/db/bundb/emoji_test.go b/internal/db/bundb/emoji_test.go
new file mode 100644
index 000000000..0a1546d91
--- /dev/null
+++ b/internal/db/bundb/emoji_test.go
@@ -0,0 +1,42 @@
+/*
+ GoToSocial
+ Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org
+
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU Affero General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Affero General Public License for more details.
+
+ You should have received a copy of the GNU Affero General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package bundb_test
+
+import (
+ "context"
+ "testing"
+
+ "github.com/stretchr/testify/suite"
+)
+
+type EmojiTestSuite struct {
+ BunDBStandardTestSuite
+}
+
+func (suite *EmojiTestSuite) TestGetCustomEmojis() {
+ emojis, err := suite.db.GetCustomEmojis(context.Background())
+
+ suite.NoError(err)
+ suite.Equal(1, len(emojis))
+ suite.Equal("rainbow", emojis[0].Shortcode)
+}
+
+func TestEmojiTestSuite(t *testing.T) {
+ suite.Run(t, new(EmojiTestSuite))
+}
diff --git a/internal/db/db.go b/internal/db/db.go
index 24205542c..57549f588 100644
--- a/internal/db/db.go
+++ b/internal/db/db.go
@@ -35,6 +35,7 @@ type DB interface {
Admin
Basic
Domain
+ Emoji
Instance
Media
Mention
diff --git a/internal/db/emoji.go b/internal/db/emoji.go
new file mode 100644
index 000000000..f2694b460
--- /dev/null
+++ b/internal/db/emoji.go
@@ -0,0 +1,31 @@
+/*
+ GoToSocial
+ Copyright (C) 2021-2022 GoToSocial Authors admin@gotosocial.org
+
+ This program is free software: you can redistribute it and/or modify
+ it under the terms of the GNU Affero General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ This program is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU Affero General Public License for more details.
+
+ You should have received a copy of the GNU Affero General Public License
+ along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+package db
+
+import (
+ "context"
+
+ "github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
+)
+
+// Emoji contains functions for getting emoji in the database.
+type Emoji interface {
+ // GetCustomEmojis gets all custom emoji for the instance
+ GetCustomEmojis(ctx context.Context) ([]*gtsmodel.Emoji, Error)
+}