68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package service
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type URLService struct {
|
|
secret string
|
|
serverHost string
|
|
expiryDuration time.Duration
|
|
}
|
|
|
|
func NewURLService(secret, serverHost string, expiryMinutes int) *URLService {
|
|
if expiryMinutes <= 0 {
|
|
expiryMinutes = 60
|
|
}
|
|
return &URLService{
|
|
secret: secret,
|
|
serverHost: serverHost,
|
|
expiryDuration: time.Duration(expiryMinutes) * time.Minute,
|
|
}
|
|
}
|
|
|
|
func (s *URLService) sign(fileID string, expires int64) string {
|
|
data := fmt.Sprintf("%s:%d", fileID, expires)
|
|
mac := hmac.New(sha256.New, []byte(s.secret))
|
|
mac.Write([]byte(data))
|
|
return hex.EncodeToString(mac.Sum(nil))
|
|
}
|
|
|
|
func (s *URLService) GenerateDownloadURL(fileUUID string) string {
|
|
expires := time.Now().Add(s.expiryDuration).Unix()
|
|
sig := s.sign(fileUUID, expires)
|
|
|
|
return fmt.Sprintf(
|
|
"%s/api/v1/files/download/%s?expires=%d&sig=%s",
|
|
s.serverHost,
|
|
fileUUID,
|
|
expires,
|
|
sig,
|
|
)
|
|
}
|
|
|
|
func (s *URLService) GenerateThumbnailURL(thumbUUID string) string {
|
|
expires := time.Now().Add(s.expiryDuration).Unix()
|
|
sig := s.sign(thumbUUID, expires)
|
|
|
|
return fmt.Sprintf(
|
|
"%s/api/v1/thumbnails/%s?expires=%d&sig=%s",
|
|
s.serverHost,
|
|
thumbUUID,
|
|
expires,
|
|
sig,
|
|
)
|
|
}
|
|
|
|
func (s *URLService) Validate(fileID, sig string, expires int64) bool {
|
|
if time.Now().Unix() > expires {
|
|
return false
|
|
}
|
|
expected := s.sign(fileID, expires)
|
|
return hmac.Equal([]byte(sig), []byte(expected))
|
|
}
|