From 8de4ee75ad9b2c4e56f132ca33e3fa68e53b1e9b Mon Sep 17 00:00:00 2001
From: Christopher Bisset <chris@gurucomputing.com.au>
Date: Wed, 20 Jul 2022 22:00:10 +1000
Subject: [PATCH] enable routing functionality

---
 src/lib/common/apiFunctions.svelte            | 74 ++++++++++++++++++-
 src/lib/common/classes.ts                     |  9 +++
 src/lib/devices/DeviceCard.svelte             |  2 +-
 .../devices/DeviceCard/DeviceRoutes.svelte    | 49 +++++++++++-
 4 files changed, 131 insertions(+), 3 deletions(-)

diff --git a/src/lib/common/apiFunctions.svelte b/src/lib/common/apiFunctions.svelte
index d3fdd3d..65fc62e 100644
--- a/src/lib/common/apiFunctions.svelte
+++ b/src/lib/common/apiFunctions.svelte
@@ -1,5 +1,5 @@
 <script context="module" lang="ts">
-	import { Device, PreAuthKey, User } from '$lib/common/classes';
+	import { Device, PreAuthKey, Route, User } from '$lib/common/classes';
 
 	export async function getUsers(): Promise<any> {
 		// variables in local storage
@@ -169,6 +169,78 @@
 		return headscaleDevices;
 	}
 
+	export async function getDeviceRoutes(deviceID: string): Promise<Route> {
+		// variables in local storage
+		let headscaleURL = localStorage.getItem('headscaleURL') || '';
+		let headscaleAPIKey = localStorage.getItem('headscaleAPIKey') || '';
+
+		// endpoint url for getting users
+		let endpointURL = '/api/v1/machine/' + deviceID + '/routes';
+
+		//returning variables
+		let headscaleRoute = new Route();
+		let headscaleDeviceResponse: Response = new Response();
+
+		await fetch(headscaleURL + endpointURL, {
+			method: 'GET',
+			headers: {
+				Accept: 'application/json',
+				Authorization: `Bearer ${headscaleAPIKey}`
+			}
+		})
+			.then((response) => {
+				if (response.ok) {
+					// return the api data
+					headscaleDeviceResponse = response;
+				} else {
+					return response.text().then((text) => {
+						throw JSON.parse(text).message;
+					});
+				}
+			})
+			.catch((error) => {
+				throw error;
+			});
+
+		await headscaleDeviceResponse.json().then((data) => {
+			headscaleRoute = data.routes;
+		});
+		return headscaleRoute;
+	}
+
+	export async function enableDeviceRoute(deviceID: string, route: string): Promise<any> {
+		// variables in local storage
+		let headscaleURL = localStorage.getItem('headscaleURL') || '';
+		let headscaleAPIKey = localStorage.getItem('headscaleAPIKey') || '';
+
+		// endpoint url for getting users
+		let endpointURL = '/api/v1/machine/' + deviceID + '/routes?routes=' + route.replace('/','%2F');
+
+		//returning variables
+		let headscaleDeviceResponse: Response = new Response();
+
+		await fetch(headscaleURL + endpointURL, {
+			method: 'POST',
+			headers: {
+				Accept: 'application/json',
+				Authorization: `Bearer ${headscaleAPIKey}`
+			}
+		})
+			.then((response) => {
+				if (response.ok) {
+					// return the api data
+					headscaleDeviceResponse = response;
+				} else {
+					return response.text().then((text) => {
+						throw JSON.parse(text).message;
+					});
+				}
+			})
+			.catch((error) => {
+				throw error;
+			});
+	}
+
 	export async function getPreauthKeys(userName: string): Promise<PreAuthKey[]> {
 		// variables in local storage
 		let headscaleURL = localStorage.getItem('headscaleURL') || '';
diff --git a/src/lib/common/classes.ts b/src/lib/common/classes.ts
index e8f79a6..47eb42c 100644
--- a/src/lib/common/classes.ts
+++ b/src/lib/common/classes.ts
@@ -10,6 +10,15 @@ export class Device {
   }
 }
 
+export class Route {
+  advertisedRoutes: string[] = [''];
+  enabledRoutes: string[] = [''];
+
+  public constructor(init?: Partial<Route>) {
+    Object.assign(this, init);
+  }
+}
+
 export class PreAuthKey {
   public namespace: string = '';
   public id: string = '';
diff --git a/src/lib/devices/DeviceCard.svelte b/src/lib/devices/DeviceCard.svelte
index 979d9e4..3b01a92 100644
--- a/src/lib/devices/DeviceCard.svelte
+++ b/src/lib/devices/DeviceCard.svelte
@@ -61,7 +61,7 @@
 							<MoveDevice {device} />
 						</tr>
 						<tr>
-							<DeviceRoutes />
+							<DeviceRoutes {device} />
 						</tr>
 					</tbody>
 				</table>
diff --git a/src/lib/devices/DeviceCard/DeviceRoutes.svelte b/src/lib/devices/DeviceCard/DeviceRoutes.svelte
index 9123691..bde17e6 100644
--- a/src/lib/devices/DeviceCard/DeviceRoutes.svelte
+++ b/src/lib/devices/DeviceCard/DeviceRoutes.svelte
@@ -1,2 +1,49 @@
+<script lang="ts">
+	import { getDeviceRoutes, enableDeviceRoute } from '$lib/common/apiFunctions.svelte';
+	import { Device, Route } from '$lib/common/classes';
+	import { onMount } from 'svelte';
+	import { alertStore } from '$lib/common/stores';
+
+	export let device = new Device();
+	let routesList = new Route();
+
+	onMount(async () => {
+		getDeviceRoutesAction();
+	});
+
+	function getDeviceRoutesAction() {
+		getDeviceRoutes(device.id)
+			.then((routes) => {
+				routesList = routes;
+			})
+			.catch((error) => {
+				$alertStore = error;
+			});
+	}
+
+  function enableDeviceRouteAction(route: string) {
+    enableDeviceRoute(device.id, route)
+			.then((response) => {
+				getDeviceRoutesAction();
+			})
+			.catch((error) => {
+				$alertStore = error;
+			});
+  }
+</script>
+
 <th>Device Routes</th>
-<td>blah</td>
\ No newline at end of file
+<td
+	><ul class="list-disc list-inside">
+		{#each routesList.advertisedRoutes as route}
+			<li>
+				{route}
+				{#if routesList.enabledRoutes.includes(route)}
+					<div class="btn btn-xs capitalize bg-success text-success-content mx-1">active</div>
+				{:else}
+					<button on:click={() => {enableDeviceRouteAction(route)}} type="button" class="btn btn-xs tooltip capitalize bg-secondary text-secondary-content mx-1" data-tip="press to enable route">pending</button>
+				{/if}
+			</li>
+		{/each}
+	</ul></td
+>