From 712bf925f8ae84a5a2fee87b8aa3c969df48fd50 Mon Sep 17 00:00:00 2001
From: jackyzha0 <j.zhao2k19@gmail.com>
Date: Sat, 16 May 2020 22:42:48 -0700
Subject: allow language saving

---
 backend/db/db.go      | 9 +++++----
 backend/db/schemas.go | 5 +++--
 2 files changed, 8 insertions(+), 6 deletions(-)

(limited to 'backend/db')

diff --git a/backend/db/db.go b/backend/db/db.go
index df112d0..258b4df 100644
--- a/backend/db/db.go
+++ b/backend/db/db.go
@@ -28,7 +28,7 @@ const TitleLimit = 100
 const ContentLimit = 100000
 
 // creates a new paste with title, content and hash, returns the hash of the created paste
-func New(ip, content, expiry, title, password string) (string, error) {
+func New(ip, content, expiry, title, password, lang string) (string, error) {
 	// generate hash from ip
 	hash := security.GenerateURI(ip)
 
@@ -40,9 +40,10 @@ func New(ip, content, expiry, title, password string) (string, error) {
 
 	// create new struct
 	new := Paste{
-		Hash:    hash,
-		Content: content,
-		Title:   title,
+		Hash:     hash,
+		Content:  content,
+		Title:    title,
+		Language: lang,
 	}
 
 	// if there is a password, encrypt content and hash the password
diff --git a/backend/db/schemas.go b/backend/db/schemas.go
index d3551fc..a3b55d0 100644
--- a/backend/db/schemas.go
+++ b/backend/db/schemas.go
@@ -10,9 +10,10 @@ import (
 type Paste struct {
 	ID       bson.ObjectId `bson:"_id,omitempty"`
 	Hash     string
-	Content  string
-	Expiry   time.Time `bson:"expiry"`
 	Title    string
+	Content  string
+	Language string
 	Password string
+	Expiry   time.Time `bson:"expiry"`
 	Salt     []byte
 }
-- 
cgit v1.2.3