Add AllowIFrames helper

kiwiz · kiwiz · commit 5638f900816a · 2021-12-30T17:15:25.000-05:00
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)
 

Original file line number	Diff line number	Diff line change
`@@ -702,6 +702,8 @@ func (p *Policy) AllowURLSchemeWithCustomPolicy(`
`702`	`702`	`return p`
`703`	`703`	`}`
`704`	`704`
	`705`	`+// RequireSandboxOnIFrame will result in all iframe tags having a sandbox="" tag`
	`706`	`+// Any sandbox values not specified here will be filtered from the generated HTML`
`705`	`707`	`func (p *Policy) RequireSandboxOnIFrame(vals ...SandboxValue) {`
`706`	`708`	`p.requireSandboxOnIFrame = make(map[string]bool)`
`707`	`709`