284 lines
12 KiB
Vue
284 lines
12 KiB
Vue
<script setup lang="ts">
|
|
import { ref, computed, watch, nextTick } from 'vue';
|
|
import { useFloating, offset, flip, shift, autoUpdate, size } from '@floating-ui/vue';
|
|
import type { Entity } from '@triplit/client';
|
|
import type schema from '#triplit/schema';
|
|
import type { ModelWithProvider, ProviderWithModels } from '~/composables/useModels';
|
|
import { sortByReleaseDate } from '~/utils/sort';
|
|
|
|
const { openDialog } = useDialog();
|
|
const { allModels } = useModels();
|
|
const { addShortcut } = useKeyboardShortcuts();
|
|
|
|
const props = defineProps<{
|
|
providers: ProviderWithModels[];
|
|
addHotkey?: boolean;
|
|
}>();
|
|
|
|
const selectedModel = defineModel<ModelWithProvider | null>();
|
|
|
|
const isOpen = ref(false);
|
|
const dropdownButton = ref<HTMLElement | null>(null);
|
|
const dropdownContent = ref<HTMLElement | null>(null);
|
|
const searchInputRef = ref<HTMLInputElement | null>(null);
|
|
|
|
const navigatingWithKeyboard = ref(false);
|
|
const focusedOptionId = ref<string | null>(null);
|
|
|
|
const { floatingStyles, placement } = useFloating(dropdownButton, dropdownContent, {
|
|
placement: 'bottom-start',
|
|
whileElementsMounted: autoUpdate,
|
|
middleware: [
|
|
offset(6),
|
|
flip(),
|
|
shift({ padding: 10 }),
|
|
size({
|
|
apply({ availableHeight, elements }) {
|
|
Object.assign(elements.floating.style, {
|
|
maxHeight: `${Math.min(availableHeight - 20, 460)}px`,
|
|
});
|
|
},
|
|
padding: 10,
|
|
}),
|
|
],
|
|
transform: false,
|
|
});
|
|
|
|
const transformOrigin = computed(() =>
|
|
placement.value.startsWith('top')
|
|
? 'transform-origin-bottom-center'
|
|
: 'transform-origin-top-center'
|
|
);
|
|
|
|
const toggleDropdown = () => {
|
|
isOpen.value = !isOpen.value;
|
|
};
|
|
|
|
const searchQuery = ref('');
|
|
|
|
const filteredProviders = computed(() => {
|
|
return filterProvidersWithModel(props.providers, searchQuery.value).filter(p => p.enabled);
|
|
});
|
|
|
|
const flatOptions = computed(() => {
|
|
const options: { id: string; model: Entity<typeof schema, 'models'>; provider: Entity<typeof schema, 'providers'> }[] = [];
|
|
for (const provider of filteredProviders.value) {
|
|
const enabledModels = provider.models.filter(m => m.enabled).sort((a, b) => b.releasedAt && a.releasedAt ? b.releasedAt.getTime() - a.releasedAt.getTime() : 0);
|
|
for (const model of enabledModels) {
|
|
options.push({ id: model.id, model, provider });
|
|
}
|
|
}
|
|
return options;
|
|
});
|
|
|
|
const focusedIndex = computed(() => {
|
|
if (!focusedOptionId.value) return -1;
|
|
return flatOptions.value.findIndex(o => o.id === focusedOptionId.value);
|
|
});
|
|
|
|
const setFocusToOption = (index: number) => {
|
|
if (flatOptions.value.length === 0) return;
|
|
navigatingWithKeyboard.value = true;
|
|
|
|
const clampedIndex = Math.max(0, Math.min(index, flatOptions.value.length - 1));
|
|
focusedOptionId.value = flatOptions.value[clampedIndex]!.id;
|
|
scrollFocusedIntoView();
|
|
};
|
|
|
|
const scrollFocusedIntoView = () => {
|
|
nextTick(() => {
|
|
const focusedEl = document.getElementById(`model-option-${focusedOptionId.value}`);
|
|
focusedEl?.scrollIntoView({ block: 'nearest' });
|
|
});
|
|
};
|
|
|
|
const selectModel = (model: Entity<typeof schema, 'models'>, provider: Entity<typeof schema, 'providers'>) => {
|
|
selectedModel.value = { ...model, provider };
|
|
closeDropdown();
|
|
};
|
|
|
|
watch(() => props.providers, () => {
|
|
if (selectedModel.value) {
|
|
const modelStillExists = allModels.value.some(model => model.id === selectedModel.value?.id);
|
|
|
|
if (!modelStillExists) {
|
|
let firstEnabled: ModelWithProvider | null = null;
|
|
|
|
for (const provider of props.providers) {
|
|
const enabledModel = provider.models.find(m => m.enabled);
|
|
if (enabledModel) {
|
|
firstEnabled = { ...enabledModel, provider };
|
|
break;
|
|
}
|
|
}
|
|
|
|
selectedModel.value = firstEnabled;
|
|
}
|
|
}
|
|
});
|
|
|
|
watch(isOpen, (open) => {
|
|
if (open) {
|
|
nextTick(() => {
|
|
searchInputRef.value?.focus();
|
|
});
|
|
} else {
|
|
focusedOptionId.value = null;
|
|
searchQuery.value = '';
|
|
}
|
|
});
|
|
|
|
const closeDropdown = () => {
|
|
isOpen.value = false;
|
|
navigatingWithKeyboard.value = false;
|
|
};
|
|
|
|
const handleSearchKeyDown = (event: KeyboardEvent) => {
|
|
if (flatOptions.value.length === 0) return;
|
|
|
|
switch (event.key) {
|
|
case 'Escape':
|
|
closeDropdown();
|
|
event.preventDefault();
|
|
break;
|
|
case 'ArrowDown':
|
|
event.preventDefault();
|
|
if (focusedIndex.value === -1 || focusedIndex.value === flatOptions.value.length - 1) {
|
|
setFocusToOption(0);
|
|
} else {
|
|
setFocusToOption(focusedIndex.value + 1);
|
|
}
|
|
break;
|
|
case 'ArrowUp':
|
|
event.preventDefault();
|
|
if (focusedIndex.value === -1 || focusedIndex.value === 0) {
|
|
setFocusToOption(flatOptions.value.length - 1);
|
|
} else {
|
|
setFocusToOption(focusedIndex.value - 1);
|
|
}
|
|
break;
|
|
case 'Home':
|
|
event.preventDefault();
|
|
setFocusToOption(0);
|
|
break;
|
|
case 'End':
|
|
event.preventDefault();
|
|
setFocusToOption(flatOptions.value.length - 1);
|
|
break;
|
|
case 'Enter':
|
|
event.preventDefault();
|
|
if (focusedOptionId.value) {
|
|
const option = flatOptions.value.find(o => o.id === focusedOptionId.value);
|
|
if (option) {
|
|
selectModel(option.model, option.provider);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (flatOptions.value.length > 0) {
|
|
selectModel(flatOptions.value[0]!.model, flatOptions.value[0]!.provider);
|
|
}
|
|
|
|
break;
|
|
}
|
|
};
|
|
|
|
let unbindShortcut: () => void = () => { };
|
|
|
|
if (props.addHotkey) {
|
|
unbindShortcut = addShortcut(['ctrl', 'shift', 'm'], (event) => {
|
|
event.preventDefault();
|
|
event.stopPropagation();
|
|
toggleDropdown();
|
|
});
|
|
}
|
|
|
|
onUnmounted(() => {
|
|
unbindShortcut();
|
|
});
|
|
</script>
|
|
|
|
<template>
|
|
<div role="button" @click="toggleDropdown()" ref="dropdownButton"
|
|
class="cursor-pointer flex items-center w-fit select-none gap-2 px-3 py-1.5 rounded-lg text-sm font-medium transition-colors duration-200"
|
|
:class="[
|
|
isOpen
|
|
? 'bg-[var(--color-hover)] text-[var(--text-primary)]'
|
|
: 'text-[var(--text-secondary)] hover:text-[var(--text-primary)] hover:bg-[var(--color-hover)]',
|
|
]">
|
|
<ModelIcon v-if="selectedModel" class="text-white" :avatar="true" variant="color"
|
|
:model-id="selectedModel.externalId" size="22" />
|
|
<span class="max-w-full truncate">
|
|
{{ selectedModel ? selectedModel.name : 'Select a model' }}
|
|
</span>
|
|
<span class="i-mynaui-chevron-down text-3.5 transition-transform duration-200"
|
|
:class="{ 'rotate-180': isOpen }"></span>
|
|
</div>
|
|
|
|
<Teleport to="body">
|
|
<Transition
|
|
enter-active-class="transition-[opacity,transform] duration-150 ease-[cubic-bezier(0.5,_1,_0.89,_1)]"
|
|
enter-from-class="opacity-0 scale-95 translate-y-1" enter-to-class="opacity-100 scale-100 translate-y-0"
|
|
leave-active-class="transition-[opacity,transform] duration-100 ease-[cubic-bezier(0.5,_1,_0.89,_1)]"
|
|
leave-from-class="opacity-100 scale-100 translate-y-0" leave-to-class="opacity-0 scale-95 translate-y-1">
|
|
<KeepAlive>
|
|
<div v-if="isOpen" v-click-outside="closeDropdown" ref="dropdownContent" role="listbox"
|
|
aria-label="Select model"
|
|
:aria-activedescendant="focusedOptionId ? `model-option-${focusedOptionId}` : undefined"
|
|
:style="floatingStyles"
|
|
class="fixed z-30 w-full max-w-[420px] min-w-[280px] flex flex-col rounded-xl border border-[var(--color-border)] bg-[var(--bg-surface)] shadow-lg overflow-hidden"
|
|
:class="transformOrigin">
|
|
|
|
<div class="relative">
|
|
<span
|
|
class="i-mynaui-search absolute left-3 top-1/2 -translate-y-1/2 text-4 text-[var(--text-secondary)]"></span>
|
|
<input ref="searchInputRef" v-model="searchQuery" autocomplete="off" name="search"
|
|
@keydown="handleSearchKeyDown" type="text" placeholder="Search models..."
|
|
class="placeholder:text-[var(--text-tertiary)] w-full pl-9 pr-3 py-2 text-sm text-[var(--text-primary)] bg-transparent placeholder-[var(--text-secondary)] outline-none" />
|
|
</div>
|
|
|
|
<div class="flex-1 overflow-y-auto [scrollbar-width:thin] py-2 select-none">
|
|
<div v-if="filteredProviders.length === 0"
|
|
class="px-4 py-8 text-center text-sm text-[var(--text-secondary)]">
|
|
No models found
|
|
</div>
|
|
|
|
<div v-for="provider in filteredProviders" :key="provider.id" class="mb-2">
|
|
<div v-if="provider.models.filter(m => m.enabled).length > 0"
|
|
class="px-4 py-1.5 text-[13px] font-medium text-[var(--text-secondary)] case-capital tracking-wider flex justify-between">
|
|
{{ provider.name }}
|
|
<button
|
|
@click="closeDropdown(); openDialog(DialogType.Settings, undefined, { page: 'providers', params: provider.id })"
|
|
class="flex h-5 w-5 items-center justify-center hover:bg-[var(--color-hover)] rounded transition-colors duration-200 ease-[cubic-bezier(0.5,_1,_0.89,_1)]">
|
|
<span class="i-mynaui-cog-four text-4"></span>
|
|
</button>
|
|
</div>
|
|
|
|
<button v-for="model in provider.models.filter(m => m.enabled).sort(sortByReleaseDate)"
|
|
:key="model.id" :id="`model-option-${model.id}`" role="option"
|
|
:aria-selected="focusedOptionId === model.id" @click="selectModel(model, provider)"
|
|
class="text-white w-full min-h-9 px-4 py-2 flex items-center justify-between hover:bg-[var(--color-hover)] transition-colors duration-150"
|
|
:class="{
|
|
'bg-[var(--color-hover)]': selectedModel?.id === model.id,
|
|
'ring-2 ring-inset ring-[var(--color-accent)]': focusedOptionId === model.id && navigatingWithKeyboard
|
|
}">
|
|
<ModelInfo :model="model" size="medium" />
|
|
</button>
|
|
</div>
|
|
</div>
|
|
|
|
<div class="p-1 border-t border-[var(--color-border)]">
|
|
<button
|
|
class="flex w-full items-center gap-2 px-3 py-2 text-sm text-[var(--text-secondary)] hover:text-[var(--text-primary)] hover:bg-[var(--color-hover)] rounded-lg transition-colors duration-150"
|
|
@click="closeDropdown(); openDialog(DialogType.Settings, undefined, { page: 'providers' });">
|
|
<span class="i-mynaui-cog-four text-4.5"></span>
|
|
<span>Manage Providers</span>
|
|
<span class="i-mynaui-arrow-right text-4 ml-auto"></span>
|
|
</button>
|
|
</div>
|
|
</div>
|
|
</KeepAlive>
|
|
</Transition>
|
|
</Teleport>
|
|
</template>
|