107 lines
3.1 KiB
Go
107 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/rsa"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"time"
|
|
|
|
"github.com/RichardKnop/machinery/v1"
|
|
"github.com/RichardKnop/machinery/v1/config"
|
|
"github.com/go-redis/redis"
|
|
cache "github.com/patrickmn/go-cache"
|
|
"github.com/spf13/viper"
|
|
activitypub "github.com/yukimochi/Activity-Relay/ActivityPub"
|
|
keyloader "github.com/yukimochi/Activity-Relay/KeyLoader"
|
|
state "github.com/yukimochi/Activity-Relay/State"
|
|
)
|
|
|
|
var (
|
|
version string
|
|
|
|
// Actor : Relay's Actor
|
|
Actor activitypub.Actor
|
|
|
|
// WebfingerResource : Relay's Webfinger resource
|
|
WebfingerResource activitypub.WebfingerResource
|
|
|
|
hostURL *url.URL
|
|
hostPrivatekey *rsa.PrivateKey
|
|
relayState state.RelayState
|
|
machineryServer *machinery.Server
|
|
actorCache *cache.Cache
|
|
)
|
|
|
|
func initConfig() {
|
|
viper.SetConfigName("config")
|
|
viper.AddConfigPath(".")
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
fmt.Println("Config file is not exists. Use environment variables.")
|
|
viper.BindEnv("actor_pem")
|
|
viper.BindEnv("redis_url")
|
|
viper.BindEnv("relay_bind")
|
|
viper.BindEnv("relay_domain")
|
|
viper.BindEnv("relay_servicename")
|
|
} else {
|
|
Actor.Summary = viper.GetString("relay_summary")
|
|
Actor.Icon = activitypub.Image{URL: viper.GetString("relay_icon")}
|
|
Actor.Image = activitypub.Image{URL: viper.GetString("relay_image")}
|
|
}
|
|
Actor.Name = viper.GetString("relay_servicename")
|
|
|
|
hostURL, _ = url.Parse("https://" + viper.GetString("relay_domain"))
|
|
hostPrivatekey, _ = keyloader.ReadPrivateKeyRSAfromPath(viper.GetString("actor_pem"))
|
|
redisOption, err := redis.ParseURL(viper.GetString("redis_url"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
redisClient := redis.NewClient(redisOption)
|
|
relayState = state.NewState(redisClient, true)
|
|
relayState.ListenNotify(nil)
|
|
machineryConfig := &config.Config{
|
|
Broker: viper.GetString("redis_url"),
|
|
DefaultQueue: "relay",
|
|
ResultBackend: viper.GetString("redis_url"),
|
|
ResultsExpireIn: 5,
|
|
}
|
|
machineryServer, err = machinery.NewServer(machineryConfig)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
Actor.GenerateSelfKey(hostURL, &hostPrivatekey.PublicKey)
|
|
actorCache = cache.New(5*time.Minute, 10*time.Minute)
|
|
WebfingerResource.GenerateFromActor(hostURL, &Actor)
|
|
|
|
fmt.Println("Welcome to YUKIMOCHI Activity-Relay [Server]", version)
|
|
fmt.Println(" - Configurations")
|
|
fmt.Println("RELAY DOMAIN : ", hostURL.Host)
|
|
fmt.Println("REDIS URL : ", viper.GetString("redis_url"))
|
|
fmt.Println("BIND ADDRESS : ", viper.GetString("relay_bind"))
|
|
fmt.Println(" - Blocked Domain")
|
|
domains, _ := redisClient.HKeys("relay:config:blockedDomain").Result()
|
|
for _, domain := range domains {
|
|
fmt.Println(domain)
|
|
}
|
|
fmt.Println(" - Limited Domain")
|
|
domains, _ = redisClient.HKeys("relay:config:limitedDomain").Result()
|
|
for _, domain := range domains {
|
|
fmt.Println(domain)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// Load Config
|
|
initConfig()
|
|
|
|
http.HandleFunc("/.well-known/webfinger", handleWebfinger)
|
|
http.HandleFunc("/actor", handleActor)
|
|
http.HandleFunc("/inbox", func(w http.ResponseWriter, r *http.Request) {
|
|
handleInbox(w, r, decodeActivity)
|
|
})
|
|
|
|
http.ListenAndServe(viper.GetString("relay_bind"), nil)
|
|
}
|