3 files changed, 62 insertions, 0 deletions
diff --git a/app/config.go b/app/config.go
index 2faa69d..b790f71 100755
--- a/app/config.go
+++ b/app/config.go
@@ -34,6 +34,7 @@ type config struct {
 	Cache         cacheConfig `json:"cache"`
 	Proxy         bool        `json:"proxy"`
 	Nsfw          bool        `json:"nsfw"`
+	HideAI        bool        `json:"hide-ai"`
 	UserAgent     string      `json:"user-agent"`
 	DownloadProxy string      `json:"download-proxy"`
 	StaticPath    string      `json:"static-path"`
diff --git a/app/parsers.go b/app/parsers.go
index 68fdbe4..2027503 100755
--- a/app/parsers.go
+++ b/app/parsers.go
@@ -92,6 +92,9 @@ func (s skunkyart) DeviationList(devs []devianter.Deviation, allowAtom bool, con
 
 	for i, l := 0, len(devs); i < l; i++ {
 		data := &devs[i]
+		if data.AI && CFG.HideAI {
+			continue
+		}
 		if preview, fullview := ParseMedia(s.Host, data.Media, 320), ParseMedia(s.Host, data.Media); !data.NSFW || CFG.Nsfw {
 			if allowAtom && s.Atom {
 				s.Writer.Header().Add("Content-Type", "application/atom+xml")
diff --git a/app/parsers_test.go b/app/parsers_test.go
new file mode 100644
index 0000000..3d42bb4
--- /dev/null
+++ b/app/parsers_test.go
@@ -0,0 +1,58 @@
+package app
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/krazywarez/devianter"
+)
+
+// aiListDevs returns one human-made and one AI-flagged deviation, each with a
+// distinct title so the rendered output can be searched for either one.
+func aiListDevs() []devianter.Deviation {
+	human := devianter.Deviation{Title: "HumanArt"}
+	human.Author.Username = "alice"
+
+	robot := devianter.Deviation{Title: "RobotArt", AI: true}
+	robot.Author.Username = "bob"
+
+	return []devianter.Deviation{human, robot}
+}
+
+// TestDeviationListHidesAIWhenConfigured is the regression test for the hide-ai
+// option: with it on, deviations flagged data.AI must not appear in a listing at
+// all, while human-made ones are untouched.
+func TestDeviationListHidesAIWhenConfigured(t *testing.T) {
+	nsfw, hide := CFG.Nsfw, CFG.HideAI
+	CFG.Nsfw, CFG.HideAI = true, true
+	defer func() { CFG.Nsfw, CFG.HideAI = nsfw, hide }()
+
+	out := skunkyart{Host: "http://localhost"}.DeviationList(aiListDevs(), false)
+
+	if !strings.Contains(out, "HumanArt") {
+		t.Error("human deviation was dropped, want it kept")
+	}
+	if strings.Contains(out, "RobotArt") {
+		t.Error("AI deviation rendered while hide-ai is on, want it omitted")
+	}
+	if strings.Contains(out, "🤖") {
+		t.Error("AI marker present while hide-ai is on")
+	}
+}
+
+// TestDeviationListShowsAIByDefault pins down that the filter is opt-in: with
+// hide-ai off the AI deviation is still listed and still carries its marker.
+func TestDeviationListShowsAIByDefault(t *testing.T) {
+	nsfw, hide := CFG.Nsfw, CFG.HideAI
+	CFG.Nsfw, CFG.HideAI = true, false
+	defer func() { CFG.Nsfw, CFG.HideAI = nsfw, hide }()
+
+	out := skunkyart{Host: "http://localhost"}.DeviationList(aiListDevs(), false)
+
+	if !strings.Contains(out, "RobotArt") {
+		t.Error("AI deviation dropped while hide-ai is off, want it kept")
+	}
+	if !strings.Contains(out, "🤖") {
+		t.Error("AI marker missing while hide-ai is off")
+	}
+}