handler.go

package main

import (
	"fmt"
	"log"
	"strings"
	"time"

	"github.com/bwmarrin/discordgo"
)

// timelineチャンネルを登録する
func registTimelineChannel(s *discordgo.Session, m *discordgo.MessageCreate) {
	if m.Author.ID == s.State.User.ID {
		return
	}

	if m.Message.Content == "!timeline" {
		// timelineチャンネルとしてそのチャンネルをSQLに登録
		// 既に登録している場合、アップデート
		insert, err := db.Prepare("INSERT INTO timeline_channel(guild_id, timeline_id) VALUES(?,?) ON DUPLICATE KEY UPDATE timeline_id = ?")
		if err != nil {
			log.Println(err)
			return
		}
		insert.Exec(m.GuildID, m.ChannelID, m.ChannelID)
		defer insert.Close()
		s.ChannelMessageSend(m.ChannelID, "Set this channel on the timeline")
	}
	return
}

// timelineにメッセージを送る
func sendTimeline(s *discordgo.Session, m *discordgo.MessageCreate) {
	log.Println("create event")
	// bot自身の発言は処理しない
	if m.Author.ID == s.State.User.ID {
		return
	}

	if m.Message.Content == "!timeline" {
		return
	}

	// メッセージが送られたチャンネルを取得
	reciveMessageChannel, err := s.Channel(m.ChannelID)
	if err != nil {
		fmt.Println("Error: could not fetch Channel info,", err)
		return
	}

	// メッセージが送られたチャンネルの名前の prefix が `times_` であれば、処理を続ける
	if strings.HasPrefix(reciveMessageChannel.Name, "times_") {
		timelineChannelID, err := searchTimelineChannelID(m.GuildID)
		if err != nil {
			log.Println(err)
		}
		// timeline チャンネルがない==メッセージの送り先がないため、終了
		if timelineChannelID == "" {
			return
		}

		// timelineチャンネルがある場合、そこに送る
		// messageUpdateならば編集処理、messageCreateならば送信＆DB登録処理
		messageURL := "https://discord.com/channels/" + m.GuildID + "/" + m.ChannelID + "/" + m.ID
		// embedでないメッセージの場合、ImageURLを入れると落ちる
		var reciveMessageImageURL string
		if len(m.Attachments) != 0 {
			reciveMessageImageURL = m.Attachments[0].URL
		}
		messageImage := &discordgo.MessageEmbedImage{
			URL: reciveMessageImageURL,
		}
		messageEmbedFooter := &discordgo.MessageEmbedFooter{
			Text: reciveMessageChannel.Name,
		}
		messageEmbedAuthor := &discordgo.MessageEmbedAuthor{
			Name:    m.Author.Username,
			IconURL: m.Author.AvatarURL(""),
		}

		messageEmbed := &discordgo.MessageEmbed{
			URL:         messageURL,
			Type:        "",
			Title:       "",
			Description: m.Message.Content,
			Timestamp:   time.Now().Format(time.RFC3339),
			Color:       0x90ee90,
			Footer:      messageEmbedFooter,
			Image:       messageImage,
			Thumbnail:   &discordgo.MessageEmbedThumbnail{},
			Video:       &discordgo.MessageEmbedVideo{},
			Provider:    &discordgo.MessageEmbedProvider{},
			Author:      messageEmbedAuthor,
			Fields:      []*discordgo.MessageEmbedField{},
		}
		// 当該メッセージURL+タイムラインのコンテンツと2つ送信する
		linkMessage, err := s.ChannelMessageSend(timelineChannelID, messageURL)
		if err != nil {
			log.Println(err)
		}
		timelineMessage, err := s.ChannelMessageSendEmbed(timelineChannelID, messageEmbed)
		if err != nil {
			log.Println(err)
		}
		// timelineにbotが投稿したメッセージのID, timesに投稿されたメッセージのIDをSQLに登録
		ins, err := db.Prepare("INSERT INTO timeline_message(link_message_id,timeline_message_id, original_message_id) VALUES(?,?,?)")
		ins.Exec(linkMessage.ID, timelineMessage.ID, m.Message.ID)
		defer ins.Close()
	}
	return
}

// timelineのメッセージを編集する
func editTimeline(s *discordgo.Session, mup *discordgo.MessageUpdate) {
	log.Println("update event")

	// embedMessege をsend するとなぜかupdateイベントが起こってnilポインタ参照して落ちる
	// それの回避のため、nilポインタかどうかを確かめている
	if mup.Author == nil || mup.Author.ID == s.State.User.ID {
		return
	}

	// 編集されたメッセージが、既にtimeline_messageテーブルに登録されていれば、
	// 編集された内容をtimelineにも反映する
	timelineMessageID, err := getTimelineMessageID(s, mup.Message.ID)
	if err != nil {
		log.Println(err)
		return
	}
	reciveMessageChannel, err := s.Channel(mup.ChannelID)
	if err != nil {
		log.Println(err)
		return
	}
	if timelineMessageID != "" {
		// 更新点の反映
		var reciveMessageImageURL string
		if len(mup.Attachments) != 0 {
			reciveMessageImageURL = mup.Attachments[0].URL
		}
		messageImage := &discordgo.MessageEmbedImage{
			URL: reciveMessageImageURL,
		}
		messageEmbedFooter := &discordgo.MessageEmbedFooter{
			Text: reciveMessageChannel.Name,
		}
		messageEmbedAuthor := &discordgo.MessageEmbedAuthor{
			Name:    mup.Author.Username,
			IconURL: mup.Author.AvatarURL(""),
		}
		messageEmbed := &discordgo.MessageEmbed{
			URL:         "",
			Type:        "",
			Title:       "",
			Description: mup.Message.Content,
			Timestamp:   "",
			Color:       0x90ee90,
			Footer:      messageEmbedFooter,
			Image:       messageImage,
			Thumbnail:   &discordgo.MessageEmbedThumbnail{},
			Video:       &discordgo.MessageEmbedVideo{},
			Provider:    &discordgo.MessageEmbedProvider{},
			Author:      messageEmbedAuthor,
			Fields:      []*discordgo.MessageEmbedField{},
		}

		timelineChannelID, err := searchTimelineChannelID(mup.GuildID)
		if err != nil {
			log.Println(err)
		}
		log.Println(timelineChannelID, timelineMessageID, messageEmbed)
		_, err = s.ChannelMessageEditEmbed(timelineChannelID, timelineMessageID, messageEmbed)
		if err != nil {
			log.Println(err)
		}
	}
	return
}

// timelineのメッセージを削除する
func delTimeline(s *discordgo.Session, mdel *discordgo.MessageDelete) {
	log.Println("delete event")

	// 削除されたメッセージのIDを元に、タイムライン側もメッセージIDを取得して削除する
	// timelineをdelした時もイベントが発火するが、止める手段がないため重くなってきたら考える
	linkMessageID, timelineMessageID, err := getLinkAndTimelineMessageID(mdel.Message.ID)
	if err != nil {
		log.Println("linkand,", err)
	}
	timelineChannelID, err := searchTimelineChannelID(mdel.GuildID)
	if err != nil {
		log.Println("tlch,", err)
	}
	s.ChannelMessageDelete(timelineChannelID, linkMessageID)
	s.ChannelMessageDelete(timelineChannelID, timelineMessageID)

	// タイムライン側を削除したら、当該SQLのデータも削除
	err = delTimelineFromDB(mdel.Message.ID)
	if err != nil {
		log.Println("deldb,", err)
	}
	return
}