working mongo doc add

2 files changed, 20 insertions, 2 deletions

diff --git a/db/db.go b/db/db.go
index 5c35495..a6f439e 100644
--- a/db/db.go
+++ b/db/db.go
@@ -3,6 +3,7 @@ package db
 import (
 	"os"
 
+	"github.com/jackyzha0/ctrl-v/hashing"
 	"github.com/joho/godotenv"
 	log "github.com/sirupsen/logrus"
 )
@@ -21,8 +22,10 @@ func init() {
 	initSessions(mUser, mPass, mIP)
 }
 
-// creates a new
-func New(hash, content string) error {
+// creates a new paste with content and hash
+func New(ip, content string) error {
+	// generate hash from ip
+	hash := hashing.GenerateURI(ip)
 
 	// create new struct
 	new := Paste{
@@ -35,3 +38,8 @@ func New(hash, content string) error {
 	insertErr := insert(new)
 	return insertErr
 }
+
+// lookup
+func Lookup(hash string) (Paste, error) {
+	return fetch(hash)
+}
diff --git a/db/mongo.go b/db/mongo.go
index c3e31dd..1c694e7 100644
--- a/db/mongo.go
+++ b/db/mongo.go
@@ -6,6 +6,7 @@ import (
 	"net"
 
 	"github.com/globalsign/mgo"
+	"github.com/globalsign/mgo/bson"
 	log "github.com/sirupsen/logrus"
 )
 
@@ -39,5 +40,14 @@ func initSessions(user, pass, ip string) {
 }
 
 func insert(new Paste) error {
+	log.Infof("new paste struct: %+v", new)
 	return pastes.Insert(new)
 }
+
+func fetch(hash string) (Paste, error) {
+	p := Paste{}
+
+	q := bson.M{"hash": hash}
+	err := pastes.Find(q).One(&p)
+	return p, err
+}