1// SPDX-FileCopyrightText: Amolith <amolith@secluded.site>
 2//
 3// SPDX-License-Identifier: AGPL-3.0-or-later
 4
 5package person
 6
 7import (
 8	"fmt"
 9
10	"git.secluded.site/go-lunatask"
11	"git.secluded.site/lune/internal/client"
12	"git.secluded.site/lune/internal/completion"
13	"git.secluded.site/lune/internal/ui"
14	"git.secluded.site/lune/internal/validate"
15	"github.com/spf13/cobra"
16)
17
18// UpdateCmd updates a person. Exported for potential use by shortcuts.
19var UpdateCmd = &cobra.Command{
20	Use:   "update ID",
21	Short: "Update a person",
22	Long: `Update an existing person in Lunatask.
23
24Accepts a UUID or lunatask:// deep link.
25Only specified flags are modified; other fields remain unchanged.`,
26	Args: cobra.ExactArgs(1),
27	RunE: runUpdate,
28}
29
30func init() {
31	UpdateCmd.Flags().String("first", "", "First name")
32	UpdateCmd.Flags().String("last", "", "Last name")
33	UpdateCmd.Flags().StringP("relationship", "r", "", "Relationship strength")
34
35	_ = UpdateCmd.RegisterFlagCompletionFunc("relationship", completion.Relationships)
36}
37
38func runUpdate(cmd *cobra.Command, args []string) error {
39	id, err := validate.Reference(args[0])
40	if err != nil {
41		return err
42	}
43
44	apiClient, err := client.New()
45	if err != nil {
46		return err
47	}
48
49	builder := apiClient.NewPersonUpdate(id)
50
51	applyUpdateNames(cmd, builder)
52
53	if err := applyUpdateRelationship(cmd, builder); err != nil {
54		return err
55	}
56
57	person, err := builder.Update(cmd.Context())
58	if err != nil {
59		return err
60	}
61
62	fmt.Fprintln(cmd.OutOrStdout(), ui.Success.Render("Updated person: "+person.ID))
63
64	return nil
65}
66
67func applyUpdateNames(cmd *cobra.Command, builder *lunatask.PersonUpdateBuilder) {
68	if first, _ := cmd.Flags().GetString("first"); first != "" {
69		builder.FirstName(first)
70	}
71
72	if last, _ := cmd.Flags().GetString("last"); last != "" {
73		builder.LastName(last)
74	}
75}
76
77func applyUpdateRelationship(cmd *cobra.Command, builder *lunatask.PersonUpdateBuilder) error {
78	rel, _ := cmd.Flags().GetString("relationship")
79	if rel == "" {
80		return nil
81	}
82
83	strength, err := validate.RelationshipStrength(rel)
84	if err != nil {
85		return err
86	}
87
88	builder.WithRelationshipStrength(strength)
89
90	return nil
91}