Add display name support for accounts

Author: etbyrd

cmd/msgvault/cmd/addaccount.go

@@ -9,7 +9,8 @@ import (
 )
 
 var (
-	headless bool
+	headless            bool
+	accountDisplayName  string
 )
 
 var addAccountCmd = &cobra.Command{
@@ -20,9 +21,10 @@ var addAccountCmd = &cobra.Command{
 By default, opens a browser for authorization. Use --headless for environments
 without a display (e.g., SSH sessions) to use device code flow instead.
 
-Example:
+Examples:
   msgvault add-account you@gmail.com
-  msgvault add-account you@gmail.com --headless`,
+  msgvault add-account you@gmail.com --headless
+  msgvault add-account you@gmail.com --display-name "Work Account"`,
 	Args: cobra.ExactArgs(1),
 	RunE: func(cmd *cobra.Command, args []string) error {
 		email := args[0]
@@ -70,11 +72,18 @@ Example:
 		}
 
 		// Create source record in database
-		_, err = s.GetOrCreateSource("gmail", email)
+		source, err := s.GetOrCreateSource("gmail", email)
 		if err != nil {
 			return fmt.Errorf("create source: %w", err)
 		}
 
+		// Set display name if provided
+		if accountDisplayName != "" {
+			if err := s.UpdateSourceDisplayName(source.ID, accountDisplayName); err != nil {
+				return fmt.Errorf("set display name: %w", err)
+			}
+		}
+
 		fmt.Printf("\nAccount %s authorized successfully!\n", email)
 		fmt.Println("You can now run: msgvault sync-full", email)
 
@@ -84,5 +93,6 @@ Example:
 
 func init() {
 	addAccountCmd.Flags().BoolVar(&headless, "headless", false, "Use device code flow for headless environments")
+	addAccountCmd.Flags().StringVar(&accountDisplayName, "display-name", "", "Display name for the account (e.g., \"Work\", \"Personal\")")
 	rootCmd.AddCommand(addAccountCmd)
 }

cmd/msgvault/cmd/update_account.go

@@ -0,0 +1,57 @@
+package cmd
+
+import (
+	"fmt"
+
+	"github.com/spf13/cobra"
+	"github.com/wesm/msgvault/internal/store"
+)
+
+var updateDisplayName string
+
+var updateAccountCmd = &cobra.Command{
+	Use:   "update-account <email>",
+	Short: "Update account settings",
+	Long: `Update settings for an existing account.
+
+Currently supports updating the display name for an account.
+
+Examples:
+  msgvault update-account you@gmail.com --display-name "Work"
+  msgvault update-account you@gmail.com --display-name "Personal Email"`,
+	Args: cobra.ExactArgs(1),
+	RunE: func(cmd *cobra.Command, args []string) error {
+		email := args[0]
+
+		if updateDisplayName == "" {
+			return fmt.Errorf("nothing to update: use --display-name to set a display name")
+		}
+
+		dbPath := cfg.DatabasePath()
+		s, err := store.Open(dbPath)
+		if err != nil {
+			return fmt.Errorf("open database: %w", err)
+		}
+		defer s.Close()
+
+		source, err := s.GetSourceByIdentifier(email)
+		if err != nil {
+			return fmt.Errorf("get account: %w", err)
+		}
+		if source == nil {
+			return fmt.Errorf("account not found: %s", email)
+		}
+
+		if err := s.UpdateSourceDisplayName(source.ID, updateDisplayName); err != nil {
+			return fmt.Errorf("update display name: %w", err)
+		}
+
+		fmt.Printf("Updated account %s: display name set to %q\n", email, updateDisplayName)
+		return nil
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(updateAccountCmd)
+	updateAccountCmd.Flags().StringVar(&updateDisplayName, "display-name", "", "Set the display name for the account")
+}

internal/store/sync.go

@@ -245,6 +245,16 @@ func (s *Store) UpdateSourceSyncCursor(sourceID int64, cursor string) error {
 	return err
 }
 
+// UpdateSourceDisplayName updates the display name for a source.
+func (s *Store) UpdateSourceDisplayName(sourceID int64, displayName string) error {
+	_, err := s.db.Exec(`
+		UPDATE sources
+		SET display_name = ?, updated_at = datetime('now')
+		WHERE id = ?
+	`, displayName, sourceID)
+	return err
+}
+
 // GetSourceByIdentifier returns a source by its identifier (email address).
 func (s *Store) GetSourceByIdentifier(identifier string) (*Source, error) {
 	row := s.db.QueryRow(`