package main import ( "os" "log" "time" "flag" "github.com/ChannelMeter/iso8601duration" "git.1750studios.com/gronkhDE/gogronkh/config" "git.1750studios.com/gronkhDE/gogronkh/database" "git.1750studios.com/gronkhDE/gogronkh/image" "git.1750studios.com/gronkhDE/gogronkh/youtube" ) var mode string var slug string var name string var aslug string var youtubeid string var poster string var newslug string var oldslug string var newseason int64 func InitCli() { flag.StringVar(&mode, "mode", "", "Specifies the action to be done. Can be add_lt and merge_lps") // For add_lt flag.StringVar(&slug, "slug", "", "Slug as it would be on gronkh.de") flag.StringVar(&name, "name", "", "Name as it would be on gronkh.de") flag.StringVar(&aslug, "author", "", "Author slug") flag.StringVar(&youtubeid, "youtube", "", "YouTube ID of the video") flag.StringVar(&poster, "poster", "", "Poster URL that should be used") // For merge_lps flag.StringVar(&newslug, "newslug", "", "Slug of the final LP") flag.StringVar(&oldslug, "oldslug", "", "Current slug of LP to be merged") flag.Int64Var(&newseason, "newseason", 0, "Season the episodes of the old LP will become") } func DoCli() { flag.Parse() if mode != "" { if mode == "add_lt" { if (slug == "" || name == "" || aslug == "" || youtubeid == "" || poster == "") { log.Fatalf("slug, name, author, youtube, poster must be specified") } ParseCliLT() } else if mode == "merge_lps" { if (oldslug == "" || newslug == "" || newseason == 0) { log.Fatalf("oldslug, newslug, newseason must be specified") } MergeLPs() } os.Exit(0) } } func ParseCliLT() { var err error var LT database.LetsTest LT.Slug.String = slug LT.Name.String = name LT.PosterS.String, LT.PosterB.String, err = image.ResizeCover(poster) if err != nil { log.Printf("WAR LT %s: Error resizing cover: %+v", slug, err) } var AT database.Author database.Db.Where("slug = ?", aslug).First(&AT) LT.AuthorID = AT.ID LT.Youtube.String = youtubeid ytres, err := youtube.GetVideos([]string {"snippet", "statistics", "status", "contentDetails"}, []string {LT.Youtube.String}, config.C.YoutubeKey) if err != nil || len(ytres.Items) == 0 || ytres.Items[0].Status.UploadStatus != "processed" { log.Printf("ERR LT %s: Video %s is private (%+v)", slug, LT.Youtube.String, err) return } LT.Descr.String = ytres.Items[0].Snippet.Description if thumb, ok := ytres.Items[0].Snippet.Thumbnails["maxres"]; ok { LT.ThumbS.String, LT.ThumbB.String, err = image.ResizeThumb(thumb.Url) } else if thumb, ok := ytres.Items[0].Snippet.Thumbnails["high"]; ok { LT.ThumbS.String, LT.ThumbB.String, err = image.ResizeThumb(thumb.Url) } if err != nil { log.Printf("WAR LT %s: Error resizing thumbnail: %+v", slug, err) } LT.Aired, err = time.Parse(time.RFC3339Nano, ytres.Items[0].Snippet.PublishedAt) if err != nil { log.Printf("ERR LT %s: Failed to parse aired", slug) return } dur, err := duration.FromString(ytres.Items[0].ContentDetails.Duration) if err == nil { LT.Duration.Int64 = int64(dur.ToDuration().Seconds()) } else { log.Printf("ERR LT %s: Failed to parse duration", slug) return } LT.Rating.Float64, LT.Votes.Int64 = youtube.GetRatingAndVotesWithRes(ytres) if err := database.Db.Create(<).Error; err != nil { log.Printf("ERR LT %s: Could not be added to databse (%+v)", slug, err) } } func MergeLPs() { var OldLP database.LetsPlay var NewLP database.LetsPlay database.Db.Where("slug = ?", oldslug).First(&OldLP) database.Db.Where("slug = ?", newslug).First(&NewLP) database.Db.Exec("UPDATE episodes SET season=? WHERE lets_play_id = ?", newseason, OldLP.ID) database.Db.Exec("UPDATE episodes SET lets_play_id=? WHERE lets_play_id = ?", NewLP.ID, OldLP.ID) database.Db.Exec("UPDATE lets_plays SET aired=NULL WHERE id = ?", OldLP.ID) OldLP.MergeID.Int64 = int64(NewLP.ID) OldLP.MergeSeason.Int64 = newseason OldLP.Aired = time.Time{} database.Db.Save(&OldLP) }