working mongo doc add

1 files changed, 10 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)
+}