feat: complete generate

This commit is contained in:
Rogee
2024-12-04 20:02:30 +08:00
parent d4981f99c2
commit 975eba4183
3 changed files with 24 additions and 6 deletions

View File

@@ -1,8 +1,10 @@
package task
package tasks
import (
"backend/modules/medias"
"backend/modules/tasks"
"backend/providers/app"
"backend/providers/postgres"
"git.ipao.vip/rogeecn/atom"
"git.ipao.vip/rogeecn/atom/container"
@@ -12,6 +14,7 @@ import (
func defaultProviders(providers ...container.ProviderContainer) container.Providers {
return append(container.Providers{
app.DefaultProvider(),
postgres.DefaultProvider(),
}, providers...)
}
@@ -19,15 +22,16 @@ func Command() atom.Option {
return atom.Command(
atom.Name("tasks"),
atom.Short("run tasks"),
atom.Providers(defaultProviders().With(
tasks.Provide,
)),
atom.Command(
atom.Name("discover"),
atom.Arguments(func(cmd *cobra.Command) {
cmd.Flags().String("from", "", "from path")
cmd.Flags().String("to", "", "to path")
}),
atom.Providers(defaultProviders().With(
tasks.Provide,
medias.Provide,
)),
atom.RunE(func(cmd *cobra.Command, args []string) error {
return container.Container.Invoke(func(task *tasks.DiscoverMedias) error {
from, to := cmd.Flag("from").Value.String(), cmd.Flag("to").Value.String()