From 9543e4a8eb45e3dcdf4ab5452f65817d85422069 Mon Sep 17 00:00:00 2001
From: "Casper V. Kristensen" <casper@vkristensen.dk>
Date: Mon, 5 Aug 2024 20:25:39 +0200
Subject: [PATCH] gui

---
 README.md          | 9 +++++++--
 autosurfer/main.py | 5 ++++-
 flake.nix          | 2 +-
 3 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 292f11d..639f3f0 100644
--- a/README.md
+++ b/README.md
@@ -8,11 +8,16 @@ internet surveillance](https://ulovliglogning.dk/). Privacy through obscurity?
 podman run --rm -it quay.io/caspervk/autosurfer:latest
 ```
 
+To show the Firefox GUI:
+```shell
+podman run --rm -it --network host --env DISPLAY --security-opt label=type:container_runtime_t quay.io/caspervk/autosurfer:latest
+```
+
 
 # Building
 ```shell
 nix build .#oci
 ./result | podman load
-podman run --rm -it autosurfer:0.0.1
-# podman push autosurfer:0.0.1 quay.io/caspervk/autosurfer:0.0.1
+podman run --rm -it autosurfer:dev
+# podman push autosurfer:dev quay.io/caspervk/autosurfer:latest
 ```
diff --git a/autosurfer/main.py b/autosurfer/main.py
index db88d96..a705712 100644
--- a/autosurfer/main.py
+++ b/autosurfer/main.py
@@ -1,5 +1,6 @@
 import asyncio
 import json
+import os
 
 import websockets
 from selenium import webdriver
@@ -15,7 +16,9 @@ service = webdriver.FirefoxService(
 options = webdriver.FirefoxOptions()
 # Selenium only checks /usr/bin/firefox by default
 options.binary_location = "/bin/firefox"
-options.add_argument("--headless")
+
+if not os.getenv("DISPLAY"):
+    options.add_argument("--headless")
 
 driver = webdriver.Firefox(service=service, options=options)
 driver.set_page_load_timeout(3)
diff --git a/flake.nix b/flake.nix
index e01f3e3..5cdbbc2 100644
--- a/flake.nix
+++ b/flake.nix
@@ -62,7 +62,7 @@
       # https://github.com/NixOS/nixpkgs/blob/master/pkgs/build-support/docker/examples.nix
       oci = pkgs.dockerTools.streamLayeredImage {
         name = "autosurfer";
-        tag = "0.0.1";
+        tag = "dev";
         created = builtins.substring 0 8 self.lastModifiedDate;
         contents = [
           env