Add AllowIFrames helper

microcosm-cc · Dec 30, 2021 · 5638f90 · 5638f90
1 parent 6e7247e
commit 5638f90
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/helpers.go b/helpers.go
@@ -295,3 +295,9 @@ func (p *Policy) AllowTables() {
 		CellVerticalAlign,
 	).OnElements("tbody", "tfoot")
 }
+
+func (p *Policy) AllowIFrames(vals ...SandboxValue) {
+	p.AllowAttrs("sandbox").OnElements("iframe")
+
+	p.RequireSandboxOnIFrame(vals...)
+}
diff --git a/policy.go b/policy.go
@@ -702,6 +702,8 @@ func (p *Policy) AllowURLSchemeWithCustomPolicy(
 	return p
 }
 
+// RequireSandboxOnIFrame will result in all iframe tags having a sandbox="" tag
+// Any sandbox values not specified here will be filtered from the generated HTML
 func (p *Policy) RequireSandboxOnIFrame(vals ...SandboxValue) {
 	p.requireSandboxOnIFrame = make(map[string]bool)