hmn/src/admintools/adminproject.go

232 lines
5.3 KiB
Go
Raw Normal View History

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)
}