hmn/src/admintools/adminproject.go

package admintools

import (
	"context"
	"fmt"

	"git.handmade.network/hmn/hmn/src/db"
	"git.handmade.network/hmn/hmn/src/models"
	"git.handmade.network/hmn/hmn/src/parsing"
	"git.handmade.network/hmn/hmn/src/website"
	"github.com/spf13/cobra"
)

func addProjectCommands(adminCommand *cobra.Command) {
	projectCommand := &cobra.Command{
		Use:   "project",
		Short: "Admin commands for managing projects",
	}
	adminCommand.AddCommand(projectCommand)

	addCreateProjectCommand(projectCommand)
	addProjectTagCommand(projectCommand)
}

func addCreateProjectCommand(projectCommand *cobra.Command) {
	createProjectCommand := &cobra.Command{
		Use:   "create",
		Short: "Create a new project",
		Run: func(cmd *cobra.Command, args []string) {
			name, _ := cmd.Flags().GetString("name")
			slug, _ := cmd.Flags().GetString("slug")
			blurb, _ := cmd.Flags().GetString("blurb")
			description, _ := cmd.Flags().GetString("description")
			personal, _ := cmd.Flags().GetBool("personal")
			userIDs, _ := cmd.Flags().GetIntSlice("userids")

			descParsed := parsing.ParseMarkdown(description, parsing.ForumRealMarkdown)

			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)

			p, err := website.FetchProject(ctx, tx, nil, models.HMNProjectID, website.ProjectsQuery{
				IncludeHidden: true,
			})
			if err != nil {
				panic(err)
			}
			hmn := p.Project

			newProjectID, err := db.QueryInt(ctx, tx,
				`
				INSERT INTO handmade_project (
					slug,
					name,
					blurb,
					description,
					color_1,
					color_2,
					featured,
					hidden,
					descparsed,
					blog_enabled,
					forum_enabled,
					all_last_updated,
					annotation_last_updated,
					blog_last_updated,
					forum_last_updated,
					lifecycle,
					date_approved,
					date_created,
					bg_flags,
					library_enabled,
					personal
				) VALUES (
					$1,							-- slug
					$2,							-- name
					$3,							-- blurb
					$4,							-- description
					$5,							-- color_1
					$6,							-- color_2
					FALSE,						-- featured
					FALSE,						-- hidden
					$7,							-- descparsed
					FALSE,						-- blog_enabled
					FALSE,						-- forum_enabled
					NOW(),						-- all_last_updated
					'epoch',					-- annotation_last_updated
					'epoch',					-- blog_last_updated
					'epoch',					-- forum_last_updated
					$8,							-- lifecycle
					NOW(),						-- date_approved
					NOW(),						-- date_created
					0,							-- bg_flags
					FALSE,						-- library_enabled
					$9							-- personal
				)
				RETURNING id
				`,
				slug,
				name,
				blurb,
				description,
				hmn.Color1,
				hmn.Color2,
				descParsed,
				models.ProjectLifecycleActive,
				personal,
			)
			if err != nil {
				panic(err)
			}

			for _, userID := range userIDs {
				_, err := tx.Exec(ctx,
					`
					INSERT INTO handmade_user_projects (user_id, project_id)
					VALUES ($1, $2)
					`,
					userID,
					newProjectID,
				)
				if err != nil {
					panic(err)
				}
			}

			err = tx.Commit(ctx)
			if err != nil {
				panic(err)
			}

			fmt.Printf("Created new project with id: %d\n", newProjectID)
		},
	}
	createProjectCommand.Flags().String("name", "", "")
	createProjectCommand.Flags().String("slug", "", "")
	createProjectCommand.Flags().String("blurb", "", "")
	createProjectCommand.Flags().String("description", "", "")
	createProjectCommand.Flags().Bool("personal", true, "")
	createProjectCommand.Flags().IntSlice("userids", nil, "")
	createProjectCommand.MarkFlagRequired("name")
	createProjectCommand.MarkFlagRequired("userids")
	projectCommand.AddCommand(createProjectCommand)
}

func addProjectTagCommand(projectCommand *cobra.Command) {
	projectTagCommand := &cobra.Command{
		Use:   "tag",
		Short: "Create or update a project's tag",
		Run: func(cmd *cobra.Command, args []string) {
			projectID, _ := cmd.Flags().GetInt("projectid")
			tag, _ := cmd.Flags().GetString("tag")

			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)

			p, err := website.FetchProject(ctx, tx, nil, projectID, website.ProjectsQuery{
				IncludeHidden: true,
			})
			if err != nil {
				panic(err)
			}

			if p.Project.TagID == nil {
				// Create a tag
				tagID, err := db.QueryInt(ctx, tx,
					`
					INSERT INTO tags (text) VALUES ($1)
					RETURNING id
					`,
					tag,
				)
				if err != nil {
					panic(err)
				}

				// Attach it to the project
				_, err = tx.Exec(ctx,
					`
					UPDATE handmade_project
					SET tag = $1
					WHERE id = $2
					`,
					tagID, projectID,
				)
				if err != nil {
					panic(err)
				}
			} else {
				// Update the text of an existing one
				_, err := tx.Exec(ctx,
					`
					UPDATE tags
					SET text = $1
					WHERE id = (SELECT tag FROM handmade_project WHERE id = $2)
					`,
					tag, projectID,
				)
				if err != nil {
					panic(err)
				}
			}

			err = tx.Commit(ctx)
			if err != nil {
				panic(err)
			}

			fmt.Printf("Project now has tag: %s\n", tag)
		},
	}
	projectTagCommand.Flags().Int("projectid", 0, "")
	projectTagCommand.Flags().String("tag", "", "")
	projectTagCommand.MarkFlagRequired("projectid")
	projectTagCommand.MarkFlagRequired("tag")
	projectCommand.AddCommand(projectTagCommand)
}