main.go

package main

import (
	"flag"
	"github.com/winterssy/music-get/common"
	"github.com/winterssy/music-get/handler"
	"github.com/winterssy/music-get/utils"
	"github.com/winterssy/music-get/utils/logger"
)

func main() {
	if len(flag.Args()) == 0 {
		logger.Error.Fatal("Missing music address")
	}

	if err := utils.BuildPathIfNotExist(common.MP3DownloadDir); err != nil {
		logger.Error.Fatalf("Failed to build path: %s: %s", common.MP3DownloadDir, err)
	}

	url := flag.Args()[0]
	req, err := handler.Parse(url)
	if err != nil {
		logger.Error.Fatal(err)
	}

	if err = req.Do(); err != nil {
		logger.Error.Fatal(err)
	}

	mp3List, err := req.Extract()
	if err != nil {
		logger.Error.Fatal(err)
	}

	n := common.MP3ConcurrentDownloadTasksNumber
	if n > common.MaxConcurrentDownloadTasksNumber {
		n = common.MaxConcurrentDownloadTasksNumber
	}
	switch {
	case n > 1:
		handler.ConcurrentDownload(mp3List, n)
	default:
		handler.SingleDownload(mp3List)
	}
}