feat: Add json presentation support; Add specifying format support

src/middlewares/presentation.ts

@@ -1,6 +1,6 @@
 import { createMiddleware } from "hono/factory";
 
-export type PresentationType = "terminal" | "browser" | "cli-browser";
+export type PresentationType = "terminal" | "browser" | "cli-browser" | "json";
 
 const TERMINAL_USER_AGENT =
 	/^(curl|wget|python-urllib|pycurl|java|go-http-client|php)/i;
@@ -12,16 +12,53 @@ const presentation = createMiddleware<{
 		presentation: PresentationType;
 	};
 }>(async (context, next) => {
-	const userAgent = context.req.header("User-Agent") || "";
+	let presentation: PresentationType = "browser";
+	const suggestedFormat =
+		context.req.query("format") || context.req.query("f");
 
-	if (TERMINAL_USER_AGENT.test(userAgent)) {
+	switch (suggestedFormat) {
-		context.set("presentation", "terminal");
+		case "terminal":
-	} else if (CLI_BROWSER_USER_AGENT.test(userAgent)) {
+		case "cli":
-		context.set("presentation", "cli-browser");
+		case "script":
-	} else {
+		case "shell":
-		context.set("presentation", "browser");
+		case "sh":
+		case "bash":
+			presentation = "terminal";
+			break;
+
+		case "browser":
+		case "default":
+		case "html":
+		case "web":
+			presentation = "browser";
+			break;
+
+		case "cli-browser":
+		case "lynx":
+		case "elinks":
+		case "w3m":
+			presentation = "cli-browser";
+			break;
+
+		case "json":
+			presentation = "json";
+			break;
+
+		default:
+			// Detect using User-Agent
+			const userAgent = context.req.header("User-Agent") || "";
+
+			if (TERMINAL_USER_AGENT.test(userAgent)) {
+				presentation = "terminal";
+			} else if (CLI_BROWSER_USER_AGENT.test(userAgent)) {
+				presentation = "cli-browser";
+			} else {
+				presentation = "browser";
+			}
 	}
 
+	context.set("presentation", presentation);
+
 	await next();
 });
 

src/middlewares/real-ip.ts

@@ -8,6 +8,7 @@ const realIP = createMiddleware<{
 }>(async (context, next) => {
 	const ip =
 		context.req.header("x-forwarded-for") ||
+		context.req.header("x-real-ip") ||
 		getConnInfo(context).remote.address;
 
 	if (!ip) {

src/utils/renderer.ts

@@ -12,6 +12,7 @@ export default async function render(
 		browser: ".html",
 		terminal: ".txt",
 		"cli-browser": ".cli.html",
+		json: ".jsonn",
 	}[presentation];
 	const key = templateName + ".njk" + extension;
 
@@ -24,5 +25,7 @@ export default async function render(
 			return context.html(content);
 		case "terminal":
 			return context.text(content.trimEnd());
+		case "json":
+			return context.json(JSON.parse(content));
 	}
 }

templates/index.njk.jsonn

@@ -0,0 +1 @@
+{ "ipv4": "{{ip}}" }

templates/port.njk.cli.html

@@ -1,6 +1,4 @@
 {% extends "base.njk.cli.html" %} {% block main %}
-<h2>Am I Open</h2>
-
 <i>{{ ip }}</i>
 
 {% if isOpen %}

templates/port.njk.html

@@ -72,8 +72,6 @@
 	}
 </style>
 {% endblock %} {% block main %}
-<h2>Am I Open</h2>
-
 <i>{{ ip }}</i>
 
 {% if isOpen %}

templates/port.njk.jsonn

@@ -0,0 +1 @@
+{"isOpen":{% if isOpen %}true{% else %}false{% endif %}}