krz/skunky-art

Alternative privacy frontend for DeviantArt.

clone: git clone https://gitbay.org/krz/skunky-art.git

db53a8bd9004098e6a1507e5826a833fd9efa954

unsigned

author: lost+skunk <skunky@ebloid.ru> · 2024-09-03T12:36:19Z

Два API-эндпоинта
 TODO.md               |  5 ++++-
 app/api.go            | 61 +++++++++++++++++++++++++++++++++++++++++----------
 app/cli.go            | 11 +++++++---
 app/config.go         |  5 +++++
 app/router.go         | 56 +++++++++++++++++++++++-----------------------
 app/util.go           | 14 +++++++++++-
 app/wrapper.go        |  1 -
 go.mod                |  2 +-
 main.go               |  2 ++
 static/html/about.htm |  4 ++--
 10 files changed, 114 insertions(+), 47 deletions(-)

diff --git a/TODO.md b/TODO.md
index 0853df6..fbca5d8 100644
--- a/TODO.md
+++ b/TODO.md
@@ -1,11 +1,13 @@
 # v1.3.x
 * Почистить говнокод
+* Добавить фильтры поиска
 * ~~Сделать порт под FreeBSD~~ ✔️
 * **Доделать парсинг описания**
 * ~~Реализовать стрипы в ежедневных артах~~ ✔️
 * ~~Исправить баг с навигацией по страницам~~ ✔️
 * Сделать нормальное отображение ошибок
 * ~~Сделать единицы в конфиге более понятными~~ ✔️
+* Добавить чекер инстанса на работоспособность
 * ~~Добавить просмотр понравившихся артов пользователю~~ ✔️
 * Добавить возможность включить темплейты в бинарник [P]
 * ~~Реализовать миниатюры и оптимизировать CSS под маленькие экраны~~ ✔️
@@ -13,9 +15,10 @@
 * **Реализовать отображение контента, отличного от картинок (видео, аудио, etc)**
 * Исправить баг с эмоджи, когда некоторые кастомные эмоции могут не отображаться
 * ~~Добавить аргумент &filename, который будет выдавать файл с нормально выглядещем именем~~ ✔️
-* ~~Улучшить систему кеширования: добавить рейтинг для удаления и копирование изображений в ОЗУ~~ BUG: почему-то всё переодически встаёт раком
+* ~~Улучшить систему кеширования: добавить рейтинг для удаления и копирование изображений в ОЗУ~~ ✔️
 # v1.4
 * Реализовать API
 * Реализовать темы
 * Перейти на арены в кеше
 * Реализовать многоязычный интерфейс
+
diff --git a/app/api.go b/app/api.go
index ddc67c1..3d0871c 100644
--- a/app/api.go
+++ b/app/api.go
@@ -1,6 +1,7 @@
 package app
 
 import (
+	"encoding/json"
 	"math/rand"
 	"strings"
 
@@ -8,27 +9,65 @@ import (
 )
 
 type API struct {
-  skunkyartLink *skunkyart
+  main *skunkyart
+}
+
+type info struct {
+  Version string `json:"version"`
+  Settings settingsParams `json:"settings"`
+}
+
+func (a API) Info() {
+  json, err := json.Marshal(info{
+    Version: a.main.Version,
+    Settings: settingsParams{
+      Nsfw: CFG.Nsfw,
+      Proxy: CFG.Proxy,
+    },
+  })
+  try(err)
+  a.main.Writer.Write(json)
+  return
+}
+
+func (a API) Error(description string, status int) {
+  a.main.Writer.WriteHeader(status)
+  var response strings.Builder
+  response.WriteString(`{"error":"`)
+  response.WriteString(description)
+  response.WriteString(`"}`)
+  wr(a.main.Writer, response.String())
 }
 
 func (a API) sendMedia(d *devianter.Deviation) {
   mediaUrl, name := devianter.UrlFromMedia(d.Media)
-  
-  var filename strings.Builder
-	filename.WriteString(`filename="`)
-  filename.WriteString(name)
-	filename.WriteString(`"`)
-	a.skunkyartLink.Writer.Header().Add("Content-Disposition", filename.String())
+	a.main.SetFilename(name)
 	
   if len(mediaUrl) != 0 {
     mediaUrl = mediaUrl[21:]
 		dot := strings.Index(mediaUrl, ".")
-    a.skunkyartLink.DownloadAndSendMedia(mediaUrl[:dot], mediaUrl[dot+11:])
+		a.main.Writer.Header().Del("Content-Type")
+    a.main.DownloadAndSendMedia(mediaUrl[:dot], mediaUrl[dot+11:])
   }
 }
 
+// TODO: сделать фильтры
 func (a API) Random() {
-  s, err := devianter.PerformSearch(string(rand.Intn(999)), rand.Intn(30), 'a')
-  try(err) 
-  a.sendMedia(&s.Results[rand.Intn(len(s.Results))])
+  for attempt := 1;; {
+    if attempt > 3 {
+      a.Error("Sorry, butt NSFW on this are disabled, and the instance failed to find a random art without NSFW", 500)
+    }
+    
+    s, err := devianter.PerformSearch(string(rand.Intn(999)), rand.Intn(30), 'a')
+    try(err)
+    deviation := &s.Results[rand.Intn(len(s.Results))]
+    
+    if deviation.NSFW && !CFG.Nsfw {
+      attempt++
+      continue
+    }
+    
+    a.sendMedia(deviation)
+    return
+  }    
 }
diff --git a/app/cli.go b/app/cli.go
index 261e103..0480bdb 100644
--- a/app/cli.go
+++ b/app/cli.go
@@ -4,19 +4,20 @@ import (
 	"bufio"
 	"bytes"
 	"encoding/json"
+	"html/template"
 	"os"
 	"time"
 )
 
 func ExecuteCommandLineArguments() {
-	const helpmsg = `SkunkyArt v1.3.1 [CSS improvements for mobile and the strips on Daily Deviations]
+	var helpmsg = `SkunkyArt v{{.Version}} [{{.Description}}]
 Usage:
 	- [-c|--config] 	| path to config
 	- [-a|--add-instance]	| generates 'instances.json' and 'INSTANCES.md' files with ur instance
 	- [-h|--help]		| returns this message
 Example:
 	./skunkyart -c config.json
-Copyright lost+skunk, X11. https://git.macaw.me/skunky/skunkyart/src/tag/v1.3.1`
+Copyright lost+skunk, X11. https://git.macaw.me/skunky/skunkyart/src/tag/v{{.Version}}`
 
 	a := os.Args[1:]
 	for n, x := range a {
@@ -28,7 +29,11 @@ Copyright lost+skunk, X11. https://git.macaw.me/skunky/skunkyart/src/tag/v1.3.1`
 				exit("Not enought arguments", 1)
 			}
 		case "-h", "--help":
-			exit(helpmsg, 0)
+			var buf bytes.Buffer
+			t := template.New("help")
+			t.Parse(helpmsg)
+			t.Execute(&buf, &Release)
+			exit(buf.String(), 0)
 		case "-a", "--add-instance":
 			addInstance()
 		}
diff --git a/app/config.go b/app/config.go
index 7d159e7..eff9ee8 100644
--- a/app/config.go
+++ b/app/config.go
@@ -11,6 +11,11 @@ import (
 	"git.macaw.me/skunky/devianter"
 )
 
+var Release struct {
+	Version string
+	Description string
+}
+
 type cache_config struct {
 	Enabled        bool
 	Path           string
diff --git a/app/router.go b/app/router.go
index 2bcaf41..cc58f93 100644
--- a/app/router.go
+++ b/app/router.go
@@ -13,10 +13,9 @@ var Host, Path string
 
 func Router() {
 	parsepath := func(path string) map[int]string {
+		path = "/"
 		if l := len(CFG.URI); len(path) > l {
 			path = path[l-1:]
-		} else {
-			path = "/"
 		}
 
 		parsedpath := make(map[int]string)
@@ -54,44 +53,42 @@ func Router() {
 
 	// функция, что управляет всем
 	handle := func(w http.ResponseWriter, r *http.Request) {
+		Path = r.URL.Path
+		path := parsepath(Path)
+		Host = "http://" + r.Host
+		
 		if h := r.Header["X-Forwarded-Proto"]; len(h) != 0 && h[0] == "https" {
-			Host = h[0] + "://" + r.Host
-		} else {
-			Host = "http://" + r.Host
+			Host = "https://" + r.Host
 		}
 
-		Path = r.URL.Path
-		path := parsepath(Path)
-		// структура с функциями
-		var skunky skunkyart
+		var skunky = skunkyart{Version: Release.Version}
+
+		arg := skunky.Args.Get
+		p, _ := strconv.Atoi(arg("p"))
+		
+		skunky.Endpoint = path[1]
+		skunky.API.main = &skunky
 		skunky.Writer = w
 		skunky.Args = r.URL.Query()
 		skunky.BasePath = CFG.URI
-
-		arg := skunky.Args.Get
 		skunky.QueryRaw = arg("q")
 		skunky.Query = u.QueryEscape(skunky.QueryRaw)
+		skunky.Page = p
 
 		if t := arg("type"); len(t) > 0 {
 			skunky.Type = rune(t[0])
 		}
-		p, _ := strconv.Atoi(arg("p"))
-		skunky.Page = p
 
 		if arg("atom") == "true" {
 			skunky.Atom = true
 		}
 
-		skunky.Endpoint = path[1]
-		skunky.API.skunkyartLink = &skunky
-
-		// пути
 		switch skunky.Endpoint {
-		default:
-			skunky.ReturnHTTPError(404)
-
+		// main
 		case "":
 			skunky.ExecuteTemplate("index.htm", "html", &CFG.URI)
+		case "about":
+			skunky.About()
 		case "post":
 			skunky.Deviation(path[2], path[3])
 		case "search":
@@ -101,33 +98,38 @@ func Router() {
 		case "group_user":
 			skunky.GRUser()
 
+		// media
 		case "media":
 			switch path[2] {
 			case "file":
 				if a := arg("filename"); a != "" {
-					var filename strings.Builder
-					filename.WriteString(`filename="`)
-					filename.WriteString(a)
-					filename.WriteString(`"`)
-					w.Header().Add("Content-Disposition", filename.String())
+					skunky.SetFilename(a)
 				}
 				skunky.DownloadAndSendMedia(path[3], next(path, 4))
 			case "emojitar":
 				skunky.Emojitar(path[3])
 			}
-		case "about":
-			skunky.About()
 		case "stylesheet":
 			w.Header().Add("content-type", "text/css")
 			w.Write(open("css/skunky.css"))
 		case "favicon.ico":
 			w.Write(open("images/logo.png"))
 
+		// API
 		case "api":
+			w.Header().Add("Content-Type", "application/json")
 			switch path[2] {
+				case "instance":
+					skunky.API.Info()
 				case "random":
 					skunky.API.Random()
+				default:
+					skunky.API.Error("Not Found", 404)
 			}
+
+			// 404
+			default:
+				skunky.ReturnHTTPError(404)
 		}
 	}
 
diff --git a/app/util.go b/app/util.go
index 71e7d89..c992466 100644
--- a/app/util.go
+++ b/app/util.go
@@ -64,6 +64,7 @@ type skunkyart struct {
 	Query, QueryRaw    string
 
   API API
+  Version string
 
 	Templates struct {
 		About struct {
@@ -131,7 +132,7 @@ func UrlBuilder(strs ...string) string {
 	str.WriteString(CFG.URI)
 	for n, x := range strs {
 		str.WriteString(x)
-		if n+1 < l && !(strs[n+1][0] == '?' || strs[n+1][0] == '&') && !(x[0] == '?' || x[0] == '&') {
+		if n := n+1; n < l && len(strs[n]) != 0 && !(strs[n][0] == '?' || strs[n][0] == '&') && !(x[0] == '?' || x[0] == '&') {
 			str.WriteString("/")
 		}
 	}
@@ -153,6 +154,14 @@ func (s skunkyart) ReturnHTTPError(status int) {
 	wr(s.Writer, msg.String())
 }
 
+func (s skunkyart) SetFilename(name string) {
+	var filename strings.Builder
+	filename.WriteString(`filename="`)
+	filename.WriteString(name)
+	filename.WriteString(`"`)
+	s.Writer.Header().Add("Content-Disposition", filename.String())
+}
+
 type Downloaded struct {
 	Headers http.Header
 	Status  int
@@ -189,6 +198,9 @@ func ParseMedia(media devianter.Media, thumb ...int) string {
 	if len(mediaUrl) != 0 && CFG.Proxy {
 		mediaUrl = mediaUrl[21:]
 		dot := strings.Index(mediaUrl, ".")
+		if filename == "" {
+			filename = "image.gif"
+		}
 		return UrlBuilder("media", "file", mediaUrl[:dot], mediaUrl[dot+11:], "&filename=", filename)
 	}
 	return mediaUrl
diff --git a/app/wrapper.go b/app/wrapper.go
index f78d8c8..a3d251b 100644
--- a/app/wrapper.go
+++ b/app/wrapper.go
@@ -42,7 +42,6 @@ func (s skunkyart) GRUser() {
 					group.About.A = x.ModuleData.About
 					var about = &group.About.A
 					group.CreationDate = time.Unix(time.Now().Unix()-x.ModuleData.About.RegDate, 0).UTC().String()
-
 					group.About.DescriptionFormatted = ParseDescription(about.Description)
 
 					for _, val := range x.ModuleData.About.SocialLinks {
diff --git a/go.mod b/go.mod
index c7dc4e6..a3cbd52 100644
--- a/go.mod
+++ b/go.mod
@@ -2,7 +2,7 @@ module skunkyart
 
 go 1.18
 
-// replace git.macaw.me/skunky/devianter v0.2.5 => /home/skunk/projects/devianter
+replace git.macaw.me/skunky/devianter v0.2.5 => /home/skunk/projects/devianter
 require (
 	git.macaw.me/skunky/devianter v0.2.5
 	golang.org/x/net v0.27.0
diff --git a/main.go b/main.go
index 190d79b..3dc4374 100644
--- a/main.go
+++ b/main.go
@@ -9,6 +9,8 @@ import (
 )
 
 func main() {
+	app.Release.Version = "1.3.2-alpha"
+	app.Release.Description = "Two API endpoints and template embedding into binary"
 	go app.RefreshInstances()
 
 	app.ExecuteCommandLineArguments()
diff --git a/static/html/about.htm b/static/html/about.htm
index d2278d6..94bbf5c 100644
--- a/static/html/about.htm
+++ b/static/html/about.htm
@@ -44,6 +44,6 @@
                 {{end}}
             </ul>
         </details>
-        <p>Copyright <a href="https://go.kde.org/matrix/#/@softpigeones:ebloid.ru" target="_blank">lost+skunk</a>, X11. <a href="https://git.macaw.me/skunky/skunkyart/src/tag/v1.3.1" target="_blank">SkunkyArt v1.3.1</a></p>
+        <p>Copyright <a href="https://go.kde.org/matrix/#/@softpigeones:ebloid.ru" target="_blank">lost+skunk</a>, X11. <a href="https://git.macaw.me/skunky/skunkyart/src/tag/v{{.Version}}" target="_blank">SkunkyArt v{{.Version}}</a></p>
     </main>
-</html>
\ No newline at end of file
+</html>