otter-space-sdk/pkg/database/postgres.go

package database

import (
	"context"
	"database/sql"
	"embed"
	"fmt"

	"git.dragse.it/anthrove/otter-space-sdk/internal/postgres"
	"git.dragse.it/anthrove/otter-space-sdk/pkg/models"
	"git.dragse.it/anthrove/otter-space-sdk/pkg/models/graphModels"
	_ "github.com/lib/pq"
	migrate "github.com/rubenv/sql-migrate"
	log "github.com/sirupsen/logrus"
	gormPostgres "gorm.io/driver/postgres"
	"gorm.io/gorm"
)

//go:embed migrations/*.sql
var embedMigrations embed.FS

type postgresqlConnection struct {
	db *gorm.DB
}

func NewPostgresqlConnection() OtterSpace {
	return &postgresqlConnection{
		db: nil,
	}
}

func (p postgresqlConnection) Connect(ctx context.Context, endpoint string, username string, password string, database string, port int, ssl string, timezone string) error {
	dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=%s TimeZone=%s", endpoint, username, password, database, port, ssl, timezone)
	var err error

	err = p.migrateDatabase(dsn)
	if err != nil {
		return err
	}

	p.db, err = gorm.Open(gormPostgres.Open(dsn), &gorm.Config{})
	if err != nil {
		return err
	}

	return nil
}

func (p postgresqlConnection) AddUserWithRelationToSource(ctx context.Context, anthroveUserID models.AnthroveUserID, sourceDomain string, userID string, username string) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) AddSource(ctx context.Context, anthroveSource *graphModels.AnthroveSource) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) AddPost(ctx context.Context, anthrovePost *graphModels.AnthrovePost) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) AddTagWithRelationToPost(ctx context.Context, anthrovePostID models.AnthrovePostID, anthroveTag *graphModels.AnthroveTag) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) LinkPostWithSource(ctx context.Context, anthrovePostID models.AnthrovePostID, anthroveSourceDomain string, anthrovePostRelationship *graphModels.AnthrovePostRelationship) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) LinkUserWithPost(ctx context.Context, anthroveUser *graphModels.AnthroveUser, anthrovePost *graphModels.AnthrovePost) error {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) CheckUserPostLink(ctx context.Context, anthroveUserID models.AnthroveUserID, sourcePostID string, sourceUrl string) (bool, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) CheckPostNodeExistsByAnthroveID(ctx context.Context, anthrovePost *graphModels.AnthrovePost) (*graphModels.AnthrovePost, bool, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) CheckPostNodeExistsBySourceURL(ctx context.Context, sourceUrl string) (*graphModels.AnthrovePost, bool, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) CheckPostNodeExistsBySourceID(ctx context.Context, sourcePostID string) (*graphModels.AnthrovePost, bool, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetUserFavoriteCount(ctx context.Context, anthroveUserID models.AnthroveUserID) (int64, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetUserSourceLinks(ctx context.Context, anthroveUserID models.AnthroveUserID) (map[string]graphModels.AnthroveUserRelationship, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetSpecifiedUserSourceLink(ctx context.Context, anthroveUserID models.AnthroveUserID, sourceDisplayName string) (map[string]graphModels.AnthroveUserRelationship, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetAnthroveUser(ctx context.Context, anthroveUserID models.AnthroveUserID) (*graphModels.AnthroveUser, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetAllAnthroveUserIDs(ctx context.Context) ([]models.AnthroveUserID, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetUserFavoritePostsWithPagination(ctx context.Context, anthroveUserID models.AnthroveUserID, skip int, limit int) (*graphModels.FavoriteList, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetUserTagsTroughFavedPosts(ctx context.Context, anthroveUserID models.AnthroveUserID) ([]graphModels.TagsWithFrequency, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetAllTags(ctx context.Context) ([]graphModels.TagsWithFrequency, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) GetAllSources(ctx context.Context) ([]graphModels.AnthroveSource, error) {
	return postgres.GetAllSourceNodes(ctx, p.db)
}

func (p postgresqlConnection) GetSourceByURL(ctx context.Context, sourceUrl string) (*graphModels.AnthroveSource, error) {
	//TODO implement me
	panic("implement me")
}

func (p postgresqlConnection) migrateDatabase(connectionString string) error {
	dialect := "postgres"
	migrations := &migrate.EmbedFileSystemMigrationSource{FileSystem: embedMigrations, Root: "migrations"}

	db, err := sql.Open(dialect, connectionString)
	if err != nil {
		return fmt.Errorf("postgres migration: %v", err)
	}

	n, err := migrate.Exec(db, dialect, migrations, migrate.Up)
	if err != nil {
		return fmt.Errorf("postgres migration: %v", err)
	}
	if n != 0 {
		log.Infof("postgres migration: applied %d migrations!", n)

	} else {
		log.Info("postgres migration: nothing to migrate")

	}

	return nil
}