[v9]話者選択機能の実装

src/internal/api/voicevox/client.go

@@ -107,3 +107,38 @@ func (c *Client) Synthesize(
 
 	return io.ReadAll(res2.Body)
 }
+
+// GetSpeakers はVOICEVOXの話者一覧を取得します
+func (c *Client) GetSpeakers(ctx context.Context) ([]Speaker, error) {
+	req, err := http.NewRequestWithContext(
+		ctx,
+		http.MethodGet,
+		c.BaseURL+"/speakers",
+		nil,
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	if c.APIKey != "" {
+		req.Header.Set("Authorization", "ApiKey "+c.APIKey)
+	}
+
+	res, err := c.HTTP.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer res.Body.Close()
+
+	if res.StatusCode != 200 {
+		b, _ := io.ReadAll(res.Body)
+		return nil, fmt.Errorf("speakers failed: %s", string(b))
+	}
+
+	var speakers []Speaker
+	if err := json.NewDecoder(res.Body).Decode(&speakers); err != nil {
+		return nil, err
+	}
+
+	return speakers, nil
+}

src/internal/api/voicevox/types.go

@@ -1,3 +1,16 @@
 package voicevox
 
-// 将来 audio_query の struct 化したくなったらここに書く
+// Speaker は /speakers のレスポンスを表します
+type Speaker struct {
+	Name        string         `json:"name"`
+	Styles      []SpeakerStyle `json:"styles"`
+	SpeakerUUID string         `json:"speaker_uuid"`
+	Version     string         `json:"version"`
+}
+
+// SpeakerStyle は話者のスタイル情報です
+type SpeakerStyle struct {
+	ID   int    `json:"id"`
+	Name string `json:"name"`
+	Type string `json:"type"`
+}

src/internal/bot/command/general/help/help-commands.go

@@ -29,7 +29,7 @@ var HelpCommands = []HelpCommand{
 	},
 	{
 		Name:        "/tts <subcommand>",
-		Description: "読み上げにまつわるコマンドです。\nサブコマンド一覧:\n・`join` : ボイスチャンネルに参加します。\n・`leave`: ボイスチャンネルから退出します。\n・`skip` : 現在再生中の音声をスキップします。\n・`speed` : 再生速度を設定します。\n・`dict` : 読み上げ辞書の管理を行います。",
+		Description: "読み上げにまつわるコマンドです。\nサブコマンド一覧:\n・`join` : ボイスチャンネルに参加します。\n・`leave`: ボイスチャンネルから退出します。\n・`skip` : 現在再生中の音声をスキップします。\n・`speed` : 再生速度を設定します。\n・`dict` : 読み上げ辞書の管理を行います。\n・`set` : 読み上げの設定を変更します。",
 		Usage:       "基本はVC接続中に使用を推奨します。",
 	},
 	{
@@ -47,4 +47,9 @@ var HelpCommands = []HelpCommand{
 		Description: "TTSの再生速度を設定します。",
 		Usage:       "VC接続中に`/tts speed 120`のように指定します。",
 	},
+	{
+		Name:        "/tts set voice",
+		Description: "TTSの話者を選択します。",
+		Usage:       "VC接続中に`/tts set voice`で表示される選択肢から話者を選びます。",
+	},
 }

src/internal/bot/command/general/tts.go

@@ -17,6 +17,7 @@ func LoadTtsCommandContext() *discordgo.ApplicationCommand {
 			tts.LoadLeaveCommandContext(),
 			tts.LoadSkipCommandContext(),
 			tts.LoadDictCommandContext(),
+			tts.LoadSetCommandContext(),
 			tts.LoadSpeedCommandContext(),
 		},
 	}
@@ -27,6 +28,7 @@ var ttsHandler = map[string]func(ctx *internal.BotContext, s *discordgo.Session,
 	"leave": tts.Leave,
 	"skip":  tts.Skip,
 	"dict":  tts.Dict,
+	"set":   tts.Set,
 	"speed": tts.Speed,
 }
 

src/internal/bot/command/general/tts/set.go

@@ -0,0 +1,32 @@
+package tts
+
+import (
+	"unibot/internal"
+	"unibot/internal/bot/command/general/tts/set"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+func LoadSetCommandContext() *discordgo.ApplicationCommandOption {
+	return &discordgo.ApplicationCommandOption{
+		Type:        discordgo.ApplicationCommandOptionSubCommandGroup,
+		Name:        "set",
+		Description: "TTSの設定を変更します",
+		Options: []*discordgo.ApplicationCommandOption{
+			set.LoadVoiceCommandContext(),
+		},
+	}
+}
+
+var setHandler = map[string]func(ctx *internal.BotContext, s *discordgo.Session, i *discordgo.InteractionCreate){
+	"voice": set.Voice,
+}
+
+func Set(ctx *internal.BotContext, s *discordgo.Session, i *discordgo.InteractionCreate) {
+	subCommandGroup := i.ApplicationCommandData().Options[0]
+	subCommand := subCommandGroup.Options[0]
+
+	if handler, exists := setHandler[subCommand.Name]; exists {
+		handler(ctx, s, i)
+	}
+}

src/internal/bot/command/general/tts/set/voice.go

@@ -0,0 +1,126 @@
+package set
+
+import (
+	"log"
+	"time"
+	"unibot/internal"
+	"unibot/internal/bot/ttsutil"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+func LoadVoiceCommandContext() *discordgo.ApplicationCommandOption {
+	return &discordgo.ApplicationCommandOption{
+		Type:        discordgo.ApplicationCommandOptionSubCommand,
+		Name:        "voice",
+		Description: "読み上げの話者を設定します",
+	}
+}
+
+func Voice(ctx *internal.BotContext, s *discordgo.Session, i *discordgo.InteractionCreate) {
+	config := ctx.Config
+	memberID, requesterName, requesterAvatar := ttsutil.GetInteractionUser(i)
+	if requesterName == "" {
+		log.Println("Voice: missing user information on interaction")
+		_, err := s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
+			Embeds: &[]*discordgo.MessageEmbed{
+				{
+					Title:       "エラー",
+					Description: "ユーザー情報の取得に失敗しました。",
+					Color:       config.Colors.Error,
+					Timestamp:   time.Now().Format(time.RFC3339),
+				},
+			},
+			Flags: discordgo.MessageFlagsEphemeral,
+		})
+		if err != nil {
+			log.Println("Failed to edit deferred interaction:", err)
+		}
+		return
+	}
+
+	done := make(chan struct{})
+	go func() {
+		select {
+		case <-done:
+			return
+		case <-time.After(3 * time.Minute):
+			_, err := s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
+				Embeds: &[]*discordgo.MessageEmbed{
+					{
+						Title:       "エラー",
+						Description: "話者情報の取得に失敗しました。",
+						Color:       config.Colors.Error,
+						Footer: &discordgo.MessageEmbedFooter{
+							Text:    "Requested by " + requesterName,
+							IconURL: requesterAvatar,
+						},
+						Timestamp: time.Now().Format(time.RFC3339),
+					},
+				},
+			})
+			if err != nil {
+				log.Println("Failed to edit deferred interaction on timeout:", err)
+			}
+		}
+	}()
+	defer close(done)
+
+	speakers, err := ttsutil.FetchSpeakers(ctx)
+	if err != nil {
+		log.Println("Failed to fetch speakers:", err)
+		_, err := s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
+			Embeds: &[]*discordgo.MessageEmbed{
+				{
+					Title:       "エラー",
+					Description: "話者情報の取得に失敗しました。",
+					Color:       config.Colors.Error,
+					Footer: &discordgo.MessageEmbedFooter{
+						Text:    "Requested by " + requesterName,
+						IconURL: requesterAvatar,
+					},
+					Timestamp: time.Now().Format(time.RFC3339),
+				},
+			},
+			Flags: discordgo.MessageFlagsEphemeral,
+		})
+		if err != nil {
+			log.Println("Failed to edit deferred interaction:", err)
+		}
+		return
+	}
+
+	pages := ttsutil.BuildSpeakerPages(speakers, ttsutil.SpeakerPageSize)
+	if len(pages) == 0 {
+		_, err := s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
+			Embeds: &[]*discordgo.MessageEmbed{
+				{
+					Title:       "エラー",
+					Description: "話者情報が取得できませんでした。",
+					Color:       config.Colors.Error,
+					Footer: &discordgo.MessageEmbedFooter{
+						Text:    "Requested by " + requesterName,
+						IconURL: requesterAvatar,
+					},
+					Timestamp: time.Now().Format(time.RFC3339),
+				},
+			},
+			Flags: discordgo.MessageFlagsEphemeral,
+		})
+		if err != nil {
+			log.Println("Failed to edit deferred interaction:", err)
+		}
+		return
+	}
+
+	currentSpeakerID := ttsutil.GetCurrentSpeakerID(ctx, memberID)
+	content, components := ttsutil.BuildVoiceMessage(0, pages, currentSpeakerID)
+
+	_, err = s.InteractionResponseEdit(i.Interaction, &discordgo.WebhookEdit{
+		Content:    &content,
+		Components: &components,
+	})
+	if err != nil {
+		log.Println("Failed to edit deferred interaction:", err)
+	}
+}

src/internal/bot/handler/interaction.go

@@ -29,13 +29,36 @@ func handleApplicationCommand(ctx *internal.BotContext, s *discordgo.Session, i
 		response.Data = &discordgo.InteractionResponseData{
 			Flags: discordgo.MessageFlagsEphemeral,
 		}
+	} else if isTtsSetVoice(i) {
+		response.Data = &discordgo.InteractionResponseData{
+			Flags: discordgo.MessageFlagsEphemeral,
+		}
 	}
 	s.InteractionRespond(i.Interaction, response)
 	if entry, ok := command.Handlers[name]; ok {
 		entry.Handler(ctx, s, i)
 	}
 }
 
+func isTtsSetVoice(i *discordgo.InteractionCreate) bool {
+	if i.ApplicationCommandData().Name != "tts" {
+		return false
+	}
+	options := i.ApplicationCommandData().Options
+	if len(options) == 0 {
+		return false
+	}
+	group := options[0]
+	if group.Type != discordgo.ApplicationCommandOptionSubCommandGroup || group.Name != "set" {
+		return false
+	}
+	if len(group.Options) == 0 {
+		return false
+	}
+	sub := group.Options[0]
+	return sub.Type == discordgo.ApplicationCommandOptionSubCommand && sub.Name == "voice"
+}
+
 func handleMessageComponent(ctx *internal.BotContext, s *discordgo.Session, i *discordgo.InteractionCreate) {
 	customID := i.MessageComponentData().CustomID