From 33541bdd48dfdba6039b4e563f80de5699a1955d Mon Sep 17 00:00:00 2001
From: Romain de Laage <romain.delaage@rdelaage.ovh>
Date: Fri, 19 Feb 2021 17:09:09 +0100
Subject: [PATCH] Now have already posted memory

---
 bot.go | 76 ++++++++++++++++++++++++++++++++++++++++++----------------
 1 file changed, 55 insertions(+), 21 deletions(-)

diff --git a/bot.go b/bot.go
index bb87faa..b37a700 100644
--- a/bot.go
+++ b/bot.go
@@ -20,6 +20,7 @@ package main
 
 import (
 	"io/ioutil"
+	"bufio"
 	"bytes"
 	"log"
 	"net/http"
@@ -65,9 +66,10 @@ func main() {
 	api_token := os.Getenv("MM_ASSOS_TOKEN")
 	api_endpointmm := os.Getenv("MM_ASSOS_ENDPOINTMM")
 	api_endpointassos := os.Getenv("MM_ASSOS_ENDPOINTASSOS")
+	db_path := os.Getenv("MM_ASSOS_DBPATH")
 
-	if channel_id == "" || api_token == "" || api_endpointmm == "" || api_endpointassos == "" {
-		log.Fatalln("MM_ASSOS_CHANNEL, MM_ASSOS_ENDPOINTMM, MM_ASSOS_ENDPOINTASSOS or MM_ASSOS_TOKEN not defined")
+	if channel_id == "" || api_token == "" || api_endpointmm == "" || api_endpointassos == "" || db_path == "" {
+		log.Fatalln("MM_ASSOS_CHANNEL, MM_ASSOS_ENDPOINTMM, MM_ASSOS_ENDPOINTASSOS, MM_ASSOS_DBPATH or MM_ASSOS_TOKEN not defined")
 		os.Exit(1)
 	}
 
@@ -92,34 +94,66 @@ func main() {
 	json.Unmarshal(body, &articles)
 
 	for i := 0; i < len(articles); i++ {
-		var message Message
-		message.Channel_id = channel_id
-		message.Message_content = "# " + articles[i].Title + "\n" + "par *" + articles[i].Owned_by.Name + "*\n\n" + articles[i].Content
-
-		json_content, err := json.Marshal(message)
+		db_file, err := os.OpenFile(db_path, os.O_CREATE|os.O_RDWR, 0777)
 		if err != nil {
 			log.Fatalln(err)
 			os.Exit(1)
 		}
 
-		req, err := http.NewRequest("POST", api_endpointmm, bytes.NewBuffer(json_content))
-		if err != nil {
-			log.Fatalln(err)
-			os.Exit(1)
+		scanner := bufio.NewScanner(db_file)
+		scanner.Split(bufio.ScanLines)
+		var entries []string
+
+		for scanner.Scan() {
+			entries = append(entries, scanner.Text())
 		}
 
-		req.Header.Set("Content-Type", "application/json")
-		req.Header.Set("Authorization", "Bearer " + api_token)
+		in_db := 0
 
-		client := &http.Client{}
-		resp, err := client.Do(req)
-		if err != nil {
-			log.Fatalln(err)
-			os.Exit(1)
+		for _, entry := range entries {
+			if articles[i].Id == entry {
+				in_db = 1
+			}
 		}
-		if resp.StatusCode != 201 {
-			log.Fatalln("Error while posting message, server returned " + resp.Status)
-			os.Exit(1)
+
+		if in_db == 0 {
+			var message Message
+			message.Channel_id = channel_id
+			message.Message_content = "# " + articles[i].Title + "\n" + "par *" + articles[i].Owned_by.Name + "*\n\n" + articles[i].Content
+
+			json_content, err := json.Marshal(message)
+			if err != nil {
+				log.Fatalln(err)
+				os.Exit(1)
+			}
+
+			req, err := http.NewRequest("POST", api_endpointmm, bytes.NewBuffer(json_content))
+			if err != nil {
+				log.Fatalln(err)
+				os.Exit(1)
+			}
+
+			req.Header.Set("Content-Type", "application/json")
+			req.Header.Set("Authorization", "Bearer " + api_token)
+
+			client := &http.Client{}
+			resp, err := client.Do(req)
+			if err != nil {
+				log.Fatalln(err)
+				os.Exit(1)
+			}
+			if resp.StatusCode != 201 {
+				log.Fatalln("Error while posting message, server returned " + resp.Status)
+				os.Exit(1)
+			}
+
+			_, err = db_file.WriteString(articles[i].Id + "\n")
+			if err != nil {
+				log.Fatalln(err)
+				os.Exit(1)
+			}
 		}
+
+		db_file.Close()
 	}
 }