#!/bin/bash

DOMAIN=""
MANUAL_PORT=5090
JELLYFIN_PORT=5091
OLLAMA_PORT=5092

read -p "Has the command 'cloudflared tunnel login' been executed? (y/N): " LOGGED_IN
LOGGED_IN=$(echo "$LOGGED_IN" | tr '[:upper:]' '[:lower:]')

if [[ -z "$LOGGED_IN" || "$LOGGED_IN" == "n" ]]; then
	echo "You can still use Cloudflare for free, by generating random URLs."
	echo "Enter the following into a terminal:";
	echo "    cloudflared tunnel --url=http://localhost:PORT";
	echo "to get a URL a process, for example. Replace PORT with one of the following:";
	echo "    $MANUAL_PORT (The manual mirror's port.)";
	echo "    $JELLYFIN_PORT (The Jellyfin port.)";
	echo "    $OLLAMA_PORT (The Ollama & Open Web-UI port.)";
	echo "For more information, visit raspi.vintagecoding.net#cloudflared";
	exit;
fi

read -p "What is your domain name? e.g., vintagecoding.net: " DOMAIN;

echo "export TUNNEL_ORIGIN_CERT=$HOME/.cloudflared/cert.pem" >> $HOME/.bashrc;
source $HOME/.bashrc;

cloudflared tunnel create raspi;
TUNNEL_ID=$(cloudflared tunnel list | awk '/raspi/ {print $1}');

cd;
echo "tunnel: $TUNNEL_ID
credentials-file: $(pwd)/.cloudflared/$TUNNEL_ID.json
ingress:
  - hostname: raspi.$DOMAIN
    service: http://localhost:$MANUAL_PORT
  - hostname: media.$DOMAIN
    service: http://localhost:$JELLYFIN_PORT
  - hostname: ai.$DOMAIN
    service: http://localhost:$OLLAMA_PORT
  - service: http_status:404" | tee $HOME/.cloudflared/config.yaml;

cloudflared tunnel route dns $TUNNEL_ID raspi.$DOMAIN;
cloudflared tunnel route dns $TUNNEL_ID media.$DOMAIN;
cloudflared tunnel route dns $TUNNEL_ID ai.$DOMAIN;
	
echo "[Unit]
Description=Cloudflare Tunnel Daemon
After=network.target

[Service]
Type=simple
User=$(whoami)
ExecStart=/usr/bin/cloudflared tunnel run raspi
Restart=always

[Install]
WantedBy=multi-user.target" | sudo tee /etc/systemd/system/cloudflared.service;

sudo systemctl daemon-reload && sudo systemctl enable cloudflared.service && sudo systemctl start cloudflared.service;

sudo sed -i -e 's/server_name localhost/servername ai.$DOMAIN/' /etc/nginx/sites-available/open-webui;
sudo sed -i -e 's/WEBUI_BASE_URL=localhost/WEBUI_BASE_URL=https://ai.$DOMAIN/' /etc/systemd/system/open-webui.service;

sudo systemctl daemon-reload && sudo systemctl restart open-webui.service;