Add fetch-messages subcmd

Author: thinca

subcmd/fetch_messages.go

@@ -0,0 +1,66 @@
+package subcmd
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strconv"
+	"time"
+
+	"github.com/vim-jp/slacklog-generator/internal/slacklog"
+)
+
+func FetchMessages(args []string) error {
+	slackToken := os.Getenv("SLACK_TOKEN")
+	if slackToken == "" {
+		return fmt.Errorf("$SLACK_TOKEN required")
+	}
+
+	if len(args) < 2 {
+		fmt.Println("Usage: go run scripts/main.go update-messages {data-dir} {yyyy-mm-dd}")
+		return nil
+	}
+
+	dataDir := filepath.Clean(args[0])
+
+	loc, err := time.LoadLocation("Asia/Tokyo")
+	if err != nil {
+		return err
+	}
+
+	date, err := time.ParseInLocation("2006-01-02", args[1], loc)
+	if err != nil {
+		return err
+	}
+	dateEndTime := date.AddDate(0, 0, 1)
+
+	channelsJSONPath := filepath.Join(dataDir, "channels.json")
+	channelTable, err := slacklog.NewChannelTable(channelsJSONPath, []string{"*"})
+	if err != nil {
+		return err
+	}
+
+	for _, channel := range channelTable.Channels {
+		channelDir := filepath.Join(dataDir, channel.ID)
+
+		request := slacklog.APIRequest{
+			SlackToken: slackToken,
+			APIMethod:  "conversations.history",
+			ExtraParams: map[string]string{
+				"channel": channel.ID,
+				"oldest":  strconv.FormatInt(date.Unix(), 10) + ".000000",
+				"latest":  strconv.FormatInt(dateEndTime.Unix(), 10) + ".000000",
+				"limit":   "200",
+			},
+			JSONKey: "messages",
+			Reverse: true,
+		}
+		outFile := filepath.Join(channelDir, date.Format("2006-01-02")+".json")
+		err := slacklog.DownloadEntitiesToFile(request, outFile)
+		if err != nil {
+			return err
+		}
+	}
+
+	return nil
+}

subcmd/subcmd.go

@@ -13,6 +13,7 @@ func Run() error {
     convert-exported-logs
     download-emoji
     download-files
+    fetch-messages
     generate-html
     update-channel-list
     update-user-list`)
@@ -28,6 +29,8 @@ func Run() error {
 		return DownloadEmoji(args)
 	case "download-files":
 		return DownloadFiles(args)
+	case "fetch-messages":
+		return FetchMessages(args)
 	case "generate-html":
 		return GenerateHTML(args)
 	case "update-channel-list":