From dd8999a45f7e9f8f00c873ba6214f37b8461c401 Mon Sep 17 00:00:00 2001
From: Sean Morley <zipsm15@gmail.com>
Date: Mon, 19 Aug 2024 16:32:08 -0400
Subject: [PATCH] Transportation changes

---
 .../0004_transportation_end_date.py           | 18 +++++++
 backend/server/adventures/models.py           | 10 ++++
 backend/server/adventures/serializers.py      |  2 +-
 .../src/lib/components/AdventureModal.svelte  | 28 +++++-----
 .../lib/components/EditTransportation.svelte  | 37 ++++++++++++-
 .../lib/components/NewTransportation.svelte   | 23 +++++++-
 .../lib/components/TransportationCard.svelte  | 30 +++++++----
 frontend/src/lib/index.ts                     | 53 +++++++++----------
 frontend/src/lib/types.ts                     |  1 +
 9 files changed, 150 insertions(+), 52 deletions(-)
 create mode 100644 backend/server/adventures/migrations/0004_transportation_end_date.py

diff --git a/backend/server/adventures/migrations/0004_transportation_end_date.py b/backend/server/adventures/migrations/0004_transportation_end_date.py
new file mode 100644
index 0000000..04f3b57
--- /dev/null
+++ b/backend/server/adventures/migrations/0004_transportation_end_date.py
@@ -0,0 +1,18 @@
+# Generated by Django 5.0.8 on 2024-08-19 20:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('adventures', '0003_adventure_end_date'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='transportation',
+            name='end_date',
+            field=models.DateTimeField(blank=True, null=True),
+        ),
+    ]
diff --git a/backend/server/adventures/models.py b/backend/server/adventures/models.py
index fbc94db..7df214f 100644
--- a/backend/server/adventures/models.py
+++ b/backend/server/adventures/models.py
@@ -55,6 +55,10 @@ class Adventure(models.Model):
     updated_at = models.DateTimeField(auto_now=True)
 
     def clean(self):
+        if self.date and self.end_date and self.date > self.end_date:
+            raise ValidationError('The start date must be before the end date. Start date: ' + str(self.date) + ' End date: ' + str(self.end_date))
+        if self.end_date and not self.date:
+            raise ValidationError('Adventures must have an end date. Adventure: ' + self.name)
         if self.collection:
             if self.collection.is_public and not self.is_public:
                 raise ValidationError('Adventures associated with a public collection must be public. Collection: ' + self.trip.name + ' Adventure: ' + self.name)
@@ -100,6 +104,7 @@ class Transportation(models.Model):
     rating = models.FloatField(blank=True, null=True)
     link = models.URLField(blank=True, null=True)
     date = models.DateTimeField(blank=True, null=True)
+    end_date = models.DateTimeField(blank=True, null=True)
     flight_number = models.CharField(max_length=100, blank=True, null=True)
     from_location = models.CharField(max_length=200, blank=True, null=True)
     to_location = models.CharField(max_length=200, blank=True, null=True)
@@ -109,6 +114,11 @@ class Transportation(models.Model):
     updated_at = models.DateTimeField(auto_now=True)
 
     def clean(self):
+        print(self.date)
+        if self.date and self.end_date and self.date > self.end_date:
+            raise ValidationError('The start date must be before the end date. Start date: ' + str(self.date) + ' End date: ' + str(self.end_date))
+        if self.end_date and self.date is 'null':
+            raise ValidationError('Transportations must have an end date. Transportation: ' + self.name)
         if self.collection:
             if self.collection.is_public and not self.is_public:
                 raise ValidationError('Transportations associated with a public collection must be public. Collection: ' + self.collection.name + ' Transportation: ' + self.name)
diff --git a/backend/server/adventures/serializers.py b/backend/server/adventures/serializers.py
index 620f659..9acd0b2 100644
--- a/backend/server/adventures/serializers.py
+++ b/backend/server/adventures/serializers.py
@@ -51,7 +51,7 @@ class TransportationSerializer(serializers.ModelSerializer):
         fields = [
             'id', 'user_id', 'type', 'name', 'description', 'rating', 
             'link', 'date', 'flight_number', 'from_location', 'to_location', 
-            'is_public', 'collection', 'created_at', 'updated_at'
+            'is_public', 'collection', 'created_at', 'updated_at', 'end_date'
         ]
         read_only_fields = ['id', 'created_at', 'updated_at', 'user_id']
 
diff --git a/frontend/src/lib/components/AdventureModal.svelte b/frontend/src/lib/components/AdventureModal.svelte
index 1af13d8..405bab3 100644
--- a/frontend/src/lib/components/AdventureModal.svelte
+++ b/frontend/src/lib/components/AdventureModal.svelte
@@ -383,7 +383,7 @@
 					</div>
 
 					<div>
-						<label for="date">Date (or start date)</label><br />
+						<label for="date">{adventure.date ? 'Start Date' : 'Date'}</label><br />
 						<input
 							type="date"
 							id="date"
@@ -394,18 +394,20 @@
 							class="input input-bordered w-full"
 						/>
 					</div>
-					<div>
-						<label for="end_date">End Date</label><br />
-						<input
-							type="date"
-							id="end_date"
-							name="end_date"
-							min={startDate || ''}
-							max={endDate || ''}
-							bind:value={adventure.end_date}
-							class="input input-bordered w-full"
-						/>
-					</div>
+					{#if adventure.date}
+						<div>
+							<label for="end_date">End Date</label><br />
+							<input
+								type="date"
+								id="end_date"
+								name="end_date"
+								min={startDate || ''}
+								max={endDate || ''}
+								bind:value={adventure.end_date}
+								class="input input-bordered w-full"
+							/>
+						</div>
+					{/if}
 					<div>
 						<!-- link -->
 						<div>
diff --git a/frontend/src/lib/components/EditTransportation.svelte b/frontend/src/lib/components/EditTransportation.svelte
index d304d4b..289f165 100644
--- a/frontend/src/lib/components/EditTransportation.svelte
+++ b/frontend/src/lib/components/EditTransportation.svelte
@@ -40,6 +40,9 @@
 	if (transportationToEdit.date) {
 		transportationToEdit.date = transportationToEdit.date.slice(0, 19);
 	}
+	if (transportationToEdit.end_date) {
+		transportationToEdit.end_date = transportationToEdit.end_date.slice(0, 19);
+	}
 
 	function close() {
 		dispatch('close');
@@ -53,6 +56,20 @@
 
 	async function handleSubmit(event: Event) {
 		event.preventDefault();
+		// make sure end_date is not before start_date
+		if (
+			transportationToEdit.end_date &&
+			transportationToEdit.date &&
+			transportationToEdit.date > transportationToEdit.end_date
+		) {
+			addToast('error', 'End date cannot be before start date');
+			return;
+		}
+		// make sure end_date has a start_date
+		if (transportationToEdit.end_date && !transportationToEdit.date) {
+			addToast('error', 'Please provide a start date');
+			return;
+		}
 		const form = event.target as HTMLFormElement;
 		const formData = new FormData(form);
 
@@ -147,7 +164,9 @@
 					</div>
 					<div class="mb-2">
 						<label for="start_date"
-							>Date & Time <Calendar class="inline-block mb-1 w-6 h-6" /></label
+							>{transportationToEdit.date ? 'Start' : ''}Date & Time <Calendar
+								class="inline-block mb-1 w-6 h-6"
+							/></label
 						><br />
 						<input
 							type="datetime-local"
@@ -159,6 +178,22 @@
 							class="input input-bordered w-full max-w-xs mt-1"
 						/>
 					</div>
+					{#if transportationToEdit.date}
+						<div class="mb-2">
+							<label for="end_date"
+								>End Date & Time <Calendar class="inline-block mb-1 w-6 h-6" /></label
+							><br />
+							<input
+								type="datetime-local"
+								id="end_date"
+								name="end_date"
+								min={fullStartDate || ''}
+								max={fullEndDate || ''}
+								bind:value={transportationToEdit.end_date}
+								class="input input-bordered w-full max-w-xs mt-1"
+							/>
+						</div>
+					{/if}
 					<div class="mb-2">
 						<label for="rating">Rating <Star class="inline-block mb-1 w-6 h-6" /></label><br />
 						<input
diff --git a/frontend/src/lib/components/NewTransportation.svelte b/frontend/src/lib/components/NewTransportation.svelte
index a49b12b..782ed3a 100644
--- a/frontend/src/lib/components/NewTransportation.svelte
+++ b/frontend/src/lib/components/NewTransportation.svelte
@@ -72,6 +72,12 @@
 		const form = event.target as HTMLFormElement;
 		const formData = new FormData(form);
 
+		// make sure there is a start date if there is an end date
+		if (formData.get('end_date') && !formData.get('date')) {
+			addToast('error', 'Please provide a start date');
+			return;
+		}
+
 		const response = await fetch(`/api/transportations/`, {
 			method: 'POST',
 			body: formData
@@ -168,7 +174,7 @@
 					</div>
 					<div class="mb-2">
 						<label for="start_date"
-							>Date & Time <Calendar class="inline-block mb-1 w-6 h-6" /></label
+							>Start Date & Time <Calendar class="inline-block mb-1 w-6 h-6" /></label
 						><br />
 						<input
 							type="datetime-local"
@@ -179,6 +185,21 @@
 							class="input input-bordered w-full max-w-xs mt-1"
 						/>
 					</div>
+
+					<div class="mb-2">
+						<label for="end_date"
+							>End Date & Time <Calendar class="inline-block mb-1 w-6 h-6" /></label
+						><br />
+						<input
+							type="datetime-local"
+							id="end_date"
+							name="end_date"
+							min={fullStartDate || ''}
+							max={fullEndDate || ''}
+							class="input input-bordered w-full max-w-xs mt-1"
+						/>
+					</div>
+
 					<div class="mb-2">
 						<label for="rating">Rating <Star class="inline-block mb-1 w-6 h-6" /></label><br />
 						<input
diff --git a/frontend/src/lib/components/TransportationCard.svelte b/frontend/src/lib/components/TransportationCard.svelte
index dd2a53a..efd5001 100644
--- a/frontend/src/lib/components/TransportationCard.svelte
+++ b/frontend/src/lib/components/TransportationCard.svelte
@@ -11,6 +11,7 @@
 	import { addToast } from '$lib/toasts';
 
 	import Plus from '~icons/mdi/plus';
+	import ArrowDownThick from '~icons/mdi/arrow-down-thick';
 
 	const dispatch = createEventDispatcher();
 
@@ -39,19 +40,30 @@
 </script>
 
 <div
-	class="card min-w-max lg:w-96 md:w-80 sm:w-60 xs:w-40 bg-primary-content shadow-xl overflow-hidden text-base-content"
+	class="card w-full max-w-xs sm:max-w-sm md:max-w-md lg:max-w-md xl:max-w-md bg-primary-content shadow-xl text-base-content"
 >
 	<div class="card-body">
 		<h2 class="card-title overflow-ellipsis">{transportation.name}</h2>
 		<div class="badge badge-secondary">{transportation.type}</div>
-		{#if transportation.from_location && transportation.to_location}
-			<p class="text-sm">
-				{transportation.from_location} to {transportation.to_location}
-			</p>
-		{/if}
-		{#if transportation.date}
-			{new Date(transportation.date).toLocaleString(undefined, { timeZone: 'UTC' })}
-		{/if}
+		<div>
+			{#if transportation.from_location}
+				<p class="break-words text-wrap">{transportation.from_location}</p>
+			{/if}
+			{#if transportation.to_location}
+				<ArrowDownThick class="w-6 h-6" />
+				<p class="break-words text-wrap">{transportation.to_location}</p>
+			{/if}
+		</div>
+		<div>
+			{#if transportation.date}
+				<p>{new Date(transportation.date).toLocaleString(undefined, { timeZone: 'UTC' })}</p>
+			{/if}
+			{#if transportation.end_date}
+				<ArrowDownThick class="w-6 h-6" />
+				<p>{new Date(transportation.end_date).toLocaleString(undefined, { timeZone: 'UTC' })}</p>
+			{/if}
+		</div>
+
 		{#if user?.pk === transportation.user_id}
 			<div class="card-actions justify-end">
 				<button on:click={deleteTransportation} class="btn btn-secondary"
diff --git a/frontend/src/lib/index.ts b/frontend/src/lib/index.ts
index 39196a9..057e8bf 100644
--- a/frontend/src/lib/index.ts
+++ b/frontend/src/lib/index.ts
@@ -37,31 +37,6 @@ export async function exportData() {
 		visitedRegions
 	};
 
-	async function convertImages() {
-		const promises = data.adventures.map(async (adventure, i) => {
-			if (adventure.image) {
-				const res = await fetch(adventure.image);
-				const blob = await res.blob();
-				const base64 = await blobToBase64(blob);
-				adventure.image = base64;
-				data.adventures[i].image = adventure.image;
-			}
-		});
-
-		await Promise.all(promises);
-	}
-
-	function blobToBase64(blob: Blob): Promise<string> {
-		return new Promise((resolve, reject) => {
-			const reader = new FileReader();
-			reader.readAsDataURL(blob);
-			reader.onloadend = () => resolve(reader.result as string);
-			reader.onerror = (error) => reject(error);
-		});
-	}
-
-	await convertImages();
-
 	const blob = new Blob([JSON.stringify(data)], { type: 'application/json' });
 	return URL.createObjectURL(blob);
 }
@@ -92,7 +67,19 @@ export function groupAdventuresByDate(
 	adventures.forEach((adventure) => {
 		if (adventure.date) {
 			const adventureDate = new Date(adventure.date).toISOString().split('T')[0];
-			if (groupedAdventures[adventureDate]) {
+			if (adventure.end_date) {
+				const endDate = new Date(adventure.end_date).toISOString().split('T')[0];
+				const currentDate = new Date(startDate);
+				for (let i = 0; i < numberOfDays; i++) {
+					currentDate.setDate(startDate.getDate() + i);
+					const dateString = currentDate.toISOString().split('T')[0];
+					if (dateString >= adventureDate && dateString <= endDate) {
+						if (groupedAdventures[dateString]) {
+							groupedAdventures[dateString].push(adventure);
+						}
+					}
+				}
+			} else if (groupedAdventures[adventureDate]) {
 				groupedAdventures[adventureDate].push(adventure);
 			}
 		}
@@ -118,7 +105,19 @@ export function groupTransportationsByDate(
 	transportations.forEach((transportation) => {
 		if (transportation.date) {
 			const transportationDate = new Date(transportation.date).toISOString().split('T')[0];
-			if (groupedTransportations[transportationDate]) {
+			if (transportation.end_date) {
+				const endDate = new Date(transportation.end_date).toISOString().split('T')[0];
+				const currentDate = new Date(startDate);
+				for (let i = 0; i < numberOfDays; i++) {
+					currentDate.setDate(startDate.getDate() + i);
+					const dateString = currentDate.toISOString().split('T')[0];
+					if (dateString >= transportationDate && dateString <= endDate) {
+						if (groupedTransportations[dateString]) {
+							groupedTransportations[dateString].push(transportation);
+						}
+					}
+				}
+			} else if (groupedTransportations[transportationDate]) {
 				groupedTransportations[transportationDate].push(transportation);
 			}
 		}
diff --git a/frontend/src/lib/types.ts b/frontend/src/lib/types.ts
index 3811a20..251c6e1 100644
--- a/frontend/src/lib/types.ts
+++ b/frontend/src/lib/types.ts
@@ -105,6 +105,7 @@ export type Transportation = {
 	rating: number | null;
 	link: string | null;
 	date: string | null; // ISO 8601 date string
+	end_date: string | null; // ISO 8601 date string
 	flight_number: string | null;
 	from_location: string | null;
 	to_location: string | null;