109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
/*
|
|
Yet another powerful customizable ActivityPub relay server written in Go.
|
|
Run Activity-Relay
|
|
API Server
|
|
./Activity-Relay -c <Path of config file> server
|
|
Config
|
|
YAML Format
|
|
ACTOR_PEM: actor.pem
|
|
REDIS_URL: redis://localhost:6379
|
|
RELAY_BIND: 0.0.0.0:8080
|
|
RELAY_DOMAIN: relay.toot.yukimochi.jp
|
|
RELAY_SERVICENAME: YUKIMOCHI Toot Relay Service
|
|
JOB_CONCURRENCY: 50
|
|
RELAY_SUMMARY: |
|
|
YUKIMOCHI Toot Relay Service is Running by Activity-Relay
|
|
RELAY_ICON: https://example.com/example_icon.png
|
|
RELAY_IMAGE: https://example.com/example_image.png
|
|
Environment Variable
|
|
This is Optional : When config file not exist, use environment variables.
|
|
- ACTOR_PEM
|
|
- REDIS_URL
|
|
- RELAY_BIND
|
|
- RELAY_DOMAIN
|
|
- RELAY_SERVICENAME
|
|
- JOB_CONCURRENCY
|
|
- RELAY_SUMMARY
|
|
- RELAY_ICON
|
|
- RELAY_IMAGE
|
|
*/
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
"github.com/yukimochi/Activity-Relay/api"
|
|
"github.com/yukimochi/Activity-Relay/models"
|
|
)
|
|
|
|
var (
|
|
version string
|
|
|
|
globalConfig *models.RelayConfig
|
|
)
|
|
|
|
func main() {
|
|
var app = buildCommand()
|
|
app.PersistentFlags().StringP("config", "c", "config.yml", "Path of config file.")
|
|
|
|
app.Execute()
|
|
}
|
|
|
|
func buildCommand() *cobra.Command {
|
|
var server = &cobra.Command{
|
|
Use: "server",
|
|
Short: "Activity-Relay API Server",
|
|
Long: "Activity-Relay API Server is providing WebFinger API, ActivityPub inbox",
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
initConfig(cmd, args)
|
|
fmt.Println(globalConfig.DumpWelcomeMessage("API Server", version))
|
|
err := api.Entrypoint(globalConfig, version)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err.Error())
|
|
os.Exit(1)
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var app = &cobra.Command{
|
|
Short: "YUKIMOCHI Activity-Relay",
|
|
Long: "YUKIMOCHI Activity-Relay - ActivityPub Relay Server",
|
|
}
|
|
app.AddCommand(server)
|
|
|
|
return app
|
|
}
|
|
|
|
func initConfig(cmd *cobra.Command, args []string) {
|
|
configPath := cmd.Flag("config").Value.String()
|
|
file, err := os.Open(configPath)
|
|
defer file.Close()
|
|
|
|
if err == nil {
|
|
viper.SetConfigType("yaml")
|
|
viper.ReadConfig(file)
|
|
} else {
|
|
fmt.Fprintln(os.Stderr, "Config file not exist. Use environment variables.")
|
|
|
|
viper.BindEnv("ACTOR_PEM")
|
|
viper.BindEnv("REDIS_URL")
|
|
viper.BindEnv("RELAY_BIND")
|
|
viper.BindEnv("RELAY_DOMAIN")
|
|
viper.BindEnv("RELAY_SERVICENAME")
|
|
viper.BindEnv("JOB_CONCURRENCY")
|
|
viper.BindEnv("RELAY_SUMMARY")
|
|
viper.BindEnv("RELAY_ICON")
|
|
viper.BindEnv("RELAY_IMAGE")
|
|
}
|
|
|
|
globalConfig, err = models.NewRelayConfig()
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, err.Error())
|
|
os.Exit(1)
|
|
}
|
|
}
|