handlers.go

@@ -12,7 +12,6 @@ import (
 	"time"
 
 	"gitea.hashru.nl/dsprenkels/rushlink/internal/db"
-	"github.com/google/uuid"
 	"github.com/gorilla/mux"
 	"github.com/pkg/errors"
 )
@@ -94,7 +93,6 @@ func (rl *rushlink) viewPasteHandlerFlags(w http.ResponseWriter, r *http.Request
 	vars := mux.Vars(r)
 	key := vars["key"]
 	var p *db.Paste
-	var fu *db.FileUpload
 	var notFound bool
 	err := rl.db.Transaction(func(tx *db.Database) error {
 		var err error
@@ -102,14 +100,6 @@ func (rl *rushlink) viewPasteHandlerFlags(w http.ResponseWriter, r *http.Request
 		if err != nil {
 			return err
 		}
-		if p != nil && p.Type == db.PasteTypeFileUpload {
-			var id uuid.UUID
-			copy(id[:], p.Content)
-			fu, err = db.GetFileUpload(tx, id)
-			if err != nil {
-				return err
-			}
-		}
 		return nil
 	})
 	if notFound {
@@ -123,12 +113,12 @@ func (rl *rushlink) viewPasteHandlerFlags(w http.ResponseWriter, r *http.Request
 		rl.renderError(w, r, status, err.Error())
 		return
 	}
-	rl.viewPasteHandlerInner(w, r, flags, p, fu)
+	rl.viewPasteHandlerInner(w, r, flags, p)
 }
 
-func (rl *rushlink) viewPasteHandlerInner(w http.ResponseWriter, r *http.Request, flags viewPaste, p *db.Paste, fu *db.FileUpload) {
+func (rl *rushlink) viewPasteHandlerInner(w http.ResponseWriter, r *http.Request, flags viewPaste, p *db.Paste) {
 	if flags&viewShowMeta != 0 {
-		rl.viewPasteHandlerInnerMeta(w, r, p, fu)
+		rl.viewPasteHandlerInnerMeta(w, r, p)
 		return
 	}
 
@@ -136,10 +126,7 @@ func (rl *rushlink) viewPasteHandlerInner(w http.ResponseWriter, r *http.Request
 	case db.PasteStatePresent:
 		switch p.Type {
 		case db.PasteTypeFileUpload:
-			if fu == nil {
+			rl.viewFileUploadHandler(w, r, p.FileUpload)
-				panic(fmt.Sprintf("file for id %v does not exist in database\n", string(p.Content)))
-			}
-			rl.viewFileUploadHandler(w, r, fu)
 			return
 		case db.PasteTypeRedirect:
 			if flags&viewNoRedirect != 0 {
@@ -191,7 +178,7 @@ func (rl *rushlink) viewFileUploadHandler(w http.ResponseWriter, r *http.Request
 	http.ServeContent(w, r, fu.FileName, modtime, file)
 }
 
-func (rl *rushlink) viewPasteHandlerInnerMeta(w http.ResponseWriter, r *http.Request, p *db.Paste, fu *db.FileUpload) {
+func (rl *rushlink) viewPasteHandlerInnerMeta(w http.ResponseWriter, r *http.Request, p *db.Paste) {
 	var cd canDelete
 	deleteToken := getDeleteTokenFromRequest(r)
 	if deleteToken != "" {
@@ -203,8 +190,8 @@ func (rl *rushlink) viewPasteHandlerInnerMeta(w http.ResponseWriter, r *http.Req
 	}
 
 	var fileExt string
-	if fu != nil {
+	if p.FileUpload != nil {
-		fileExt = fu.Ext()
+		fileExt = p.FileUpload.Ext()
 	}
 	data := map[string]interface{}{
 		"Paste":           p,
@@ -290,7 +277,7 @@ func (rl *rushlink) newFileUploadPasteHandler(w http.ResponseWriter, r *http.Req
 			panic(errors.Wrap(err, "saving fileUpload in db"))
 		}
 
-		paste, err = shortenFileUploadID(tx, fu.PubID)
+		paste, err = shortenFileUploadID(tx, fu)
 		return err
 	}); err != nil {
 		panic(err)
@@ -370,25 +357,39 @@ func (rl *rushlink) deletePasteHandler(w http.ResponseWriter, r *http.Request) {
 //
 // Returns the new paste key if the fileUpload was successfully added to the
 // database
-func shortenFileUploadID(tx *db.Database, id uuid.UUID) (*db.Paste, error) {
+func shortenFileUploadID(tx *db.Database, fu *db.FileUpload) (*db.Paste, error) {
-	return shorten(tx, db.PasteTypeFileUpload, id[:])
+	// Generate the paste key
+	pasteKey, err := db.GeneratePasteKey(tx, highOnlineEntropy)
+	if err != nil {
+		return nil, errors.Wrap(err, "generating paste key")
+	}
+
+	// Also generate a deleteToken
+	deleteToken, err := db.GenerateDeleteToken()
+	if err != nil {
+		return nil, errors.Wrap(err, "generating delete token")
+	}
+
+	// Store the new
+	p := db.Paste{
+		Type:        db.PasteTypeFileUpload,
+		State:       db.PasteStatePresent,
+		FileUpload:  fu,
+		Key:         pasteKey,
+		DeleteToken: deleteToken,
+	}
+	if err := p.Save(tx); err != nil {
+		return nil, err
+	}
+	return &p, nil
 }
 
 // Add a new URL to the database
 //
 // Returns the new paste key if the url was successfully shortened
 func shortenURL(tx *db.Database, userURL *url.URL) (*db.Paste, error) {
-	return shorten(tx, db.PasteTypeRedirect, []byte(userURL.String()))
-}
-
-// Add a paste (of any kind) to the database with arbitrary content.
-func shorten(tx *db.Database, ty db.PasteType, content []byte) (*db.Paste, error) {
 	// Generate the paste key
-	var keyEntropy int
+	pasteKey, err := db.GeneratePasteKey(tx, 0)
-	if ty == db.PasteTypeFileUpload || ty == db.PasteTypePaste {
-		keyEntropy = highOnlineEntropy
-	}
-	pasteKey, err := db.GeneratePasteKey(tx, keyEntropy)
 	if err != nil {
 		return nil, errors.Wrap(err, "generating paste key")
 	}
@@ -401,9 +402,9 @@ func shorten(tx *db.Database, ty db.PasteType, content []byte) (*db.Paste, error
 
 	// Store the new key
 	p := db.Paste{
-		Type:        ty,
+		Type:        db.PasteTypeRedirect,
 		State:       db.PasteStatePresent,
-		Content:     content,
+		URL:         userURL.String(),
 		Key:         pasteKey,
 		DeleteToken: deleteToken,
 	}

internal/db/db.go

@@ -35,7 +35,11 @@ var (
 		LogLevel:      logger.Warn,
 		Colorful:      true,
 	})
-	gormConfig = gorm.Config{Logger: gormLogger, PrepareStmt: true}
+	gormConfig = gorm.Config{
+		Logger:                                   gormLogger,
+		PrepareStmt:                              true,
+		DisableForeignKeyConstraintWhenMigrating: true,
+	}
 )
 
 // OpenDBFromEnvironment tries to open an SQL database, described by

internal/db/fileupload.go

@@ -38,6 +38,8 @@ type FileUpload struct {
 	// UUID publically identifies this FileUpload.
 	PubID uuid.UUID `gorm:"uniqueIndex"`
 
+	PasteID uint
+
 	// FileName contains the original filename of this FileUpload.
 	FileName string
 

internal/db/migrate.go

@@ -38,8 +38,69 @@ func Gormigrate(db *gorm.DB) *gormigrate.Gormigrate {
 				}
 				return tx.AutoMigrate(&FileUpload{}, &Paste{})
 			},
-			Rollback: func(tx *gorm.DB) error {
+		},
-				return tx.Migrator().DropTable(&FileUpload{}, &Paste{})
+		{
+			ID: "202107231722",
+			Migrate: func(tx *gorm.DB) error {
+				// Update the schema
+				type FileUpload struct {
+					ID          uint            `gorm:"primaryKey"`
+					State       FileUploadState `gorm:"index"`
+					PubID       uuid.UUID       `gorm:"uniqueIndex"`
+					PasteID     uint
+					FileName    string
+					ContentType string
+					Checksum    uint32
+					CreatedAt   time.Time
+					UpdatedAt   time.Time
+					DeletedAt   gorm.DeletedAt
+				}
+				type Paste struct {
+					ID          uint       `gorm:"primaryKey"`
+					Type        PasteType  `gorm:"index"`
+					State       PasteState `gorm:"index"`
+					Content     []byte
+					URL         string
+					FileUpload  *FileUpload
+					Key         string `gorm:"uniqueIndex"`
+					DeleteToken string
+					CreatedAt   time.Time
+					UpdatedAt   time.Time
+					DeletedAt   gorm.DeletedAt
+				}
+				err := tx.AutoMigrate(&FileUpload{}, &Paste{})
+				if err != nil {
+					return err
+				}
+
+				// Migrate the data
+				pastes := make([]Paste, 0)
+				if err := tx.Model(&Paste{}).Find(&pastes).Error; err != nil {
+					return err
+				}
+				for _, p := range pastes {
+					switch p.Type {
+					case PasteTypeRedirect:
+						p.URL = string(p.Content)
+						tx.Model(p).Select("Content", "URL").Updates(p)
+					case PasteTypeFileUpload:
+						var id uuid.UUID
+						var fus []FileUpload
+						copy(id[:], p.Content)
+						if err := db.Unscoped().Limit(1).Where("pub_id = ?", id).Find(&fus).Error; err != nil {
+							return err
+						}
+						p.FileUpload = &fus[0]
+						tx.Model(p).Select("Content", "FileUpload").Updates(p)
+					default:
+						continue
+					}
+				}
+
+				// Currently there is a bug in GORM, which causes a nil ptr
+				// dereference when you try dropping a column in an sqlite3
+				// database.
+				return nil
 			},
 		},
 	})

internal/db/paste.go

@@ -8,9 +8,9 @@ import (
 	"strings"
 	"time"
 
-	"github.com/google/uuid"
 	"github.com/pkg/errors"
 	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
 )
 
 // PasteType describes the type of Paste (i.e. file, redirect, [...]).
@@ -24,7 +24,8 @@ type Paste struct {
 	ID          uint       `gorm:"primaryKey"`
 	Type        PasteType  `gorm:"index"`
 	State       PasteState `gorm:"index"`
-	Content     []byte
+	URL         string
+	FileUpload  *FileUpload
 	Key         string `gorm:"uniqueIndex"`
 	DeleteToken string
 	CreatedAt   time.Time
@@ -146,7 +147,7 @@ func ValidatePasteKey(key string) error {
 // the key format first.
 func GetPasteNoValidate(db *gorm.DB, key string) (*Paste, error) {
 	var ps []Paste
-	if err := db.Unscoped().Limit(1).Where("key = ?", key).Find(&ps).Error; err != nil {
+	if err := db.Unscoped().Preload(clause.Associations).Where("key = ?", key).Find(&ps).Error; err != nil {
 		return nil, err
 	}
 	if len(ps) == 0 {
@@ -163,16 +164,8 @@ func (p *Paste) Save(db *gorm.DB) error {
 // Delete deletes this Paste from the database.
 func (p *Paste) Delete(db *gorm.DB, fs *FileStore) error {
 	// Remove the (maybe) attached file
-	if p.Type == PasteTypeFileUpload {
+	if p.FileUpload != nil {
-		fuID, err := uuid.FromBytes(p.Content)
+		if err := p.FileUpload.Delete(db, fs); err != nil {
-		if err != nil {
-			return errors.Wrap(err, "failed to parse uuid")
-		}
-		fu, err := GetFileUpload(db, fuID)
-		if err != nil {
-			return errors.Wrap(err, "failed to find file in database")
-		}
-		if err := fu.Delete(db, fs); err != nil {
 			return errors.Wrap(err, "failed to remove file")
 		}
 	}
@@ -180,7 +173,9 @@ func (p *Paste) Delete(db *gorm.DB, fs *FileStore) error {
 	// Wipe the old paste
 	p.Type = PasteTypeUndef
 	p.State = PasteStateDeleted
-	p.Content = []byte{}
+	p.URL = ""
+	p.FileUpload = nil
+
 	if err := db.Save(&p).Error; err != nil {
 		return errors.Wrap(err, "failed to wipe paste in database")
 	}
@@ -199,10 +194,9 @@ func (p *Paste) RedirectURL() *url.URL {
 	if p.Type != PasteTypeRedirect {
 		panic("expected p.Type to be PasteTypeRedirect")
 	}
-	rawurl := string(p.Content)
+	urlParse, err := url.Parse(p.URL)
-	urlParse, err := url.Parse(rawurl)
 	if err != nil {
-		panic(errors.Wrapf(err, "invalid URL ('%v') in database for key '%v'", rawurl, p.Key))
+		panic(errors.Wrapf(err, "invalid URL ('%v') in database for key '%v'", p.URL, p.Key))
 	}
 	return urlParse
 }