| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | /* | 
					
						
							|  |  |  |    GoToSocial | 
					
						
							|  |  |  |    Copyright (C) 2021 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" | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	"crypto/rand" | 
					
						
							|  |  |  | 	"crypto/rsa" | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	"errors" | 
					
						
							|  |  |  | 	"fmt" | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	"net" | 
					
						
							|  |  |  | 	"net/mail" | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	"regexp" | 
					
						
							| 
									
										
										
										
											2021-03-03 18:12:02 +01:00
										 |  |  | 	"strings" | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	"time" | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | 	"github.com/go-fed/activity/pub" | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	"github.com/go-pg/pg/extra/pgdebug" | 
					
						
							|  |  |  | 	"github.com/go-pg/pg/v10" | 
					
						
							|  |  |  | 	"github.com/go-pg/pg/v10/orm" | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	"github.com/sirupsen/logrus" | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/internal/config" | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/internal/db/gtsmodel" | 
					
						
							|  |  |  | 	mastotypes "github.com/superseriousbusiness/gotosocial/internal/mastotypes/mastomodel" | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	"github.com/superseriousbusiness/gotosocial/internal/util" | 
					
						
							|  |  |  | 	"golang.org/x/crypto/bcrypt" | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | // postgresService satisfies the DB interface | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | type postgresService struct { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	config       *config.Config | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | 	conn         *pg.DB | 
					
						
							|  |  |  | 	log          *logrus.Entry | 
					
						
							|  |  |  | 	cancel       context.CancelFunc | 
					
						
							|  |  |  | 	federationDB pub.Database | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // newPostgresService returns a postgresService derived from the provided config, which implements the go-fed DB interface. | 
					
						
							|  |  |  | // Under the hood, it uses https://github.com/go-pg/pg to create and maintain a database connection. | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | func newPostgresService(ctx context.Context, c *config.Config, log *logrus.Entry) (DB, error) { | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	opts, err := derivePGOptions(c) | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("could not create postgres service: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	log.Debugf("using pg options: %+v", opts) | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	readyChan := make(chan interface{}) | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	opts.OnConnect = func(ctx context.Context, c *pg.Conn) error { | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 		close(readyChan) | 
					
						
							|  |  |  | 		return nil | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// create a connection | 
					
						
							|  |  |  | 	pgCtx, cancel := context.WithCancel(ctx) | 
					
						
							|  |  |  | 	conn := pg.Connect(opts).WithContext(pgCtx) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	// this will break the logfmt format we normally log in, | 
					
						
							|  |  |  | 	// since we can't choose where pg outputs to and it defaults to | 
					
						
							|  |  |  | 	// stdout. So use this option with care! | 
					
						
							|  |  |  | 	if log.Logger.GetLevel() >= logrus.TraceLevel { | 
					
						
							|  |  |  | 		conn.AddQueryHook(pgdebug.DebugHook{ | 
					
						
							|  |  |  | 			// Print all queries. | 
					
						
							|  |  |  | 			Verbose: true, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	// actually *begin* the connection so that we can tell if the db is there | 
					
						
							|  |  |  | 	// and listening, and also trigger the opts.OnConnect function passed in above | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	if err := conn.Ping(ctx); err != nil { | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 		cancel() | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("db connection error: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	// print out discovered postgres version | 
					
						
							|  |  |  | 	var version string | 
					
						
							|  |  |  | 	if _, err = conn.QueryOneContext(ctx, pg.Scan(&version), "SELECT version()"); err != nil { | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 		cancel() | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("db connection error: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	log.Infof("connected to postgres version: %s", version) | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// make sure the opts.OnConnect function has been triggered | 
					
						
							|  |  |  | 	// and closed the ready channel | 
					
						
							|  |  |  | 	select { | 
					
						
							|  |  |  | 	case <-readyChan: | 
					
						
							|  |  |  | 		log.Infof("postgres connection ready") | 
					
						
							|  |  |  | 	case <-time.After(5 * time.Second): | 
					
						
							|  |  |  | 		cancel() | 
					
						
							|  |  |  | 		return nil, errors.New("db connection timeout") | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	ps := &postgresService{ | 
					
						
							|  |  |  | 		config: c, | 
					
						
							|  |  |  | 		conn:   conn, | 
					
						
							|  |  |  | 		log:    log, | 
					
						
							|  |  |  | 		cancel: cancel, | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	federatingDB := newFederatingDB(ps, c) | 
					
						
							|  |  |  | 	ps.federationDB = federatingDB | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// we can confidently return this useable postgres service now | 
					
						
							|  |  |  | 	return ps, nil | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | /* | 
					
						
							|  |  |  | 	HANDY STUFF | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // derivePGOptions takes an application config and returns either a ready-to-use *pg.Options | 
					
						
							|  |  |  | // with sensible defaults, or an error if it's not satisfied by the provided config. | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | func derivePGOptions(c *config.Config) (*pg.Options, error) { | 
					
						
							|  |  |  | 	if strings.ToUpper(c.DBConfig.Type) != dbTypePostgres { | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("expected db type of %s but got %s", dbTypePostgres, c.DBConfig.Type) | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-04 12:07:24 +01:00
										 |  |  | 	// validate port | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	if c.DBConfig.Port == 0 { | 
					
						
							| 
									
										
										
										
											2021-03-04 12:07:24 +01:00
										 |  |  | 		return nil, errors.New("no port set") | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// validate address | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	if c.DBConfig.Address == "" { | 
					
						
							| 
									
										
										
										
											2021-03-04 12:07:24 +01:00
										 |  |  | 		return nil, errors.New("no address set") | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	ipv4Regex := regexp.MustCompile(`^(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$`) | 
					
						
							|  |  |  | 	hostnameRegex := regexp.MustCompile(`^(?:[a-z0-9]+(?:-[a-z0-9]+)*\.)+[a-z]{2,}$`) | 
					
						
							|  |  |  | 	if !hostnameRegex.MatchString(c.DBConfig.Address) && !ipv4Regex.MatchString(c.DBConfig.Address) && c.DBConfig.Address != "localhost" { | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("address %s was neither an ipv4 address nor a valid hostname", c.DBConfig.Address) | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// validate username | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	if c.DBConfig.User == "" { | 
					
						
							| 
									
										
										
										
											2021-03-04 12:07:24 +01:00
										 |  |  | 		return nil, errors.New("no user set") | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	// validate that there's a password | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	if c.DBConfig.Password == "" { | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 		return nil, errors.New("no password set") | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// validate database | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 	if c.DBConfig.Database == "" { | 
					
						
							| 
									
										
										
										
											2021-03-04 12:07:24 +01:00
										 |  |  | 		return nil, errors.New("no database set") | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | 	// We can rely on the pg library we're using to set | 
					
						
							|  |  |  | 	// sensible defaults for everything we don't set here. | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	options := &pg.Options{ | 
					
						
							| 
									
										
										
										
											2021-03-04 14:38:18 +01:00
										 |  |  | 		Addr:            fmt.Sprintf("%s:%d", c.DBConfig.Address, c.DBConfig.Port), | 
					
						
							|  |  |  | 		User:            c.DBConfig.User, | 
					
						
							|  |  |  | 		Password:        c.DBConfig.Password, | 
					
						
							|  |  |  | 		Database:        c.DBConfig.Database, | 
					
						
							|  |  |  | 		ApplicationName: c.ApplicationName, | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return options, nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	FEDERATION FUNCTIONALITY | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | func (ps *postgresService) Federation() pub.Database { | 
					
						
							|  |  |  | 	return ps.federationDB | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* | 
					
						
							|  |  |  | 	BASIC DB FUNCTIONALITY | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) CreateTable(i interface{}) error { | 
					
						
							|  |  |  | 	return ps.conn.Model(i).CreateTable(&orm.CreateTableOptions{ | 
					
						
							|  |  |  | 		IfNotExists: true, | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) DropTable(i interface{}) error { | 
					
						
							|  |  |  | 	return ps.conn.Model(i).DropTable(&orm.DropTableOptions{ | 
					
						
							|  |  |  | 		IfExists: true, | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-02 22:52:31 +01:00
										 |  |  | func (ps *postgresService) Stop(ctx context.Context) error { | 
					
						
							|  |  |  | 	ps.log.Info("closing db connection") | 
					
						
							|  |  |  | 	if err := ps.conn.Close(); err != nil { | 
					
						
							|  |  |  | 		// only cancel if there's a problem closing the db | 
					
						
							|  |  |  | 		ps.cancel() | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							| 
									
										
										
										
											2021-03-02 18:26:30 +01:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | func (ps *postgresService) IsHealthy(ctx context.Context) error { | 
					
						
							|  |  |  | 	return ps.conn.Ping(ctx) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | func (ps *postgresService) CreateSchema(ctx context.Context) error { | 
					
						
							|  |  |  | 	models := []interface{}{ | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		(*gtsmodel.Account)(nil), | 
					
						
							|  |  |  | 		(*gtsmodel.Status)(nil), | 
					
						
							|  |  |  | 		(*gtsmodel.User)(nil), | 
					
						
							| 
									
										
										
										
											2021-03-05 18:31:12 +01:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	ps.log.Info("creating db schema") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	for _, model := range models { | 
					
						
							|  |  |  | 		err := ps.conn.Model(model).CreateTable(&orm.CreateTableOptions{ | 
					
						
							|  |  |  | 			IfNotExists: true, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			return err | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	ps.log.Info("db schema created") | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | func (ps *postgresService) GetByID(id string, i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(i).Where("id = ?", id).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) GetWhere(key string, value interface{}, i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(i).Where("? = ?", pg.Safe(key), value).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | // func (ps *postgresService) GetWhereMany(i interface{}, where ...model.Where) error { | 
					
						
							|  |  |  | // 	return nil | 
					
						
							|  |  |  | // } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | func (ps *postgresService) GetAll(i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(i).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) Put(i interface{}) error { | 
					
						
							|  |  |  | 	_, err := ps.conn.Model(i).Insert(i) | 
					
						
							|  |  |  | 	return err | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) UpdateByID(id string, i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if _, err := ps.conn.Model(i).OnConflict("(id) DO UPDATE").Insert(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) UpdateOneByID(id string, key string, value interface{}, i interface{}) error { | 
					
						
							|  |  |  | 	_, err := ps.conn.Model(i).Set("? = ?", pg.Safe(key), value).Where("id = ?", id).Update() | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | 	return err | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) DeleteByID(id string, i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if _, err := ps.conn.Model(i).Where("id = ?", id).Delete(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) DeleteWhere(key string, value interface{}, i interface{}) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if _, err := ps.conn.Model(i).Where("? = ?", pg.Safe(key), value).Delete(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* | 
					
						
							|  |  |  | 	HANDY SHORTCUTS | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetAccountByUserID(userID string, account *gtsmodel.Account) error { | 
					
						
							|  |  |  | 	user := >smodel.User{ | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		ID: userID, | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	if err := ps.conn.Model(user).Where("id = ?", userID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	if err := ps.conn.Model(account).Where("id = ?", user.AccountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetFollowRequestsForAccountID(accountID string, followRequests *[]gtsmodel.FollowRequest) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(followRequests).Where("target_account_id = ?", accountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetFollowingByAccountID(accountID string, following *[]gtsmodel.Follow) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(following).Where("account_id = ?", accountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetFollowersByAccountID(accountID string, followers *[]gtsmodel.Follow) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(followers).Where("target_account_id = ?", accountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetStatusesByAccountID(accountID string, statuses *[]gtsmodel.Status) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(statuses).Where("account_id = ?", accountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetStatusesByTimeDescending(accountID string, statuses *[]gtsmodel.Status, limit int) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	q := ps.conn.Model(statuses).Order("created_at DESC") | 
					
						
							|  |  |  | 	if limit != 0 { | 
					
						
							|  |  |  | 		q = q.Limit(limit) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	if accountID != "" { | 
					
						
							|  |  |  | 		q = q.Where("account_id = ?", accountID) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	if err := q.Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetLastStatusForAccountID(accountID string, status *gtsmodel.Status) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(status).Order("created_at DESC").Limit(1).Where("account_id = ?", accountID).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) IsUsernameAvailable(username string) error { | 
					
						
							|  |  |  | 	// if no error we fail because it means we found something | 
					
						
							|  |  |  | 	// if error but it's not pg.ErrNoRows then we fail | 
					
						
							|  |  |  | 	// if err is pg.ErrNoRows we're good, we found nothing so continue | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	if err := ps.conn.Model(>smodel.Account{}).Where("username = ?", username).Where("domain = ?", nil).Select(); err == nil { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		return fmt.Errorf("username %s already in use", username) | 
					
						
							|  |  |  | 	} else if err != pg.ErrNoRows { | 
					
						
							|  |  |  | 		return fmt.Errorf("db error: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func (ps *postgresService) IsEmailAvailable(email string) error { | 
					
						
							|  |  |  | 	// parse the domain from the email | 
					
						
							|  |  |  | 	m, err := mail.ParseAddress(email) | 
					
						
							|  |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		return fmt.Errorf("error parsing email address %s: %s", email, err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	domain := strings.Split(m.Address, "@")[1] // domain will always be the second part after @ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// check if the email domain is blocked | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	if err := ps.conn.Model(>smodel.EmailDomainBlock{}).Where("domain = ?", domain).Select(); err == nil { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		// fail because we found something | 
					
						
							|  |  |  | 		return fmt.Errorf("email domain %s is blocked", domain) | 
					
						
							|  |  |  | 	} else if err != pg.ErrNoRows { | 
					
						
							|  |  |  | 		// fail because we got an unexpected error | 
					
						
							|  |  |  | 		return fmt.Errorf("db error: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// check if this email is associated with a user already | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	if err := ps.conn.Model(>smodel.User{}).Where("email = ?", email).WhereOr("unconfirmed_email = ?", email).Select(); err == nil { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		// fail because we found something | 
					
						
							|  |  |  | 		return fmt.Errorf("email %s already in use", email) | 
					
						
							|  |  |  | 	} else if err != pg.ErrNoRows { | 
					
						
							|  |  |  | 		// fail because we got an unexpected error | 
					
						
							|  |  |  | 		return fmt.Errorf("db error: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) NewSignup(username string, reason string, requireApproval bool, email string, password string, signUpIP net.IP, locale string, appID string) (*gtsmodel.User, error) { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	key, err := rsa.GenerateKey(rand.Reader, 2048) | 
					
						
							|  |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		ps.log.Errorf("error creating new rsa key: %s", err) | 
					
						
							|  |  |  | 		return nil, err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	uris := util.GenerateURIs(username, ps.config.Protocol, ps.config.Host) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	a := >smodel.Account{ | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		Username:              username, | 
					
						
							|  |  |  | 		DisplayName:           username, | 
					
						
							|  |  |  | 		Reason:                reason, | 
					
						
							|  |  |  | 		URL:                   uris.UserURL, | 
					
						
							|  |  |  | 		PrivateKey:            key, | 
					
						
							|  |  |  | 		PublicKey:             &key.PublicKey, | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		ActorType:             gtsmodel.ActivityStreamsPerson, | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		URI:                   uris.UserURI, | 
					
						
							| 
									
										
										
										
											2021-04-02 21:15:17 +02:00
										 |  |  | 		InboxURL:              uris.InboxURI, | 
					
						
							|  |  |  | 		OutboxURL:             uris.OutboxURI, | 
					
						
							|  |  |  | 		FollowersURL:          uris.FollowersURI, | 
					
						
							|  |  |  | 		FeaturedCollectionURL: uris.CollectionURI, | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	if _, err = ps.conn.Model(a).Insert(); err != nil { | 
					
						
							|  |  |  | 		return nil, err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	pw, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost) | 
					
						
							|  |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		return nil, fmt.Errorf("error hashing password: %s", err) | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	u := >smodel.User{ | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		AccountID:              a.ID, | 
					
						
							|  |  |  | 		EncryptedPassword:      string(pw), | 
					
						
							|  |  |  | 		SignUpIP:               signUpIP, | 
					
						
							|  |  |  | 		Locale:                 locale, | 
					
						
							|  |  |  | 		UnconfirmedEmail:       email, | 
					
						
							|  |  |  | 		CreatedByApplicationID: appID, | 
					
						
							|  |  |  | 		Approved:               !requireApproval, // if we don't require moderator approval, just pre-approve the user | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	if _, err = ps.conn.Model(u).Insert(); err != nil { | 
					
						
							|  |  |  | 		return nil, err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return u, nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) SetHeaderOrAvatarForAccountID(mediaAttachment *gtsmodel.MediaAttachment, accountID string) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	_, err := ps.conn.Model(mediaAttachment).Insert() | 
					
						
							| 
									
										
										
										
											2021-03-22 22:26:54 +01:00
										 |  |  | 	return err | 
					
						
							| 
									
										
										
										
											2021-03-14 17:56:16 +01:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetHeaderForAccountID(header *gtsmodel.MediaAttachment, accountID string) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(header).Where("account_id = ?", accountID).Where("header = ?", true).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) GetAvatarForAccountID(avatar *gtsmodel.MediaAttachment, accountID string) error { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.conn.Model(avatar).Where("account_id = ?", accountID).Where("avatar = ?", true).Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			return ErrNoEntries{} | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		return err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-04 23:17:10 +02:00
										 |  |  | func (ps *postgresService) Blocked(account1 string, account2 string) (bool, error) { | 
					
						
							|  |  |  | 	var blocked bool | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	if err := ps.conn.Model(>smodel.Block{}). | 
					
						
							| 
									
										
										
										
											2021-04-04 23:17:10 +02:00
										 |  |  | 		Where("account_id = ?", account1).Where("target_account_id = ?", account2). | 
					
						
							|  |  |  | 		WhereOr("target_account_id = ?", account1).Where("account_id = ?", account2). | 
					
						
							|  |  |  | 		Select(); err != nil { | 
					
						
							|  |  |  | 		if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 			blocked = false | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 			return blocked, nil | 
					
						
							| 
									
										
										
										
											2021-04-04 23:17:10 +02:00
										 |  |  | 		} else { | 
					
						
							|  |  |  | 			return blocked, err | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	blocked = true | 
					
						
							|  |  |  | 	return blocked, nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | /* | 
					
						
							|  |  |  | 	CONVERSION FUNCTIONS | 
					
						
							|  |  |  | */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // AccountToMastoSensitive takes an internal account model and transforms it into an account ready to be served through the API. | 
					
						
							|  |  |  | // The resulting account fits the specifications for the path /api/v1/accounts/verify_credentials, as described here: | 
					
						
							|  |  |  | // https://docs.joinmastodon.org/methods/accounts/. Note that it's *sensitive* because it's only meant to be exposed to the user | 
					
						
							|  |  |  | // that the account actually belongs to. | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) AccountToMastoSensitive(a *gtsmodel.Account) (*mastotypes.Account, error) { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	// we can build this sensitive account easily by first getting the public account.... | 
					
						
							|  |  |  | 	mastoAccount, err := ps.AccountToMastoPublic(a) | 
					
						
							|  |  |  | 	if err != nil { | 
					
						
							|  |  |  | 		return nil, err | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// then adding the Source object to it... | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// check pending follow requests aimed at this account | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	fr := []gtsmodel.FollowRequest{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetFollowRequestsForAccountID(a.ID, &fr); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting follow requests: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	var frc int | 
					
						
							|  |  |  | 	if fr != nil { | 
					
						
							|  |  |  | 		frc = len(fr) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	mastoAccount.Source = &mastotypes.Source{ | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		Privacy:             util.ParseMastoVisFromGTSVis(a.Privacy), | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 		Sensitive:           a.Sensitive, | 
					
						
							|  |  |  | 		Language:            a.Language, | 
					
						
							|  |  |  | 		Note:                a.Note, | 
					
						
							|  |  |  | 		Fields:              mastoAccount.Fields, | 
					
						
							|  |  |  | 		FollowRequestsCount: frc, | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return mastoAccount, nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) AccountToMastoPublic(a *gtsmodel.Account) (*mastotypes.Account, error) { | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	// count followers | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	followers := []gtsmodel.Follow{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetFollowersByAccountID(a.ID, &followers); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting followers: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	var followersCount int | 
					
						
							|  |  |  | 	if followers != nil { | 
					
						
							|  |  |  | 		followersCount = len(followers) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// count following | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	following := []gtsmodel.Follow{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetFollowingByAccountID(a.ID, &following); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting following: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	var followingCount int | 
					
						
							|  |  |  | 	if following != nil { | 
					
						
							|  |  |  | 		followingCount = len(following) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// count statuses | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	statuses := []gtsmodel.Status{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetStatusesByAccountID(a.ID, &statuses); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting last statuses: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	var statusesCount int | 
					
						
							|  |  |  | 	if statuses != nil { | 
					
						
							|  |  |  | 		statusesCount = len(statuses) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// check when the last status was | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	lastStatus := >smodel.Status{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetLastStatusForAccountID(a.ID, lastStatus); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting last status: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	var lastStatusAt string | 
					
						
							|  |  |  | 	if lastStatus != nil { | 
					
						
							|  |  |  | 		lastStatusAt = lastStatus.CreatedAt.Format(time.RFC3339) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// build the avatar and header URLs | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	avi := >smodel.MediaAttachment{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetAvatarForAccountID(avi, a.ID); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting avatar: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	aviURL := avi.File.Path | 
					
						
							|  |  |  | 	aviURLStatic := avi.Thumbnail.Path | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 	header := >smodel.MediaAttachment{} | 
					
						
							| 
									
										
										
										
											2021-04-01 20:46:45 +02:00
										 |  |  | 	if err := ps.GetHeaderForAccountID(avi, a.ID); err != nil { | 
					
						
							|  |  |  | 		if _, ok := err.(ErrNoEntries); !ok { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("error getting header: %s", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	headerURL := header.File.Path | 
					
						
							|  |  |  | 	headerURLStatic := header.Thumbnail.Path | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	// get the fields set on this account | 
					
						
							|  |  |  | 	fields := []mastotypes.Field{} | 
					
						
							|  |  |  | 	for _, f := range a.Fields { | 
					
						
							|  |  |  | 		mField := mastotypes.Field{ | 
					
						
							|  |  |  | 			Name:  f.Name, | 
					
						
							|  |  |  | 			Value: f.Value, | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		if !f.VerifiedAt.IsZero() { | 
					
						
							|  |  |  | 			mField.VerifiedAt = f.VerifiedAt.Format(time.RFC3339) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		fields = append(fields, mField) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	var acct string | 
					
						
							|  |  |  | 	if a.Domain != "" { | 
					
						
							|  |  |  | 		// this is a remote user | 
					
						
							|  |  |  | 		acct = fmt.Sprintf("%s@%s", a.Username, a.Domain) | 
					
						
							|  |  |  | 	} else { | 
					
						
							|  |  |  | 		// this is a local user | 
					
						
							|  |  |  | 		acct = a.Username | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return &mastotypes.Account{ | 
					
						
							|  |  |  | 		ID:             a.ID, | 
					
						
							|  |  |  | 		Username:       a.Username, | 
					
						
							|  |  |  | 		Acct:           acct, | 
					
						
							|  |  |  | 		DisplayName:    a.DisplayName, | 
					
						
							|  |  |  | 		Locked:         a.Locked, | 
					
						
							|  |  |  | 		Bot:            a.Bot, | 
					
						
							|  |  |  | 		CreatedAt:      a.CreatedAt.Format(time.RFC3339), | 
					
						
							|  |  |  | 		Note:           a.Note, | 
					
						
							|  |  |  | 		URL:            a.URL, | 
					
						
							|  |  |  | 		Avatar:         aviURL, | 
					
						
							|  |  |  | 		AvatarStatic:   aviURLStatic, | 
					
						
							|  |  |  | 		Header:         headerURL, | 
					
						
							|  |  |  | 		HeaderStatic:   headerURLStatic, | 
					
						
							|  |  |  | 		FollowersCount: followersCount, | 
					
						
							|  |  |  | 		FollowingCount: followingCount, | 
					
						
							|  |  |  | 		StatusesCount:  statusesCount, | 
					
						
							|  |  |  | 		LastStatusAt:   lastStatusAt, | 
					
						
							|  |  |  | 		Emojis:         nil, // TODO: implement this | 
					
						
							|  |  |  | 		Fields:         fields, | 
					
						
							|  |  |  | 	}, nil | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) MentionStringsToMentions(targetAccounts []string, originAccountID string, statusID string) ([]*gtsmodel.Mention, error) { | 
					
						
							|  |  |  | 	menchies := []*gtsmodel.Mention{} | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 	for _, a := range targetAccounts { | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		// A mentioned account looks like "@test@example.org" or just "@test" for a local account | 
					
						
							|  |  |  | 		// -- we can guarantee this from the regex that targetAccounts should have been derived from. | 
					
						
							|  |  |  | 		// But we still need to do a bit of fiddling to get what we need here -- the username and domain (if given). | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | 		// 1.  trim off the first @ | 
					
						
							|  |  |  | 		t := strings.TrimPrefix(a, "@") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// 2. split the username and domain | 
					
						
							|  |  |  | 		s := strings.Split(t, "@") | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		// 3. if it's length 1 it's a local account, length 2 means remote, anything else means something is wrong | 
					
						
							|  |  |  | 		var local bool | 
					
						
							|  |  |  | 		switch len(s) { | 
					
						
							|  |  |  | 		case 1: | 
					
						
							|  |  |  | 			local = true | 
					
						
							|  |  |  | 		case 2: | 
					
						
							|  |  |  | 			local = false | 
					
						
							|  |  |  | 		default: | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("mentioned account format '%s' was not valid", a) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		var username, domain string | 
					
						
							|  |  |  | 		username = s[0] | 
					
						
							|  |  |  | 		if !local { | 
					
						
							|  |  |  | 			domain = s[1] | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// 4. check we now have a proper username and domain | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		if username == "" || (!local && domain == "") { | 
					
						
							|  |  |  | 			return nil, fmt.Errorf("username or domain for '%s' was nil", a) | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		// okay we're good now, we can start pulling accounts out of the database | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		mentionedAccount := >smodel.Account{} | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		var err error | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		if local { | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 			// local user -- should have a null domain | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 			err = ps.conn.Model(mentionedAccount).Where("username = ?", username).Where("? IS NULL", pg.Ident("domain")).Select() | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		} else { | 
					
						
							|  |  |  | 			// remote user -- should have domain defined | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 			err = ps.conn.Model(mentionedAccount).Where("username = ?", username).Where("? = ?", pg.Ident("domain"), domain).Select() | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			if err == pg.ErrNoRows { | 
					
						
							|  |  |  | 				// no result found for this username/domain so just don't include it as a mencho and carry on about our business | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 				ps.log.Debugf("no account found with username '%s' and domain '%s', skipping it", username, domain) | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 				continue | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			// a serious error has happened so bail | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 			return nil, fmt.Errorf("error getting account with username '%s' and domain '%s': %s", username, domain, err) | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 		// id, createdAt and updatedAt will be populated by the db, so we have everything we need! | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		menchies = append(menchies, >smodel.Mention{ | 
					
						
							| 
									
										
										
										
											2021-04-04 23:17:10 +02:00
										 |  |  | 			StatusID:        statusID, | 
					
						
							| 
									
										
										
										
											2021-04-03 19:40:15 +02:00
										 |  |  | 			OriginAccountID: originAccountID, | 
					
						
							|  |  |  | 			TargetAccountID: mentionedAccount.ID, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return menchies, nil | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 
 | 
					
						
							|  |  |  | // for now this function doesn't really use the database, but it's here because: | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | // A) it probably will later and | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | // B) it's v. similar to MentionStringsToMentions | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) TagStringsToTags(tags []string, originAccountID string, statusID string) ([]*gtsmodel.Tag, error) { | 
					
						
							|  |  |  | 	newTags := []*gtsmodel.Tag{} | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 	for _, t := range tags { | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		newTags = append(newTags, >smodel.Tag{ | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 			Name: t, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return newTags, nil | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | func (ps *postgresService) EmojiStringsToEmojis(emojis []string, originAccountID string, statusID string) ([]*gtsmodel.Emoji, error) { | 
					
						
							|  |  |  | 	newEmojis := []*gtsmodel.Emoji{} | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 	for _, e := range emojis { | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 		emoji := >smodel.Emoji{} | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 		err := ps.conn.Model(emoji).Where("shortcode = ?", e).Where("visible_in_picker = true").Where("disabled = false").Select() | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			if err == pg.ErrNoRows { | 
					
						
							| 
									
										
										
										
											2021-04-08 23:29:35 +02:00
										 |  |  | 				// no result found for this username/domain so just don't include it as an emoji and carry on about our business | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 				ps.log.Debugf("no emoji found with shortcode %s, skipping it", e) | 
					
						
							|  |  |  | 				continue | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			// a serious error has happened so bail | 
					
						
							| 
									
										
										
										
											2021-04-04 23:17:10 +02:00
										 |  |  | 			return nil, fmt.Errorf("error getting emoji with shortcode %s: %s", e, err) | 
					
						
							| 
									
										
										
										
											2021-04-04 19:20:31 +02:00
										 |  |  | 		} | 
					
						
							|  |  |  | 		newEmojis = append(newEmojis, emoji) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return newEmojis, nil | 
					
						
							|  |  |  | } |