132 lines
2.8 KiB
Go
132 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/RichardKnop/machinery/v1/tasks"
|
|
"github.com/urfave/cli"
|
|
"github.com/yukimochi/Activity-Relay/ActivityPub"
|
|
)
|
|
|
|
func pushRegistorJob(inboxURL string, body []byte) {
|
|
job := &tasks.Signature{
|
|
Name: "registor",
|
|
RetryCount: 25,
|
|
Args: []tasks.Arg{
|
|
{
|
|
Name: "inboxURL",
|
|
Type: "string",
|
|
Value: inboxURL,
|
|
},
|
|
{
|
|
Name: "body",
|
|
Type: "string",
|
|
Value: string(body),
|
|
},
|
|
},
|
|
}
|
|
_, err := macServer.SendTask(job)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
|
|
func listFollows(c *cli.Context) error {
|
|
var err error
|
|
var domains []string
|
|
|
|
fmt.Println(" - Follow request :")
|
|
follows, err := redClient.Keys("relay:pending:*").Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
for _, follow := range follows {
|
|
domains = append(domains, strings.Replace(follow, "relay:pending:", "", 1))
|
|
}
|
|
for _, domain := range domains {
|
|
fmt.Println(domain)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func acceptFollow(c *cli.Context) error {
|
|
domain := c.String("domain")
|
|
if domain != "" {
|
|
num, err := redClient.Exists("relay:pending:" + domain).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if num == 0 {
|
|
fmt.Println("Given domain not found.")
|
|
return nil
|
|
}
|
|
|
|
fmt.Println("Accept Follow request : " + domain)
|
|
data, err := redClient.HGetAll("relay:pending:" + domain).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
activity := activitypub.Activity{
|
|
[]string{"https://www.w3.org/ns/activitystreams", "https://w3id.org/security/v1"},
|
|
data["activity_id"],
|
|
data["actor"],
|
|
data["type"],
|
|
data["object"],
|
|
nil,
|
|
nil,
|
|
}
|
|
|
|
resp := activity.GenerateResponse(hostname, "Accept")
|
|
jsonData, _ := json.Marshal(&resp)
|
|
|
|
pushRegistorJob(data["inbox_url"], jsonData)
|
|
redClient.HSet("relay:subscription:"+domain, "inbox_url", data["inbox_url"])
|
|
redClient.Del("relay:pending:" + domain)
|
|
return nil
|
|
} else {
|
|
fmt.Println("No domain given.")
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func rejectFollow(c *cli.Context) error {
|
|
domain := c.String("domain")
|
|
if domain != "" {
|
|
num, err := redClient.Exists("relay:pending:" + domain).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if num == 0 {
|
|
fmt.Println("Given domain not found.")
|
|
return nil
|
|
}
|
|
|
|
fmt.Println("Reject Follow request : " + domain)
|
|
data, err := redClient.HGetAll("relay:pending:" + domain).Result()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
activity := activitypub.Activity{
|
|
[]string{"https://www.w3.org/ns/activitystreams", "https://w3id.org/security/v1"},
|
|
data["activity_id"],
|
|
data["actor"],
|
|
data["type"],
|
|
data["object"],
|
|
nil,
|
|
nil,
|
|
}
|
|
|
|
resp := activity.GenerateResponse(hostname, "Reject")
|
|
jsonData, _ := json.Marshal(&resp)
|
|
|
|
pushRegistorJob(data["inbox_url"], jsonData)
|
|
redClient.Del("relay:pending:" + domain)
|
|
return nil
|
|
} else {
|
|
fmt.Println("No domain given.")
|
|
return nil
|
|
}
|
|
}
|