Browse Source

Add a rewrite rule for castopod episodes

Romain de Laage 4 years ago
parent
commit
808635e314

+ 6 - 0
reader/rewrite/rewrite_functions.go

@@ -282,3 +282,9 @@ func removeCustom(entryContent string, selector string) string {
 	output, _ := doc.Find("body").First().Html()
 	return output
 }
+
+func addCastopodEpisode(entryURL, entryContent string) string {
+	player := `<iframe width="650" frameborder="0" src="` + entryURL + `/embed/light"></iframe>`
+
+	return player + `<br>` + entryContent
+}

+ 2 - 0
reader/rewrite/rewriter.go

@@ -100,6 +100,8 @@ func applyRule(entryURL, entryContent string, rule rule) string {
 		} else {
 			logger.Debug("[Rewrite] Cannot find selector for remove rule %s", rule)
 		}
+	case "add_castopod_episode":
+		entryContent = addCastopodEpisode(entryURL, entryContent)
 	}
 
 	return entryContent

+ 9 - 0
reader/rewrite/rewriter_test.go

@@ -286,3 +286,12 @@ func TestRewriteRemoveCustom(t *testing.T) {
 		t.Errorf(`Not expected output: %s`, output)
 	}
 }
+
+func TestRewriteAddCastopodEpisode(t *testing.T) {
+	output := Rewriter("https://podcast.demo/@demo/episodes/test", "Episode Description", `add_castopod_episode`)
+	expected := `<iframe width="650" frameborder="0" src="https://podcast.demo/@demo/episodes/test/embed/light"></iframe><br>Episode Description`
+
+	if expected != output {
+		t.Errorf(`Not expected output: got "%s" instead of "%s"`, output, expected)
+	}
+}