100 lines
2.2 KiB
Go
100 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"h4kor/owl-blogs"
|
|
"sync"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var postId string
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(webmentionCmd)
|
|
webmentionCmd.Flags().StringVar(
|
|
&postId, "post", "",
|
|
"specify the post to send webmentions for. Otherwise, all posts will be checked.",
|
|
)
|
|
}
|
|
|
|
var webmentionCmd = &cobra.Command{
|
|
Use: "webmention",
|
|
Short: "Send webmentions for posts, optionally for a specific user",
|
|
Long: `Send webmentions for posts, optionally for a specific user`,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
repo, err := owl.OpenRepository(repoPath)
|
|
if err != nil {
|
|
println("Error opening repository: ", err.Error())
|
|
return
|
|
}
|
|
|
|
var users []owl.User
|
|
if user == "" {
|
|
// send webmentions for all users
|
|
users, err = repo.Users()
|
|
if err != nil {
|
|
println("Error getting users: ", err.Error())
|
|
return
|
|
}
|
|
} else {
|
|
// send webmentions for a specific user
|
|
user, err := repo.GetUser(user)
|
|
users = append(users, user)
|
|
if err != nil {
|
|
println("Error getting user: ", err.Error())
|
|
return
|
|
}
|
|
}
|
|
|
|
processPost := func(user owl.User, post *owl.Post) error {
|
|
println("Webmentions for post: ", post.Title())
|
|
|
|
err := post.ScanForLinks()
|
|
if err != nil {
|
|
println("Error scanning post for links: ", err.Error())
|
|
return err
|
|
}
|
|
|
|
webmentions := post.OutgoingWebmentions()
|
|
println("Found ", len(webmentions), " links")
|
|
wg := sync.WaitGroup{}
|
|
wg.Add(len(webmentions))
|
|
for _, webmention := range webmentions {
|
|
go func(webmention owl.WebmentionOut) {
|
|
defer wg.Done()
|
|
err = post.SendWebmention(webmention)
|
|
if err != nil {
|
|
println("Error sending webmentions: ", err.Error())
|
|
} else {
|
|
println("Webmention sent to ", webmention.Target)
|
|
}
|
|
}(webmention)
|
|
}
|
|
wg.Wait()
|
|
return nil
|
|
}
|
|
|
|
for _, user := range users {
|
|
if postId != "" {
|
|
// send webmentions for a specific post
|
|
post, err := user.GetPost(postId)
|
|
if err != nil {
|
|
println("Error getting post: ", err.Error())
|
|
return
|
|
}
|
|
processPost(user, &post)
|
|
return
|
|
}
|
|
|
|
posts, err := user.Posts()
|
|
if err != nil {
|
|
println("Error getting posts: ", err.Error())
|
|
}
|
|
|
|
for _, post := range posts {
|
|
processPost(user, post)
|
|
}
|
|
}
|
|
},
|
|
}
|