From b8ce1ab774aedc6a7a0292ded965a24729606fe1 Mon Sep 17 00:00:00 2001
From: Niko Abeler <niko@rerere.org>
Date: Sun, 21 Aug 2022 11:37:17 +0200
Subject: [PATCH] -unsafe flag for owl-web

---
 cmd/owl-web/main.go | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/cmd/owl-web/main.go b/cmd/owl-web/main.go
index a14970c..ffdd815 100644
--- a/cmd/owl-web/main.go
+++ b/cmd/owl-web/main.go
@@ -37,19 +37,36 @@ func main() {
 	println("-repo <repo> - Specify the repository to use. Defaults to '.'")
 	println("-port <port> - Specify the port to use, Default is '8080'")
 	println("-user <name> - Start server in single user mode.")
+	println("-unsafe - Allow unsafe html.")
 	var repoName string
 	var port int
 	var singleUserName string
-	for i, arg := range os.Args[0 : len(os.Args)-1] {
+	var allowRawHTML bool = false
+	for i, arg := range os.Args[0:len(os.Args)] {
 		if arg == "-port" {
+			if i+1 >= len(os.Args) {
+				println("-port requires a port number")
+				os.Exit(1)
+			}
 			port, _ = strconv.Atoi(os.Args[i+1])
 		}
 		if arg == "-repo" {
+			if i+1 >= len(os.Args) {
+				println("-repo requires a repopath")
+				os.Exit(1)
+			}
 			repoName = os.Args[i+1]
 		}
 		if arg == "-user" {
+			if i+1 >= len(os.Args) {
+				println("-user requires a username")
+				os.Exit(1)
+			}
 			singleUserName = os.Args[i+1]
 		}
+		if arg == "-unsafe" {
+			allowRawHTML = true
+		}
 	}
 	if repoName == "" {
 		repoName = "."
@@ -70,6 +87,7 @@ func main() {
 		println("Repository:", repoName)
 		repo, err = owl.OpenRepository(repoName)
 	}
+	repo.SetAllowRawHtml(allowRawHTML)
 
 	if err != nil {
 		println("Error opening repository: ", err.Error())