mirror of https://github.com/gabehf/massflip.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.1 KiB
107 lines
3.1 KiB
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/joho/godotenv"
|
|
"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"
|
|
"go.mongodb.org/mongo-driver/mongo/readpref"
|
|
)
|
|
|
|
var DB = openDB()
|
|
|
|
func openDB() *mongo.Client {
|
|
err := godotenv.Load()
|
|
if err != nil {
|
|
log.Fatal("Unable to load database. Shutting down...")
|
|
}
|
|
dbUsername := os.Getenv("DB_USERNAME")
|
|
dbPassword := os.Getenv("DB_PASSWORD")
|
|
|
|
serverAPIOptions := options.ServerAPI(options.ServerAPIVersion1)
|
|
clientOptions := options.Client().
|
|
ApplyURI("mongodb+srv://" + dbUsername + ":" + dbPassword + "@cluster0.tqrat.mongodb.net/myFirstDatabase?retryWrites=true&w=majority").
|
|
SetServerAPIOptions(serverAPIOptions)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
client, err := mongo.Connect(ctx, clientOptions)
|
|
err = client.Ping(ctx, readpref.Primary())
|
|
if err != nil {
|
|
log.Fatal("DB Error: " + err.Error())
|
|
}
|
|
return client
|
|
}
|
|
|
|
func DBSubtractPoints(user string, p int) error {
|
|
err := DB.Ping(context.Background(), readpref.Primary())
|
|
if err != nil {
|
|
DB = openDB()
|
|
}
|
|
userCollection := DB.Database("Users").Collection("Users")
|
|
var v ExistingAccount
|
|
err = userCollection.FindOne(context.Background(), bson.D{primitive.E{Key: "username", Value: user}}).Decode(&v)
|
|
|
|
if err != nil {
|
|
fmt.Println("* Error editing points for user " + user + ": User not found in DB")
|
|
return err
|
|
}
|
|
|
|
filter := bson.D{primitive.E{Key: "username", Value: user}}
|
|
opts := options.Update().SetUpsert(true)
|
|
update := bson.D{primitive.E{Key: "$set", Value: bson.D{primitive.E{Key: "points", Value: v.Points - p}}}}
|
|
_, err = userCollection.UpdateOne(context.TODO(), filter, update, opts)
|
|
if err != nil {
|
|
log.Println("* Error updating points in ")
|
|
}
|
|
return nil
|
|
}
|
|
func DBAddPoints(user string, p int) error {
|
|
err := DB.Ping(context.Background(), readpref.Primary())
|
|
if err != nil {
|
|
DB = openDB()
|
|
}
|
|
userCollection := DB.Database("Users").Collection("Users")
|
|
var v ExistingAccount
|
|
err = userCollection.FindOne(context.Background(), bson.D{primitive.E{Key: "username", Value: user}}).Decode(&v)
|
|
|
|
if err != nil {
|
|
fmt.Println("* Error editing points for user " + user + ": User not found in DB")
|
|
return err
|
|
}
|
|
|
|
filter := bson.D{primitive.E{Key: "username", Value: user}}
|
|
opts := options.Update().SetUpsert(true)
|
|
update := bson.D{primitive.E{Key: "$set", Value: bson.D{primitive.E{Key: "points", Value: v.Points + p}}}}
|
|
_, err = userCollection.UpdateOne(context.TODO(), filter, update, opts)
|
|
if err != nil {
|
|
log.Println("* Error updating points in ")
|
|
}
|
|
return nil
|
|
|
|
}
|
|
|
|
func DBGetUserByUsername(user string) ExistingAccount {
|
|
err := DB.Ping(context.Background(), readpref.Primary())
|
|
if err != nil {
|
|
DB = openDB()
|
|
}
|
|
userCollection := DB.Database("Users").Collection("Users")
|
|
var v ExistingAccount
|
|
err = userCollection.FindOne(context.Background(), bson.D{primitive.E{Key: "username", Value: user}}).Decode(&v)
|
|
|
|
if err != nil {
|
|
fmt.Println("* Error getting user " + user + ": Not found in DB")
|
|
var e ExistingAccount
|
|
return e
|
|
}
|
|
|
|
return v
|
|
}
|