Leon Mika
199ff9feb9
- Have got upload images appearing in the post list - Allowed for deleting uploads - Allowed for seeing the upload progress - Fixed the setting of upload properties like the MIME type - Removed the stripe exif logic with just re-encoding PNGs and JPEGs by loading them and saving them
58 lines
1.1 KiB
Go
58 lines
1.1 KiB
Go
package uploadfiles
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"lmika.dev/lmika/weiro/models"
|
|
)
|
|
|
|
type Provider struct {
|
|
baseDir string
|
|
}
|
|
|
|
func New(baseDir string) *Provider {
|
|
return &Provider{
|
|
baseDir: baseDir,
|
|
}
|
|
}
|
|
|
|
func (p *Provider) AdoptFile(site models.Site, up models.Upload, filename string) error {
|
|
fullPath := p.uploadFileName(site, up)
|
|
baseDir := filepath.Dir(fullPath)
|
|
|
|
if err := os.MkdirAll(baseDir, 0755); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := os.Rename(filename, fullPath); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Provider) OpenUpload(site models.Site, up models.Upload) (io.ReadCloser, error) {
|
|
fullPath := p.uploadFileName(site, up)
|
|
return os.Open(fullPath)
|
|
}
|
|
|
|
func (p *Provider) DeleteUpload(site models.Site, up models.Upload) error {
|
|
fullPath := p.uploadFileName(site, up)
|
|
if err := os.Remove(fullPath); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (p *Provider) UploadDir(site models.Site) string {
|
|
return filepath.Join(p.baseDir, site.GUID)
|
|
}
|
|
|
|
func (p *Provider) uploadFileName(site models.Site, up models.Upload) string {
|
|
return filepath.Join(p.baseDir, site.GUID, up.Slug)
|
|
}
|