107 lines
3.2 KiB
Go
107 lines
3.2 KiB
Go
package sql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/filebrowser/filebrowser/v2/share"
|
|
)
|
|
|
|
type shareBackend struct {
|
|
db *sql.DB
|
|
dbType string
|
|
}
|
|
|
|
type linkRecord interface {
|
|
Scan(dest ...interface{}) error
|
|
}
|
|
|
|
func InitSharesTable(db *sql.DB, dbType string) error {
|
|
sql := fmt.Sprintf("create table if not exists %s (hash text, path text, userid integer, expire integer, passwordhash text, token text)", quoteName(dbType, SharesTable))
|
|
_, err := db.Exec(sql)
|
|
checkError(err, "Fail to InitSharesTable")
|
|
return err
|
|
}
|
|
|
|
func parseLink(row linkRecord) (*share.Link, error) {
|
|
path := ""
|
|
hash := ""
|
|
userid := uint(0)
|
|
expire := int64(0)
|
|
passwordhash := ""
|
|
token := ""
|
|
err := row.Scan(&path, &hash, &userid, &expire, &passwordhash, &token)
|
|
if checkError(err, "Fail to parse record for share.Link") {
|
|
return nil, err
|
|
}
|
|
link := share.Link{}
|
|
link.Path = path
|
|
link.Hash = hash
|
|
link.UserID = userid
|
|
link.Expire = expire
|
|
link.PasswordHash = passwordhash
|
|
link.Token = token
|
|
return &link, nil
|
|
}
|
|
|
|
func queryLinks(db *sql.DB, dbType string, condition string) ([]*share.Link, error) {
|
|
sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from %s", quoteName(dbType, SharesTable))
|
|
if len(condition) > 0 {
|
|
sql = sql + " where " + condition
|
|
}
|
|
rows, err := db.Query(sql)
|
|
if checkError(err, "Fail to Query links") {
|
|
return nil, err
|
|
}
|
|
var links []*share.Link = []*share.Link{}
|
|
for rows.Next() {
|
|
link, err := parseLink(rows)
|
|
if checkError(err, "Fail to parse record for share.Link") {
|
|
continue
|
|
}
|
|
links = append(links, link)
|
|
}
|
|
return links, nil
|
|
}
|
|
|
|
func (s shareBackend) All() ([]*share.Link, error) {
|
|
return queryLinks(s.db, s.dbType, "")
|
|
}
|
|
|
|
func (s shareBackend) FindByUserID(id uint) ([]*share.Link, error) {
|
|
condition := fmt.Sprintf("userid=%d", id)
|
|
return queryLinks(s.db, s.dbType, condition)
|
|
}
|
|
|
|
func (s shareBackend) GetByHash(hash string) (*share.Link, error) {
|
|
sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from %s where hash='%s'", quoteName(s.dbType, SharesTable), hash)
|
|
return parseLink(s.db.QueryRow(sql))
|
|
}
|
|
|
|
func (s shareBackend) GetPermanent(path string, id uint) (*share.Link, error) {
|
|
sql := fmt.Sprintf("select hash, path, userid, expire, passwordhash, token from %s where path='%s' and userid=%d", quoteName(s.dbType, SharesTable), path, id)
|
|
return parseLink(s.db.QueryRow(sql))
|
|
}
|
|
|
|
func (s shareBackend) Gets(path string, id uint) ([]*share.Link, error) {
|
|
condition := fmt.Sprintf("userid=%d and path='%s'", id, path)
|
|
return queryLinks(s.db, s.dbType, condition)
|
|
}
|
|
func (s shareBackend) Save(l *share.Link) error {
|
|
sql := fmt.Sprintf("insert into %s (hash, path, userid, expire, passwordhash, token) values('%s', '%s', %d, %d, '%s', '%s')", quoteName(s.dbType, SharesTable), l.Hash, l.Path, l.UserID, l.Expire, l.PasswordHash, l.Token)
|
|
_, err := s.db.Exec(sql)
|
|
checkError(err, "Fail to Save share")
|
|
return err
|
|
}
|
|
func (s shareBackend) Delete(hash string) error {
|
|
sql := fmt.Sprintf("DELETE FROM %s WHERE hash='%s'", quoteName(s.dbType, SharesTable), hash)
|
|
_, err := s.db.Exec(sql)
|
|
checkError(err, "Fail to Delete share")
|
|
return err
|
|
}
|
|
|
|
func newShareBackend(db *sql.DB, dbType string) shareBackend {
|
|
InitSharesTable(db, dbType)
|
|
return shareBackend{db: db, dbType: dbType}
|
|
}
|