From b6a2c98ae52f52ec7726a88faa8d2ae2bc8f619f Mon Sep 17 00:00:00 2001
From: Oscar Beaumont <oscar@otbeaumont.me>
Date: Fri, 9 Sep 2022 07:57:35 +0800
Subject: [PATCH] assorted minor changes

---
 core/src/lib.rs                             |  2 +-
 packages/client/src/rspc.ts                 | 17 +++++++++++++++++
 packages/interface/src/screens/Overview.tsx | 10 ----------
 3 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/core/src/lib.rs b/core/src/lib.rs
index d4cc15401..999118211 100644
--- a/core/src/lib.rs
+++ b/core/src/lib.rs
@@ -126,7 +126,7 @@ impl Node {
 		&str,    /* Content-Type */
 		Vec<u8>, /* Body */
 	) {
-		match path.get(0).copied() {
+		match path.first().copied() {
 			Some("thumbnail") => {
 				if path.len() != 2 {
 					return (
diff --git a/packages/client/src/rspc.ts b/packages/client/src/rspc.ts
index f41ff24f5..52aaf593a 100644
--- a/packages/client/src/rspc.ts
+++ b/packages/client/src/rspc.ts
@@ -2,6 +2,8 @@ import { RSPCError, createReactQueryHooks } from '@rspc/client';
 import { LibraryArgs, Operations } from '@sd/core';
 import {
 	QueryClient,
+	UseInfiniteQueryOptions,
+	UseInfiniteQueryResult,
 	UseMutationOptions,
 	UseMutationResult,
 	UseQueryOptions,
@@ -51,6 +53,21 @@ export function useLibraryQuery<K extends LibraryQueryKey>(
 	);
 }
 
+export function useInfiniteLibraryQuery<K extends LibraryQueryKey>(
+	key: LibraryQueryArgs<K> extends null | undefined ? [K] : [K, LibraryQueryArgs<K>],
+	options?: UseInfiniteQueryOptions<LibraryQueryResult<K>, RSPCError>
+): UseInfiniteQueryResult<LibraryQueryResult<K>, RSPCError> {
+	const { library } = useCurrentLibrary();
+
+	if (!library?.uuid) throw new Error(`Attempted to do library query with no library set!`);
+	// @ts-ignore
+	return rspc.useInfiniteQuery(
+		// @ts-ignore
+		[key[0], { library_id: library?.uuid || '', arg: key[1] || null }],
+		options
+	);
+}
+
 type LibraryMutations = Extract<Operations['mutations'], { key: [string, LibraryArgs<any>] }>;
 type LibraryMutation<K extends LibraryMutationKey> = Extract<LibraryMutations, { key: [K, any] }>;
 type LibraryMutationKey = LibraryMutations['key'][0];
diff --git a/packages/interface/src/screens/Overview.tsx b/packages/interface/src/screens/Overview.tsx
index 89dc73f99..c2ce5bfbd 100644
--- a/packages/interface/src/screens/Overview.tsx
+++ b/packages/interface/src/screens/Overview.tsx
@@ -148,16 +148,6 @@ export const OverviewScreen = () => {
 				<div className="flex w-full">
 					{/* STAT CONTAINER */}
 					<div className="flex -mb-1 overflow-hidden">
-						{!libraryStatistics && (
-							<div className="mb-2 ml-2">
-								<div className="font-semibold text-gray-200">
-									<ExclamationCircleIcon className="inline w-4 h-4 mr-1 -mt-1 " /> Missing library
-								</div>
-								<span className="text-xs text-gray-400 ">
-									Ensure the library you have loaded still exists on disk
-								</span>
-							</div>
-						)}
 						{Object.entries(overviewStats).map(([key, value]) => {
 							if (!displayableStatItems.includes(key)) return null;