84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"brainbaking.com/go-jamming/app/external"
|
|
"brainbaking.com/go-jamming/common"
|
|
"brainbaking.com/go-jamming/db"
|
|
"brainbaking.com/go-jamming/rest"
|
|
"flag"
|
|
"os"
|
|
|
|
"brainbaking.com/go-jamming/app"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func main() {
|
|
zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
|
|
|
|
verboseFlag := flag.Bool("verbose", false, "Verbose mode (pretty print log, debug level)")
|
|
migrateFlag := flag.Bool("migrate", false, "Run migration scripts for the DB and exit.")
|
|
denylist := flag.String("denylist", "", "Denylist a domain name (also cleans spam from DB)")
|
|
importFile := flag.String("import", "", "Import mentions from an external source (i.e. webmention.io)")
|
|
flag.Parse()
|
|
denylisting := len(*denylist) > 1
|
|
importing := len(*importFile) > 1
|
|
|
|
// logs by default to Stderr (/var/log/syslog). Rolling files possible via lumberjack.
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
if *verboseFlag || *migrateFlag || denylisting || importing {
|
|
log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
}
|
|
|
|
if *migrateFlag {
|
|
migrate()
|
|
os.Exit(0)
|
|
}
|
|
|
|
if denylisting {
|
|
denylistDomain(*denylist)
|
|
os.Exit(0)
|
|
}
|
|
|
|
if importing {
|
|
importWebmentionFile(*importFile)
|
|
os.Exit(0)
|
|
}
|
|
|
|
log.Debug().Msg("Let's a go!")
|
|
app.Start()
|
|
}
|
|
|
|
func importWebmentionFile(file string) {
|
|
log.Info().Str("file", file).Msg("Starting import...")
|
|
|
|
config := common.Configure()
|
|
bootstrapper := external.ImportBootstrapper{
|
|
RestClient: &rest.HttpClient{},
|
|
Conf: config,
|
|
Repo: db.NewMentionRepo(config),
|
|
}
|
|
bootstrapper.Import(file)
|
|
}
|
|
|
|
func denylistDomain(domain string) {
|
|
log.Info().Str("domain", domain).Msg("Denylisting...")
|
|
config := common.Configure()
|
|
config.AddToDenylist(domain)
|
|
|
|
repo := db.NewMentionRepo(config)
|
|
for _, domain := range config.AllowedWebmentionSources {
|
|
repo.CleanupSpam(domain, config.Denylist)
|
|
}
|
|
|
|
log.Info().Msg("Denylist done, exiting.")
|
|
}
|
|
|
|
func migrate() {
|
|
log.Info().Msg("Starting db migration...")
|
|
db.Migrate()
|
|
log.Info().Msg("Migration ended, exiting.")
|
|
}
|