Обработка и постинг картинок

main.go

@@ -3,17 +3,22 @@ package main
 import (
 	"context"
 	"fmt"
+	"io"
 	"log"
+	"net/http"
 	"net/url"
 	"os"
 	"path/filepath"
 	"regexp"
+	"strings"
 	"time"
 
 	"github.com/go-yaml/yaml"
 	"github.com/mattn/go-mastodon"
 	"github.com/mmcdole/gofeed"
 	"github.com/urfave/cli/v3"
+	"golang.org/x/net/html"
+	"golang.org/x/net/html/atom"
 )
 
 type MastodonClientData struct {
@@ -48,6 +53,25 @@ func getDataFromConfig(path string) *mastodon.Config {
 	return config
 }
 
+func returnImgArray(htmlText string) []string {
+	var imgArray []string
+	parser, err := html.Parse(strings.NewReader(htmlText))
+	if err != nil {
+		log.Println(err)
+	}
+	for n := range parser.Descendants() {
+		if n.Type == html.ElementNode && n.DataAtom == atom.Img {
+			for _, img := range n.Attr {
+				if img.Key == "src" {
+					imgArray = append(imgArray, img.Val)
+				}
+			}
+		}
+	}
+
+	return imgArray
+}
+
 func hasHTMLTags(s string) bool {
 	re := regexp.MustCompile(`<(?i)[a-z][a-z0-9]*[^>]*>`)
 	return re.MatchString(s)
@@ -66,7 +90,7 @@ func getKikiConfig(path string) KikiSettings {
 	return kikiSettings
 }
 
-func ClientConfiguration(Instance string) {
+func clientConfiguration(Instance string) {
 	appConfig := &mastodon.AppConfig{
 		Server:       Instance,
 		ClientName:   "Kiki",
@@ -111,11 +135,13 @@ func ClientConfiguration(Instance string) {
 	if err != nil {
 		log.Println(err)
 	}
-	secretConfig, err := os.OpenFile("secret.conf", os.O_CREATE, 0o644)
+	log.Println(string(marshaledYaml))
+	secretConfig, err := os.OpenFile("secret.conf", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0o644)
 	if err != nil {
 		log.Println(err)
 	}
-	secretConfig.WriteString(string(marshaledYaml))
+	secretConfig.Write(marshaledYaml)
+	defer secretConfig.Close()
 }
 
 func newsText(url string) []*gofeed.Item {
@@ -144,6 +170,59 @@ func createPost(statusText string) {
 	}
 }
 
+func picBytesArray(picturesArray []string) [][]byte {
+	var picturesBytes [][]byte
+	for _, picture := range picturesArray {
+		resp, err := http.Get(picture)
+		if err != nil {
+			log.Println(err)
+			return picturesBytes
+		}
+		defer resp.Body.Close()
+		picBytes, err := io.ReadAll(resp.Body)
+		if err != nil {
+			log.Println(err)
+			return picturesBytes
+		}
+		picturesBytes = append(picturesBytes, picBytes)
+	}
+	return picturesBytes
+}
+
+func uploadPictures(filesBytes [][]byte) []*mastodon.Attachment {
+	config := getDataFromConfig("secret.conf")
+	var attachments []*mastodon.Attachment
+	mastoClient := mastodon.NewClient(config)
+	for _, file := range filesBytes {
+		att, err := mastoClient.UploadMediaFromBytes(context.Background(), file)
+		if err != nil {
+			log.Println(err)
+			return attachments
+		}
+		attachments = append(attachments, att)
+	}
+	return attachments
+}
+
+func postWithPictures(attachments []*mastodon.Attachment) {
+	var mediaID []mastodon.ID
+	config := getDataFromConfig("secret.conf")
+	mastoClient := mastodon.NewClient(config)
+	for _, attach := range attachments {
+		mediaID = append(mediaID, attach.ID)
+	}
+	toot := mastodon.Toot{
+		MediaIDs:   mediaID,
+		Visibility: "unlisted",
+		Sensitive:  true,
+	}
+	_, err := mastoClient.PostStatus(context.Background(), &toot)
+	if err != nil {
+		log.Println(err)
+	}
+
+}
+
 func main() {
 	cmd := &cli.Command{
 		Name:  "kiki",
@@ -163,7 +242,7 @@ func main() {
 						log.Println(err)
 					}
 					instanceUrlParser.Scheme = "https"
-					ClientConfiguration(instanceUrlParser.String())
+					clientConfiguration(instanceUrlParser.String())
 					return nil
 				},
 			},
@@ -183,19 +262,16 @@ func main() {
 						news := newsText(kikiConfig.RSSUri)
 						if news[0].GUID != lastGUID {
 							if !hasHTMLTags(news[0].Description) {
+								log.Println(news[0].Description)
 								createPost(news[0].Description)
 								lastGUID = news[0].GUID
 								log.Println("Пост отправлен")
+							} else {
+								attachments := uploadPictures(picBytesArray(returnImgArray(news[0].Description)))
+								postWithPictures(attachments)
+								lastGUID = news[0].GUID
 							}
 						}
-						/*for _, item := range newsText(kikiConfig.RSSUri) {
-							if item.GUID == lastGUID {
-								break
-							}
-							createPost(item.Description)
-							lastGUID = item.GUID
-							log.Println("Пост отправлен")
-						}*/
 					}
 					return nil
 				},