LilyBot.kt

@file:OptIn(PrivilegedIntent::class)

package org.hyacinthbots.lilybot

import com.kotlindiscord.kord.extensions.ExtensibleBot
import com.kotlindiscord.kord.extensions.checks.hasPermission
import com.kotlindiscord.kord.extensions.modules.extra.phishing.DetectionAction
import com.kotlindiscord.kord.extensions.modules.extra.phishing.extPhishing
import com.kotlindiscord.kord.extensions.modules.extra.pluralkit.extPluralKit
import dev.kord.common.entity.Permission
import dev.kord.gateway.Intent
import dev.kord.gateway.PrivilegedIntent
import io.github.oshai.kotlinlogging.KotlinLogging
import org.hyacinthbots.docgenerator.docsGenerator
import org.hyacinthbots.docgenerator.enums.CommandTypes
import org.hyacinthbots.docgenerator.enums.SupportedFileFormat
import org.hyacinthbots.lilybot.database.collections.WelcomeChannelCollection
import org.hyacinthbots.lilybot.database.storage.MongoDBDataAdapter
import org.hyacinthbots.lilybot.extensions.config.Config
import org.hyacinthbots.lilybot.extensions.config.ConfigOptions
import org.hyacinthbots.lilybot.extensions.config.GuildLogging
import org.hyacinthbots.lilybot.extensions.events.AutoThreading
import org.hyacinthbots.lilybot.extensions.events.MemberLogging
import org.hyacinthbots.lilybot.extensions.events.MessageDelete
import org.hyacinthbots.lilybot.extensions.events.MessageEdit
import org.hyacinthbots.lilybot.extensions.events.ModThreadInviting
import org.hyacinthbots.lilybot.extensions.moderation.LockingCommands
import org.hyacinthbots.lilybot.extensions.moderation.ModerationCommands
import org.hyacinthbots.lilybot.extensions.moderation.Report
import org.hyacinthbots.lilybot.extensions.util.GalleryChannel
import org.hyacinthbots.lilybot.extensions.util.Github
import org.hyacinthbots.lilybot.extensions.util.GuildAnnouncements
import org.hyacinthbots.lilybot.extensions.util.InfoCommands
import org.hyacinthbots.lilybot.extensions.util.ModUtilities
import org.hyacinthbots.lilybot.extensions.util.NewsChannelPublishing
import org.hyacinthbots.lilybot.extensions.util.PublicUtilities
import org.hyacinthbots.lilybot.extensions.util.Reminders
import org.hyacinthbots.lilybot.extensions.util.RoleMenu
import org.hyacinthbots.lilybot.extensions.util.StartupHooks
import org.hyacinthbots.lilybot.extensions.util.StatusPing
import org.hyacinthbots.lilybot.extensions.util.Tags
import org.hyacinthbots.lilybot.extensions.util.ThreadControl
import org.hyacinthbots.lilybot.utils.BOT_TOKEN
import org.hyacinthbots.lilybot.utils.ENVIRONMENT
import org.hyacinthbots.lilybot.utils.SENTRY_DSN
import org.hyacinthbots.lilybot.utils.database
import org.hyacinthbots.lilybot.utils.getLoggingChannelWithPerms
import org.kohsuke.github.GitHub
import org.kohsuke.github.GitHubBuilder
import org.quiltmc.community.cozy.modules.welcome.welcomeChannel
import java.io.IOException
import kotlin.io.path.Path
import kotlin.time.Duration.Companion.minutes

lateinit var github: GitHub
private val gitHubLogger = KotlinLogging.logger("GitHub Logger")

val docFile = Path("./docs/commands.md")

suspend fun main() {
	val bot = ExtensibleBot(BOT_TOKEN) {
		database(true)
		dataAdapter(::MongoDBDataAdapter)

		members {
			lockMemberRequests = true // Collect members one at a time to avoid hitting rate limits
			all()
		}

		// Enable the members intent to allow us to get accurate member counts for join logging
		intents {
			+Intent.GuildMembers
		}

		// Add the extensions to the bot
		extensions {
			add(::AutoThreading)
			add(::Config)
			add(::GalleryChannel)
			add(::Github)
			add(::GuildAnnouncements)
			add(::GuildLogging)
			add(::InfoCommands)
			add(::LockingCommands)
			add(::MemberLogging)
			add(::MessageDelete)
			add(::MessageEdit)
			add(::ModThreadInviting)
			add(::ModUtilities)
			add(::ModerationCommands)
			add(::NewsChannelPublishing)
			add(::PublicUtilities)
			add(::Reminders)
			add(::Report)
			add(::RoleMenu)
			add(::StartupHooks)
			add(::StatusPing)
			add(::Tags)
			add(::ThreadControl)

			/*
			The welcome channel extension allows users to designate a YAML file to create a channel with
			a variety of pre-built blocks.
			 */
 			welcomeChannel(WelcomeChannelCollection()) {
 				staffCommandCheck {
 					hasPermission(Permission.BanMembers)
 				}

 				getLogChannel { _, guild ->
 					getLoggingChannelWithPerms(ConfigOptions.UTILITY_LOG, guild)
 				}

 				refreshDuration = 5.minutes
 			}

			/*
			The anti-phishing extension automatically deletes and logs scam links. It also allows users to check links
			with a command. It kicks users who send scam links, rather than ban, to allow them to rejoin if they regain
			control of their account
			 */
			extPhishing {
				appName = "Lily Bot"
				detectionAction = DetectionAction.Kick
				logChannelName = "anti-phishing-logs"
				requiredCommandPermission = null
			}

			extPluralKit()

			sentry {
				enableIfDSN(SENTRY_DSN) // Use the nullable sentry function to allow the bot to be used without a DSN
			}
		}

		docsGenerator {
			enabled = true
			fileFormat = SupportedFileFormat.MARKDOWN
			commandTypes = CommandTypes.ALL
			filePath = docFile
			environment = ENVIRONMENT
			useBuiltinCommandList = true
			botName = "LilyBot"
		}

		// Connect to GitHub to allow the GitHub commands to function
		try {
			github = GitHubBuilder().build()
			gitHubLogger.info { "Connected to GitHub!" }
		} catch (exception: IOException) {
			exception.printStackTrace()
			gitHubLogger.error { "Failed to connect to GitHub!" }
		}
	}

	bot.start()
}