Initial commit

2023-08-26 10:36:05 +02:00
commit a3543ee13a
76 changed files with 6165 additions and 0 deletions
--- a/backend/database/mongo.go
+++ b/backend/database/mongo.go
@@ -0,0 +1,166 @@
+package database
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"os"
+	"time"
+
+	"github.com/rs/zerolog/log"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+
+	"codeberg.org/pluja/whishper/models"
+)
+
+type MongoDb struct {
+	client *mongo.Client
+}
+
+func NewMongoDb() *MongoDb {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+	client, err := mongo.Connect(ctx, options.Client().ApplyURI(fmt.Sprintf("mongodb://root:example@%v", os.Getenv("DB_ENDPOINT"))))
+	if err != nil {
+		log.Error().Err(err).Msg("Error connecting to mongodb")
+	}
+	return &MongoDb{
+		client: client,
+	}
+}
+
+func (m *MongoDb) GetTranscription(id string) *models.Transcription {
+	collection := m.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	oid, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		log.Printf("Error converting id to object id: %v", err)
+		return nil
+	}
+	filter := bson.D{primitive.E{Key: "_id", Value: oid}}
+	var result models.Transcription
+	err = collection.FindOne(ctx, filter).Decode(&result)
+	if err != nil {
+		log.Printf("Error getting transcription: %v", err)
+		return nil
+	}
+	return &result
+}
+
+func (m *MongoDb) DeleteTranscription(id string) error {
+	collection := m.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	oid, err := primitive.ObjectIDFromHex(id)
+	if err != nil {
+		log.Debug().Msg("Error converting id to object id.")
+		return err
+	}
+
+	filter := bson.D{primitive.E{Key: "_id", Value: oid}}
+	_, err = collection.DeleteOne(ctx, filter)
+	if err != nil {
+		log.Debug().Msg("Error deleting transcription")
+		return err
+	}
+	return nil
+}
+
+func (m *MongoDb) NewTranscription(t *models.Transcription) (*models.Transcription, error) {
+	collection := m.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+	// Create a new mongodb object id
+	i, err := collection.InsertOne(ctx, t)
+	if err != nil {
+		log.Printf("Error creating new transcription: %v", err)
+		return nil, err
+	}
+	// Set the id of the transcription to the mongodb object id
+	t.ID = i.InsertedID.(primitive.ObjectID)
+	return t, nil
+}
+
+func (s *MongoDb) GetAllTranscriptions() []*models.Transcription {
+	collection := s.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	cursor, err := collection.Find(ctx, bson.D{})
+	if err != nil {
+		log.Printf("Error getting transcriptions: %v", err)
+		return nil
+	}
+	defer cursor.Close(ctx)
+
+	var transcriptions []*models.Transcription
+	for cursor.Next(ctx) {
+		var result models.Transcription
+		err := cursor.Decode(&result)
+		if err != nil {
+			log.Printf("Error decoding transcription: %v", err)
+			return nil
+		}
+		transcriptions = append(transcriptions, &result)
+	}
+	if err := cursor.Err(); err != nil {
+		log.Printf("Error getting transcriptions: %v", err)
+		return nil
+	}
+	return transcriptions
+}
+
+func (s *MongoDb) GetPendingTranscriptions() []*models.Transcription {
+	collection := s.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	filter := bson.D{primitive.E{Key: "status", Value: models.TranscriptionStatusPending}}
+	cursor, err := collection.Find(ctx, filter)
+	if err != nil {
+		log.Printf("Error getting transcriptions: %v", err)
+		return nil
+	}
+
+	defer cursor.Close(ctx)
+	var transcriptions []*models.Transcription
+	for cursor.Next(ctx) {
+		var result models.Transcription
+		err := cursor.Decode(&result)
+		if err != nil {
+			log.Printf("Error decoding transcription: %v", err)
+			return nil
+		}
+		transcriptions = append(transcriptions, &result)
+	}
+
+	return transcriptions
+}
+
+func (m *MongoDb) UpdateTranscription(t *models.Transcription) (*models.Transcription, error) {
+	collection := m.client.Database("whishper").Collection("transcriptions")
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+	defer cancel()
+
+	filter := bson.D{primitive.E{Key: "_id", Value: t.ID}}
+	updateQuery := bson.D{primitive.E{Key: "$set", Value: t}}
+	updateResult, err := collection.UpdateOne(ctx, filter, updateQuery)
+	if err != nil {
+		return nil, err
+	}
+
+	if updateResult.MatchedCount == 0 {
+		return nil, errors.New("no documents matched the filter")
+	}
+	if updateResult.ModifiedCount == 0 {
+		return nil, errors.New("no documents were modified")
+	}
+
+	return t, nil
+}