From fe4b2f03bb9991366e26b8e573c09b6b1d267529 Mon Sep 17 00:00:00 2001
From: Mattias Persson <mattias.persson.87@gmail.com>
Date: Wed, 24 Jan 2024 20:52:09 +0100
Subject: [PATCH] Revert hass_url configuration option

---
 data/configuration.yaml                  |  1 +
 server.js                                |  1 -
 src/lib/Modal/LoginModal.svelte          | 15 ++++++++++-----
 src/lib/Socket.ts                        |  9 +++++----
 src/routes/+page.server.ts               | 23 ++++++++++++++++++++---
 src/routes/+page.svelte                  |  7 ++++++-
 src/routes/api/auth/+server.ts           |  5 +----
 src/routes/api/auth/[flow_id]/+server.ts |  5 +----
 8 files changed, 44 insertions(+), 22 deletions(-)

diff --git a/data/configuration.yaml b/data/configuration.yaml
index 0ce74eac..03a0c036 100644
--- a/data/configuration.yaml
+++ b/data/configuration.yaml
@@ -1 +1,2 @@
 locale: en
+hassUrl: http://192.168.1.241:8123
diff --git a/server.js b/server.js
index c1c5d04a..59a70f07 100644
--- a/server.js
+++ b/server.js
@@ -25,7 +25,6 @@ if (HASS_URL) {
 app.use(handler);
 
 app.listen(PORT, () => {
-	console.debug(`ENV HASS_URL: ${HASS_URL}`);
 	console.debug(`ENV PORT: ${PORT}`);
 	console.debug(`ENV ADDON ${process.env.ADDON || false}`);
 });
diff --git a/src/lib/Modal/LoginModal.svelte b/src/lib/Modal/LoginModal.svelte
index 3bd540e9..8cb975d5 100644
--- a/src/lib/Modal/LoginModal.svelte
+++ b/src/lib/Modal/LoginModal.svelte
@@ -1,5 +1,5 @@
 <script lang="ts">
-	import { configuration, lang, motion } from '$lib/Stores';
+	import { lang, motion } from '$lib/Stores';
 	import Modal from '$lib/Modal/Index.svelte';
 	import { Auth } from 'home-assistant-js-websocket';
 	import { closeModal } from 'svelte-modals';
@@ -11,6 +11,7 @@
 
 	export let isOpen: boolean;
 	export let clientId: string | undefined;
+	export let hassUrl: string | undefined;
 	export let flow_id: string | undefined;
 
 	let invalidAuth: string | undefined;
@@ -24,12 +25,16 @@
 	let code = '';
 
 	async function handleSubmit() {
-		if (!flow_id || (step === 'mfa' && code === '')) return;
+		if (!flow_id || !hassUrl || (step === 'mfa' && code === '')) {
+			console.error({ flow_id, hassUrl, code });
+			return;
+		}
 		disabled = true;
 
 		try {
 			// submit data
-			const payload = step === 'init' ? { username, password, clientId } : { code, clientId };
+			const payload =
+				step === 'init' ? { username, password, clientId, hassUrl } : { code, clientId, hassUrl };
 			const response = await fetch(`${base}/api/auth/${flow_id}`, {
 				method: 'POST',
 				headers: { 'Content-Type': 'application/json' },
@@ -67,14 +72,14 @@
 	async function handleSuccess(data: any) {
 		// update data
 		data.clientId = clientId;
-		data.hassUrl = $configuration?.hassUrl;
+		data.hassUrl = hassUrl;
 		data.expires = data.expires_in * 1000 + Date.now();
 
 		// save localStorage
 		localStorage.setItem('auth', JSON.stringify(data));
 
 		// connect to websocket
-		const auth = new Auth({ ...data, hassUrl: $configuration?.hassUrl, clientId });
+		const auth = new Auth({ ...data, hassUrl, clientId });
 		await webSocket(auth);
 
 		closeModal();
diff --git a/src/lib/Socket.ts b/src/lib/Socket.ts
index ca371692..efddbebf 100644
--- a/src/lib/Socket.ts
+++ b/src/lib/Socket.ts
@@ -14,7 +14,7 @@ import { states, connection, config, connected } from '$lib/Stores';
 import { openModal } from 'svelte-modals';
 import { base } from '$app/paths';
 
-export async function authenticate() {
+export async function authenticate(hassUrl: string) {
 	const storage = localStorage.getItem('auth');
 	const data = storage ? JSON.parse(storage) : null;
 
@@ -27,7 +27,8 @@ export async function authenticate() {
 		const clientId = genClientId();
 		openModal(async () => import('$lib/Modal/LoginModal.svelte'), {
 			clientId,
-			flow_id: await getFlowId(clientId)
+			hassUrl,
+			flow_id: await getFlowId(clientId, hassUrl)
 		});
 	}
 }
@@ -96,14 +97,14 @@ export async function webSocket(auth: Auth) {
 	}
 }
 
-export async function getFlowId(clientId: string) {
+export async function getFlowId(clientId: string, hassUrl: string) {
 	try {
 		const response = await fetch(`${base}/api/auth`, {
 			method: 'POST',
 			headers: {
 				'Content-Type': 'application/json'
 			},
-			body: JSON.stringify({ clientId })
+			body: JSON.stringify({ clientId, hassUrl })
 		});
 
 		const data = await response.json();
diff --git a/src/routes/+page.server.ts b/src/routes/+page.server.ts
index 5aa57b53..8003af47 100644
--- a/src/routes/+page.server.ts
+++ b/src/routes/+page.server.ts
@@ -1,4 +1,4 @@
-import { readFile } from 'fs/promises';
+import { readFile, writeFile } from 'fs/promises';
 import { dev } from '$app/environment';
 import yaml from 'js-yaml';
 import type { Configuration, Dashboard, Translations } from '$lib/Types';
@@ -30,7 +30,7 @@ async function loadFile(file: string) {
 /**
  * Server load function
  */
-export async function load(): Promise<{
+export async function load({ request }): Promise<{
 	configuration: Configuration;
 	dashboard: Dashboard;
 	theme: any;
@@ -42,7 +42,24 @@ export async function load(): Promise<{
 		loadFile('./data/dashboard.yaml')
 	]);
 
-	configuration.hassUrl = process.env.HASS_URL;
+	// determine hassUrl
+	let hassUrl = process.env.ADDON ? configuration.hassUrl : process.env.HASS_URL;
+
+	if (!hassUrl) {
+		const proto = request.headers.get('x-forwarded-proto');
+		const host = request.headers.get('x-forwarded-host');
+		if (proto && host) hassUrl = `${proto}://${host}`;
+	}
+
+	// update hassUrl
+	if (hassUrl && hassUrl !== configuration.hassUrl) {
+		configuration.hassUrl = hassUrl;
+		try {
+			await writeFile('./data/configuration.yaml', yaml.dump(configuration), 'utf8');
+		} catch (error) {
+			console.error('error updating configuration.yaml:', error);
+		}
+	}
 
 	// initialize keys if missing
 	dashboard.views = dashboard.views || [];
diff --git a/src/routes/+page.svelte b/src/routes/+page.svelte
index cedd578c..33e8c95c 100644
--- a/src/routes/+page.svelte
+++ b/src/routes/+page.svelte
@@ -52,7 +52,12 @@
 	/**
 	 * Connect to websocket
 	 */
-	if (browser) authenticate();
+
+	$: console.log('$configuration?.hassUrl', $configuration?.hassUrl);
+
+	if (browser && $configuration?.hassUrl) {
+		authenticate($configuration.hassUrl);
+	}
 
 	onMount(async () => {
 		/**
diff --git a/src/routes/api/auth/+server.ts b/src/routes/api/auth/+server.ts
index a64d0da2..b2035c22 100644
--- a/src/routes/api/auth/+server.ts
+++ b/src/routes/api/auth/+server.ts
@@ -1,12 +1,9 @@
 import type { RequestHandler } from './$types';
 import { json, error } from '@sveltejs/kit';
-import dotenv from 'dotenv';
-dotenv.config();
 
 export const POST: RequestHandler = async ({ request }) => {
 	try {
-		const hassUrl = process.env.HASS_URL;
-		const { clientId } = await request.json();
+		const { clientId, hassUrl } = await request.json();
 
 		const response = await fetch(`${hassUrl}/auth/login_flow`, {
 			method: 'POST',
diff --git a/src/routes/api/auth/[flow_id]/+server.ts b/src/routes/api/auth/[flow_id]/+server.ts
index c6e74764..becb5bc7 100644
--- a/src/routes/api/auth/[flow_id]/+server.ts
+++ b/src/routes/api/auth/[flow_id]/+server.ts
@@ -1,13 +1,10 @@
 import type { RequestHandler } from '@sveltejs/kit';
 import { json, error } from '@sveltejs/kit';
-import dotenv from 'dotenv';
-dotenv.config();
 
 export const POST: RequestHandler = async ({ params, request }) => {
 	try {
-		const hassUrl = process.env.HASS_URL;
 		const { flow_id } = params;
-		const { username, password, code, clientId } = await request.json();
+		const { username, password, code, clientId, hassUrl } = await request.json();
 
 		let payload;
 		if (username && password && clientId) {