diff options
Diffstat (limited to 'backend/db')
| -rw-r--r-- | backend/db/db.go | 25 | ||||
| -rw-r--r-- | backend/db/schemas.go | 1 |
2 files changed, 22 insertions, 4 deletions
diff --git a/backend/db/db.go b/backend/db/db.go index 053ba87..4451f34 100644 --- a/backend/db/db.go +++ b/backend/db/db.go @@ -24,15 +24,32 @@ func init() { initSessions(mUser, mPass, mIP) } -// creates a new paste with content and hash -func New(ip, content, expiry string) error { +const TitleLimit = 100 +const ContentLimit = 100000 + +// creates a new paste with title, content and hash +func New(ip, content, expiry, title string) error { // generate hash from ip hash := hashing.GenerateURI(ip) + // check for size of title and content + errs := "" + if len(title) > TitleLimit { + errs += fmt.Sprintf("title is longer than character limit of %d\n", TitleLimit) + } + if len(content) > ContentLimit { + errs += fmt.Sprintf("content is longer than character limit of %d\n", ContentLimit) + } + // if any errors were found + if errs != "" { + return fmt.Errorf(errs) + } + // create new struct new := Paste{ Hash: hash, Content: content, + Title: title, } // check if expiry @@ -45,8 +62,8 @@ func New(ip, content, expiry string) error { } // time is in the past - if t.After(time.Now()) { - return fmt.Errorf("err: time %s is in the past", t.String()) + if time.Now().After(t) { + return fmt.Errorf("time %s is in the past", t.String()) } new.Expiry = t diff --git a/backend/db/schemas.go b/backend/db/schemas.go index bdfa60c..62c5a11 100644 --- a/backend/db/schemas.go +++ b/backend/db/schemas.go @@ -12,4 +12,5 @@ type Paste struct { Hash string Content string Expiry time.Time `bson:"expiry"` + Title string } |