fix: issues

This commit is contained in:
Rogee
2024-12-06 16:13:28 +08:00
parent 233847f233
commit 7d9ec4ef81
5 changed files with 54 additions and 4 deletions

View File

@@ -3,8 +3,10 @@ package tasks
import (
"backend/modules/medias"
"backend/modules/tasks/discover"
"backend/modules/tasks/store"
"backend/providers/app"
"backend/providers/postgres"
"backend/providers/storage"
"git.ipao.vip/rogeecn/atom"
"git.ipao.vip/rogeecn/atom/container"
@@ -14,6 +16,7 @@ import (
func defaultProviders(providers ...container.ProviderContainer) container.Providers {
return append(container.Providers{
app.DefaultProvider(),
storage.DefaultProvider(),
postgres.DefaultProvider(),
}, providers...)
}
@@ -39,5 +42,21 @@ func Command() atom.Option {
})
}),
),
atom.Command(
atom.Name("store"),
atom.Providers(defaultProviders().With(
medias.Provide,
store.Provide,
)),
atom.Arguments(func(cmd *cobra.Command) {
cmd.Flags().String("from", "", "from path")
}),
atom.RunE(func(cmd *cobra.Command, args []string) error {
return container.Container.Invoke(func(task *store.StoreMedias) error {
from := cmd.Flag("from").Value.String()
return task.RunE(from)
})
}),
),
)
}