58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
/*
|
|
* Copyright (c) 2023 Laura Kalb <dev@lauka.net>
|
|
* The code of this project is available under the MIT license. See the LICENSE file for more info.
|
|
*
|
|
*/
|
|
|
|
package app
|
|
|
|
import (
|
|
"codeberg.org/lauralani/humble-bot/db"
|
|
"codeberg.org/lauralani/humble-bot/models"
|
|
"fmt"
|
|
"github.com/jedib0t/go-pretty/v6/table"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
"strconv"
|
|
)
|
|
|
|
func RunQueueShow(cmd *cobra.Command, args []string) {
|
|
items := db.GetAllQueuedItems()
|
|
if len(items) == 0 {
|
|
fmt.Println("The queue is empty.")
|
|
return
|
|
}
|
|
|
|
t := table.NewWriter()
|
|
t.SetStyle(table.StyleRounded)
|
|
t.SetOutputMirror(os.Stdout)
|
|
t.AppendHeader(table.Row{"ID", "Short Name", "URL"})
|
|
|
|
for _, item := range items {
|
|
t.AppendRow(table.Row{item.ID, item.Name, item.URL})
|
|
}
|
|
t.Render()
|
|
}
|
|
|
|
func RunQueueRemove(cmd *cobra.Command, args []string) {
|
|
id, err := strconv.ParseInt(args[0], 10, 64)
|
|
if err != nil {
|
|
log.Error().Str("parameter", args[0]).Msg("Parameter is not a valid number!")
|
|
os.Exit(1)
|
|
}
|
|
var bundle models.QueueItem
|
|
bundle.ID = id
|
|
|
|
err = db.Dequeue(bundle)
|
|
if err != nil {
|
|
log.Error().Int64("parameter", id).Msgf("Couldn't remove item from queue: %q", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
log.Info().Msgf("Deleted item with ID %v from queue", id)
|
|
}
|
|
|
|
func RunQueueFill(cmd *cobra.Command, args []string) {
|
|
UpdateBundles()
|
|
}
|