gtsocial-umbx

instance.go (5532B)

---

 // GoToSocial
 // Copyright (C) GoToSocial Authors admin@gotosocial.org
 // SPDX-License-Identifier: AGPL-3.0-or-later
 //
 // 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/config"
 	"github.com/superseriousbusiness/gotosocial/internal/db"
 	"github.com/superseriousbusiness/gotosocial/internal/gtsmodel"
 	"github.com/uptrace/bun"
 )
 
 type instanceDB struct {
 	conn *DBConn
 }
 
 func (i *instanceDB) CountInstanceUsers(ctx context.Context, domain string) (int, db.Error) {
 	q := i.conn.
 		NewSelect().
 		TableExpr("? AS ?", bun.Ident("accounts"), bun.Ident("account")).
 		Column("account.id").
 		Where("? != ?", bun.Ident("account.username"), domain).
 		Where("? IS NULL", bun.Ident("account.suspended_at"))
 
 	if domain == config.GetHost() || domain == config.GetAccountDomain() {
 		// if the domain is *this* domain, just count where the domain field is null
 		q = q.WhereGroup(" AND ", whereEmptyOrNull("account.domain"))
 	} else {
 		q = q.Where("? = ?", bun.Ident("account.domain"), domain)
 	}
 
 	count, err := q.Count(ctx)
 	if err != nil {
 		return 0, i.conn.ProcessError(err)
 	}
 	return count, nil
 }
 
 func (i *instanceDB) CountInstanceStatuses(ctx context.Context, domain string) (int, db.Error) {
 	q := i.conn.
 		NewSelect().
 		TableExpr("? AS ?", bun.Ident("statuses"), bun.Ident("status"))
 
 	if domain == config.GetHost() || domain == config.GetAccountDomain() {
 		// if the domain is *this* domain, just count where local is true
 		q = q.Where("? = ?", bun.Ident("status.local"), true)
 	} else {
 		// join on the domain of the account
 		q = q.
 			Join("JOIN ? AS ? ON ? = ?", bun.Ident("accounts"), bun.Ident("account"), bun.Ident("account.id"), bun.Ident("status.account_id")).
 			Where("? = ?", bun.Ident("account.domain"), domain)
 	}
 
 	count, err := q.Count(ctx)
 	if err != nil {
 		return 0, i.conn.ProcessError(err)
 	}
 	return count, nil
 }
 
 func (i *instanceDB) CountInstanceDomains(ctx context.Context, domain string) (int, db.Error) {
 	q := i.conn.
 		NewSelect().
 		TableExpr("? AS ?", bun.Ident("instances"), bun.Ident("instance"))
 
 	if domain == config.GetHost() {
 		// if the domain is *this* domain, just count other instances it knows about
 		// exclude domains that are blocked
 		q = q.
 			Where("? != ?", bun.Ident("instance.domain"), domain).
 			Where("? IS NULL", bun.Ident("instance.suspended_at"))
 	} else {
 		// TODO: implement federated domain counting properly for remote domains
 		return 0, nil
 	}
 
 	count, err := q.Count(ctx)
 	if err != nil {
 		return 0, i.conn.ProcessError(err)
 	}
 	return count, nil
 }
 
 func (i *instanceDB) GetInstance(ctx context.Context, domain string) (*gtsmodel.Instance, db.Error) {
 	instance := &gtsmodel.Instance{}
 
 	if err := i.conn.
 		NewSelect().
 		Model(instance).
 		Where("? = ?", bun.Ident("instance.domain"), domain).
 		Scan(ctx); err != nil {
 		return nil, i.conn.ProcessError(err)
 	}
 
 	return instance, nil
 }
 
 func (i *instanceDB) GetInstancePeers(ctx context.Context, includeSuspended bool) ([]*gtsmodel.Instance, db.Error) {
 	instances := []*gtsmodel.Instance{}
 
 	q := i.conn.
 		NewSelect().
 		Model(&instances).
 		Where("? != ?", bun.Ident("instance.domain"), config.GetHost())
 
 	if !includeSuspended {
 		q = q.Where("? IS NULL", bun.Ident("instance.suspended_at"))
 	}
 
 	if err := q.Scan(ctx); err != nil {
 		return nil, i.conn.ProcessError(err)
 	}
 
 	return instances, nil
 }
 
 func (i *instanceDB) GetInstanceAccounts(ctx context.Context, domain string, maxID string, limit int) ([]*gtsmodel.Account, db.Error) {
 	accounts := []*gtsmodel.Account{}
 
 	q := i.conn.NewSelect().
 		Model(&accounts).
 		Where("? = ?", bun.Ident("account.domain"), domain).
 		Order("account.id DESC")
 
 	if maxID != "" {
 		q = q.Where("? < ?", bun.Ident("account.id"), maxID)
 	}
 
 	if limit > 0 {
 		q = q.Limit(limit)
 	}
 
 	if err := q.Scan(ctx); err != nil {
 		return nil, i.conn.ProcessError(err)
 	}
 
 	if len(accounts) == 0 {
 		return nil, db.ErrNoEntries
 	}
 
 	return accounts, nil
 }
 
 func (i *instanceDB) GetInstanceModeratorAddresses(ctx context.Context) ([]string, db.Error) {
 	addresses := []string{}
 
 	// Select email addresses of approved, confirmed,
 	// and enabled moderators or admins.
 
 	q := i.conn.
 		NewSelect().
 		TableExpr("? AS ?", bun.Ident("users"), bun.Ident("user")).
 		Column("user.email").
 		Where("? = ?", bun.Ident("user.approved"), true).
 		Where("? IS NOT NULL", bun.Ident("user.confirmed_at")).
 		Where("? = ?", bun.Ident("user.disabled"), false).
 		WhereGroup(" AND ", func(q *bun.SelectQuery) *bun.SelectQuery {
 			return q.
 				Where("? = ?", bun.Ident("user.moderator"), true).
 				WhereOr("? = ?", bun.Ident("user.admin"), true)
 		}).
 		OrderExpr("? ASC", bun.Ident("user.email"))
 
 	if err := q.Scan(ctx, &addresses); err != nil {
 		return nil, i.conn.ProcessError(err)
 	}
 
 	if len(addresses) == 0 {
 		return nil, db.ErrNoEntries
 	}
 
 	return addresses, nil
 }