mirror of
https://github.com/photoprism/photoprism.git
synced 2025-12-12 00:34:13 +01:00
134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/urfave/cli/v2"
|
|
|
|
"github.com/photoprism/photoprism/internal/config"
|
|
"github.com/photoprism/photoprism/internal/photoprism"
|
|
"github.com/photoprism/photoprism/internal/photoprism/get"
|
|
"github.com/photoprism/photoprism/internal/photoprism/ytdl"
|
|
"github.com/photoprism/photoprism/pkg/clean"
|
|
"github.com/photoprism/photoprism/pkg/fs"
|
|
"github.com/photoprism/photoprism/pkg/media/http/scheme"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
// DownloadCommand configures the command name, flags, and action.
|
|
var DownloadCommand = &cli.Command{
|
|
Name: "download",
|
|
Aliases: []string{"dl"},
|
|
Usage: "Imports media from a URL",
|
|
ArgsUsage: "[url]",
|
|
Flags: []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "dest",
|
|
Aliases: []string{"d"},
|
|
Usage: "relative originals `PATH` to which the files should be imported",
|
|
},
|
|
},
|
|
Action: downloadAction,
|
|
}
|
|
|
|
// downloadAction downloads and import media from a URL.
|
|
func downloadAction(ctx *cli.Context) error {
|
|
start := time.Now()
|
|
|
|
conf, err := InitConfig(ctx)
|
|
|
|
_, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// very if copy directory exist and is writable
|
|
if conf.ReadOnly() {
|
|
return config.ErrReadOnly
|
|
}
|
|
|
|
conf.InitDb()
|
|
defer conf.Shutdown()
|
|
|
|
// Get URL from first argument.
|
|
sourceUrl, sourceErr := url.Parse(strings.TrimSpace(ctx.Args().First()))
|
|
|
|
if sourceErr != nil {
|
|
return sourceErr
|
|
} else if sourceUrl.Scheme != scheme.Http && sourceUrl.Scheme != scheme.Https {
|
|
return fmt.Errorf("invalid download URL scheme %s", clean.Log(sourceUrl.Scheme))
|
|
}
|
|
|
|
var destFolder string
|
|
if ctx.IsSet("dest") {
|
|
destFolder = clean.UserPath(ctx.String("dest"))
|
|
} else {
|
|
destFolder = conf.ImportDest()
|
|
}
|
|
|
|
log.Infof("importing media from %s to %s", sourceUrl.String(), filepath.Join(conf.OriginalsPath(), destFolder))
|
|
|
|
result, err := ytdl.New(context.Background(), sourceUrl.String(), ytdl.Options{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var downloadPath, downloadFile string
|
|
|
|
downloadPath = filepath.Join(conf.TempPath(), "download_"+rnd.Base36(12))
|
|
|
|
if err = fs.MkdirAll(downloadPath); err != nil {
|
|
return err
|
|
}
|
|
|
|
defer os.RemoveAll(downloadPath)
|
|
|
|
if dlName := clean.DlName(result.Info.Title); dlName != "" {
|
|
downloadFile = dlName + fs.ExtMp4
|
|
} else {
|
|
downloadFile = time.Now().Format("20060102_150405") + fs.ExtMp4
|
|
}
|
|
|
|
downloadFilePath := filepath.Join(downloadPath, downloadFile)
|
|
downloadResult, err := result.Download(context.Background(), "best")
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
defer downloadResult.Close()
|
|
|
|
file, err := os.Create(downloadFilePath)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err = io.Copy(file, downloadResult); err != nil {
|
|
file.Close()
|
|
return err
|
|
}
|
|
|
|
file.Close()
|
|
|
|
w := get.Import()
|
|
opt := photoprism.ImportOptionsMove(downloadPath, destFolder)
|
|
|
|
w.Start(opt)
|
|
|
|
elapsed := time.Since(start)
|
|
|
|
log.Infof("completed in %s", elapsed)
|
|
|
|
return nil
|
|
}
|