From 57d4216d2deb3aec226c5911d4eaf0dfd6fec92a Mon Sep 17 00:00:00 2001 From: Ben Visness Date: Sat, 28 Aug 2021 10:16:02 -0500 Subject: [PATCH] Add tools for creating subforums and moving threads --- src/admintools/admintools.go | 176 ++++++++++++++++++++++++++++++++++- 1 file changed, 173 insertions(+), 3 deletions(-) diff --git a/src/admintools/admintools.go b/src/admintools/admintools.go index 97c532d4..63a39931 100644 --- a/src/admintools/admintools.go +++ b/src/admintools/admintools.go @@ -5,6 +5,7 @@ import ( "errors" "fmt" "os" + "strconv" "time" "git.handmade.network/hmn/hmn/src/auth" @@ -21,6 +22,12 @@ import ( ) func init() { + adminCommand := &cobra.Command{ + Use: "admin", + Short: "Miscellaneous admin commands", + } + website.WebsiteCommand.AddCommand(adminCommand) + setPasswordCommand := &cobra.Command{ Use: "setpassword [username] [new password]", Short: "Replace a user's password", @@ -61,7 +68,7 @@ func init() { fmt.Printf("Successfully updated password for '%s'\n", canonicalUsername) }, } - website.WebsiteCommand.AddCommand(setPasswordCommand) + adminCommand.AddCommand(setPasswordCommand) activateUserCommand := &cobra.Command{ Use: "activateuser [username]", @@ -90,7 +97,7 @@ func init() { fmt.Printf("User has been successfully activated.\n\n") }, } - website.WebsiteCommand.AddCommand(activateUserCommand) + adminCommand.AddCommand(activateUserCommand) sendTestMailCommand := &cobra.Command{ Use: "sendtestmail [type] [toAddress] [toName]", @@ -126,5 +133,168 @@ func init() { } }, } - website.WebsiteCommand.AddCommand(sendTestMailCommand) + adminCommand.AddCommand(sendTestMailCommand) + + createSubforumCommand := &cobra.Command{ + Use: "createsubforum", + Short: "Create a new subforum", + Run: func(cmd *cobra.Command, args []string) { + name, _ := cmd.Flags().GetString("name") + slug, _ := cmd.Flags().GetString("slug") + blurb, _ := cmd.Flags().GetString("blurb") + parentSlug, _ := cmd.Flags().GetString("parent_slug") + projectSlug, _ := cmd.Flags().GetString("project_slug") + + ctx := context.Background() + conn := db.NewConnPool(1, 1) + defer conn.Close() + + tx, err := conn.Begin(ctx) + if err != nil { + panic(err) + } + defer tx.Rollback(ctx) + + projectId, err := db.QueryInt(ctx, tx, `SELECT id FROM handmade_project WHERE slug = $1`, projectSlug) + if err != nil { + panic(err) + } + + var parentId *int + if parentSlug == "" { + // Select the root subforum + id, err := db.QueryInt(ctx, tx, + `SELECT id FROM handmade_subforum WHERE parent_id IS NULL AND project_id = $1`, + projectId, + ) + if err != nil { + panic(err) + } + parentId = &id + } else { + // Select the parent + id, err := db.QueryInt(ctx, tx, + `SELECT id FROM handmade_subforum WHERE slug = $1 AND project_id = $2`, + parentSlug, projectId, + ) + if err != nil { + panic(err) + } + parentId = &id + } + + newId, err := db.QueryInt(ctx, tx, + ` + INSERT INTO handmade_subforum (name, slug, blurb, parent_id, project_id) + VALUES ($1, $2, $3, $4, $5) + RETURNING id + `, + name, + slug, + blurb, + parentId, + projectId, + ) + if err != nil { + panic(err) + } + + err = tx.Commit(ctx) + if err != nil { + panic(err) + } + + fmt.Printf("Created new subforum with id: %d\n", newId) + }, + } + createSubforumCommand.Flags().String("name", "", "") + createSubforumCommand.Flags().String("slug", "", "") + createSubforumCommand.Flags().String("blurb", "", "") + createSubforumCommand.Flags().String("parent_slug", "", "") + createSubforumCommand.Flags().String("project_slug", "", "") + createSubforumCommand.MarkFlagRequired("name") + createSubforumCommand.MarkFlagRequired("slug") + createSubforumCommand.MarkFlagRequired("project_slug") + adminCommand.AddCommand(createSubforumCommand) + + moveThreadsToSubforumCommand := &cobra.Command{ + Use: "movethreadstosubforum [...]", + Short: "Move threads to a subforum, changing their type if necessary", + Run: func(cmd *cobra.Command, args []string) { + projectSlug, _ := cmd.Flags().GetString("project_slug") + subforumSlug, _ := cmd.Flags().GetString("subforum_slug") + + ctx := context.Background() + conn := db.NewConnPool(1, 1) + defer conn.Close() + + tx, err := conn.Begin(ctx) + if err != nil { + panic(err) + } + defer tx.Rollback(ctx) + + projectId, err := db.QueryInt(ctx, tx, `SELECT id FROM handmade_project WHERE slug = $1`, projectSlug) + if err != nil { + panic(err) + } + + subforumId, err := db.QueryInt(ctx, tx, + `SELECT id FROM handmade_subforum WHERE slug = $1 AND project_id = $2`, + subforumSlug, projectId, + ) + if err != nil { + panic(err) + } + + var threadIds []int + for _, threadIdStr := range args { + threadId, err := strconv.Atoi(threadIdStr) + if err != nil { + fmt.Printf("Couldn't move thread '%s': couldn't parse ID\n", threadIdStr) + continue + } + threadIds = append(threadIds, threadId) + } + + threadsTag, err := tx.Exec(ctx, + ` + UPDATE handmade_thread + SET + project_id = $2, + subforum_id = $3, + personal_article_user_id = NULL, + type = 2 + WHERE + id = ANY ($1) + `, + threadIds, + projectId, + subforumId, + ) + + postsTag, err := tx.Exec(ctx, + ` + UPDATE handmade_post + SET + thread_type = 2 + WHERE + thread_id = ANY ($1) + `, + threadIds, + ) + + err = tx.Commit(ctx) + if err != nil { + panic(err) + } + + fmt.Printf("Successfully moved %d threads (and %d posts).\n", threadsTag.RowsAffected(), postsTag.RowsAffected()) + }, + } + moveThreadsToSubforumCommand.Flags().String("project_slug", "", "") + moveThreadsToSubforumCommand.Flags().String("subforum_slug", "", "") + moveThreadsToSubforumCommand.MarkFlagRequired("project_slug") + moveThreadsToSubforumCommand.MarkFlagRequired("subforum_slug") + adminCommand.AddCommand(moveThreadsToSubforumCommand) }