1 files changed, 90 insertions, 0 deletions

diff --git a/db/mongodb/client.go b/db/mongodb/client.go
new file mode 100644
index 0000000..98369b9
--- /dev/null
+++ b/db/mongodb/client.go
@@ -0,0 +1,90 @@
+package mongodb
+
+import (
+        "context"
+        "net/url"
+
+        "go.mongodb.org/mongo-driver/bson"
+        "go.mongodb.org/mongo-driver/mongo"
+        "go.mongodb.org/mongo-driver/mongo/options"
+        "go.mongodb.org/mongo-driver/x/mongo/driver/connstring"
+
+        "code.crute.us/mcrute/golib/vault"
+)
+
+type Mongo struct {
+        client *mongo.Client
+        db     *mongo.Database
+}
+
+func Connect(ctx context.Context, uri, materialSet string) (*Mongo, error) {
+        db := &Mongo{}
+
+        cred, err := vault.GetVaultDatabase(materialSet)
+        if err != nil {
+                return nil, err
+        }
+
+        u, err := url.Parse(uri)
+        if err != nil {
+                return nil, err
+        }
+        u.User = url.UserPassword(cred.Username, cred.Password)
+
+        cs, err := connstring.ParseAndValidate(u.String())
+        if err != nil {
+                return nil, err
+        }
+
+        client, err := mongo.Connect(ctx, options.Client().ApplyURI(u.String()))
+        if err != nil {
+                return nil, err
+        }
+
+        db.client = client
+        db.db = client.Database(cs.Database)
+
+        return db, nil
+}
+
+func (m *Mongo) Collection(name string) *mongo.Collection {
+        return m.db.Collection(name)
+}
+
+func (m *Mongo) FindAll(ctx context.Context, cn string, out interface{}) error {
+        res, err := m.db.Collection(cn).Find(ctx, bson.D{})
+        if err != nil {
+                return err
+        }
+
+        if err = res.All(ctx, out); err != nil {
+                return err
+        }
+
+        return nil
+}
+
+func (m *Mongo) FindOneById(ctx context.Context, cn string, id string, out interface{}) error {
+        if err := m.db.Collection(cn).FindOne(ctx, bson.M{"_id": id}).Decode(out); err != nil {
+                return err
+        }
+
+        return nil
+}
+
+func (m *Mongo) InsertOne(ctx context.Context, cn string, in interface{}) error {
+        _, err := m.db.Collection(cn).InsertOne(ctx, in)
+        if err != nil {
+                return err
+        }
+
+        return nil
+}
+
+func (m *Mongo) ReplaceOneById(ctx context.Context, cn string, id string, in interface{}) error {
+        if _, err := m.db.Collection(cn).ReplaceOne(ctx, bson.M{"_id": id}, in); err != nil {
+                return err
+        }
+
+        return nil
+}