main.go

package main

import (
	"log"
	"os"
	"os/signal"
	"regexp"

	"github.com/robfig/cron/v3"
	"github.com/tpc3/Noa/lib/config"
	"github.com/tpc3/Noa/lib/markov"
	misskeyapi "github.com/tpc3/Noa/lib/misskeyApi"
)

func main() {
	botID := misskeyapi.MisskeyGetuserID(config.Loadconfig.Misskey.Token)
	c := cron.New()
	c.AddFunc("@hourly", func() { runMarkov(botID) })
	c.AddFunc("@weekly", func() { config.GetTlds() })
	c.Start()
	log.Print("Noa is now running!")
	stop := make(chan os.Signal, 1)
	signal.Notify(stop, os.Interrupt)
	<-stop
}

func runMarkov(botID string) {
	var text string
	markovBlock := [][]string{}
	notes, err := misskeyapi.MisskeyGetnotesRequest(config.Loadconfig.Misskey.Token, botID)
	if err != nil {
		return
	}

	// Create text
	var okBlacklist bool
	var okTld bool
	for i := 0; i < 100; i++ {
		for _, note := range notes {
			data := markov.ParseToNode(note)
			elems := markov.GetMarkovBlocks(data)
			markovBlock = append(markovBlock, elems...)
		}

		noteElemset := markov.MarkovChainExec(markovBlock)
		text = markov.TextGenerate(noteElemset)

		okBlacklist = true
		for _, v := range config.Loadconfig.TextBlacklist {
			check := regexp.MustCompile(v)
			if check.MatchString(text) {
				okBlacklist = false
			}
		}

		okTld = true
		okTld = config.CheckTld(text)

		if !okBlacklist || !okTld {
			continue
		}
	}

	if okBlacklist && okTld {
		err = misskeyapi.MisskeySendnotesRequest(config.Loadconfig.Misskey.Token, text)
		if err != nil {
			return
		}
	} else {
		log.Println("err")
	}
}