[questions][fix] fix expanded typeaheads

apps/portal/src/components/questions/card/question/BaseQuestionCard.tsx

@@ -185,7 +185,7 @@ export default function BaseQuestionCard({
       )}
       <div className="flex flex-1 flex-col items-start gap-2">
         <div className="flex items-baseline justify-between self-stretch">
-          <div className="z-10 flex items-center gap-2 text-slate-500">
+          <div className="z-10 flex flex-wrap items-center gap-2 text-slate-500">
             {showAggregateStatistics && (
               <>
                 <QuestionTypeBadge type={type} />

apps/portal/src/components/questions/forms/CreateQuestionEncounterForm.tsx

@@ -3,11 +3,15 @@ import { useState } from 'react';
 import { CheckIcon } from '@heroicons/react/20/solid';
 import { Button } from '@tih/ui';
 
+import { useCompanyOptions } from '~/components/shared/CompaniesTypeahead';
+import { useJobTitleOptions } from '~/components/shared/JobTitlesTypeahead';
 import type { Month } from '~/components/shared/MonthYearPicker';
 import MonthYearPicker from '~/components/shared/MonthYearPicker';
 
 import CompanyTypeahead from '../typeahead/CompanyTypeahead';
-import LocationTypeahead from '../typeahead/LocationTypeahead';
+import LocationTypeahead, {
+  useLocationOptions,
+} from '../typeahead/LocationTypeahead';
 import RoleTypeahead from '../typeahead/RoleTypeahead';
 
 import type { Location } from '~/types/questions';
@@ -43,6 +47,10 @@ export default function CreateQuestionEncounterForm({
     startOfMonth(new Date()),
   );
 
+  const { data: allCompanyOptions } = useCompanyOptions('');
+  const { data: allLocationOptions } = useLocationOptions('');
+  const allRoleOptions  = useJobTitleOptions('');
+
   if (submitted) {
     return (
       <div className="font-md flex items-center gap-1 rounded-full border bg-slate-50 py-1 pl-2 pr-3 text-sm text-slate-500">
@@ -53,7 +61,7 @@ export default function CreateQuestionEncounterForm({
   }
 
   return (
-    <div className="flex items-center gap-2">
+    <div className="flex items-center gap-2 flex-wrap">
       <p className="text-md text-slate-600">
         I saw this question {step <= 1 ? 'at' : step === 2 ? 'for' : 'on'}
       </p>
@@ -62,8 +70,8 @@ export default function CreateQuestionEncounterForm({
           <CompanyTypeahead
             isLabelHidden={true}
             placeholder="Company"
-            // TODO: Fix suggestions and set count back to 3
-            suggestedCount={0}
+            suggestedCount={3}
+            suggestedOptions={allCompanyOptions}
             onSelect={({ value: company }) => {
               setSelectedCompany(company);
             }}
@@ -79,7 +87,8 @@ export default function CreateQuestionEncounterForm({
           <LocationTypeahead
             isLabelHidden={true}
             placeholder="Location"
-            suggestedCount={0}
+            suggestedCount={3}
+            suggestedOptions={allLocationOptions}
             onSelect={(location) => {
               setSelectedLocation(location);
             }}
@@ -95,7 +104,8 @@ export default function CreateQuestionEncounterForm({
           <RoleTypeahead
             isLabelHidden={true}
             placeholder="Role"
-            suggestedCount={0}
+            suggestedCount={3}
+            suggestedOptions={allRoleOptions}
             onSelect={({ value: role }) => {
               setSelectedRole(role);
             }}

apps/portal/src/components/questions/typeahead/CompanyTypeahead.tsx

@@ -1,6 +1,6 @@
-import { useMemo, useState } from 'react';
+import { useState } from 'react';
 
-import { trpc } from '~/utils/trpc';
+import { useCompanyOptions } from '~/components/shared/CompaniesTypeahead';
 
 import type { ExpandedTypeaheadProps } from './ExpandedTypeahead';
 import ExpandedTypeahead from './ExpandedTypeahead';
@@ -13,22 +13,7 @@ export type CompanyTypeaheadProps = Omit<
 export default function CompanyTypeahead(props: CompanyTypeaheadProps) {
   const [query, setQuery] = useState('');
 
-  const { data: companies, isLoading } = trpc.useQuery([
-    'companies.list',
-    {
-      name: query,
-    },
-  ]);
-
-  const companyOptions = useMemo(() => {
-    return (
-      companies?.map(({ id, name }) => ({
-        id,
-        label: name,
-        value: id,
-      })) ?? []
-    );
-  }, [companies]);
+  const { data: companyOptions, isLoading } = useCompanyOptions(query);
 
   return (
     <ExpandedTypeahead

apps/portal/src/components/questions/typeahead/ExpandedTypeahead.tsx

@@ -13,11 +13,12 @@ export type ExpandedTypeaheadProps = Omit<
   'nullable' | 'onSelect'
 > &
   RequireAllOrNone<{
-    clearOnSelect?: boolean;
-    filterOption: (option: TypeaheadOption) => boolean;
     onSuggestionClick: (option: TypeaheadOption) => void;
     suggestedCount: number;
+    suggestedOptions: Array<TypeaheadOption>;
   }> & {
+    clearOnSelect?: boolean;
+    filterOption?: (option: TypeaheadOption) => boolean;
     onChange?: unknown; // Workaround: This prop is here just to absorb the onChange returned react-hook-form
     onSelect: (option: TypeaheadOption) => void;
   };
@@ -25,6 +26,7 @@ export type ExpandedTypeaheadProps = Omit<
 export default function ExpandedTypeahead({
   suggestedCount = 0,
   onSuggestionClick,
+  suggestedOptions = [],
   filterOption = () => true,
   clearOnSelect = false,
   options,
@@ -37,21 +39,22 @@ export default function ExpandedTypeahead({
     return options.filter(filterOption);
   }, [options, filterOption]);
   const suggestions = useMemo(
-    () => filteredOptions.slice(0, suggestedCount),
-    [filteredOptions, suggestedCount],
+    () => suggestedOptions.slice(0, suggestedCount),
+    [suggestedOptions, suggestedCount],
   );
 
   return (
-    <div className="flex flex-wrap gap-x-2">
+    <div className="flex flex-wrap gap-2">
       {suggestions.map((suggestion) => (
-        <Button
-          key={suggestion.id}
-          label={suggestion.label}
-          variant="tertiary"
-          onClick={() => {
-            onSuggestionClick?.(suggestion);
-          }}
-        />
+        <div key={suggestion.id} className="hidden lg:block">
+          <Button
+            label={suggestion.label}
+            variant="tertiary"
+            onClick={() => {
+              onSuggestionClick?.(suggestion);
+            }}
+          />
+        </div>
       ))}
       <div className="flex-1">
         <Typeahead

apps/portal/src/components/questions/typeahead/LocationTypeahead.tsx

@@ -16,14 +16,8 @@ export type LocationTypeaheadProps = Omit<
   onSuggestionClick?: (option: Location) => void;
 };
 
-export default function LocationTypeahead({
-  onSelect,
-  onSuggestionClick,
-  ...restProps
-}: LocationTypeaheadProps) {
-  const [query, setQuery] = useState('');
-
-  const { data: locations, isLoading } = trpc.useQuery([
+export function useLocationOptions(query: string) {
+  const { data: locations, ...restQuery } = trpc.useQuery([
     'locations.cities.list',
     {
       name: query,
@@ -43,6 +37,21 @@ export default function LocationTypeahead({
     );
   }, [locations]);
 
+  return {
+    data: locationOptions,
+    ...restQuery,
+  };
+}
+
+export default function LocationTypeahead({
+  onSelect,
+  onSuggestionClick,
+  ...restProps
+}: LocationTypeaheadProps) {
+  const [query, setQuery] = useState('');
+
+  const { data: locationOptions, isLoading } = useLocationOptions(query);
+
   return (
     <ExpandedTypeahead
       isLoading={isLoading}

apps/portal/src/components/shared/CitiesTypeahead.tsx

@@ -22,6 +22,25 @@ type Props = BaseProps &
     value?: TypeaheadOption | null;
   }>;
 
+export function useCityOptions(query: string) {
+  const { data, ...restQuery } = trpc.useQuery([
+    'locations.cities.list',
+    {
+      name: query,
+    },
+  ]);
+
+  return {
+    data:
+      data?.map(({ id, name, state }) => ({
+        id,
+        label: `${name}, ${state?.name}, ${state?.country?.name}`,
+        value: id,
+      })) ?? [],
+    ...restQuery,
+  };
+}
+
 export default function CitiesTypeahead({
   label = 'City',
   onSelect,
@@ -29,14 +48,8 @@ export default function CitiesTypeahead({
   ...props
 }: Props) {
   const [query, setQuery] = useState('');
-  const cities = trpc.useQuery([
-    'locations.cities.list',
-    {
-      name: query,
-    },
-  ]);
 
-  const { data, isLoading } = cities;
+  const { data: cityOptions, isLoading } = useCityOptions(query);
 
   return (
     <Typeahead
@@ -45,13 +58,7 @@ export default function CitiesTypeahead({
       minQueryLength={3}
       noResultsMessage="No cities found"
       nullable={true}
-      options={
-        data?.map(({ id, name, state }) => ({
-          id,
-          label: `${name}, ${state?.name}, ${state?.country?.name}`,
-          value: id,
-        })) ?? []
-      }
+      options={cityOptions}
       value={value}
       onQueryChange={setQuery}
       onSelect={onSelect}

apps/portal/src/components/shared/CompaniesTypeahead.tsx

@@ -21,12 +21,7 @@ type Props = BaseProps &
     value?: TypeaheadOption | null;
   }>;
 
-export default function CompaniesTypeahead({
-  onSelect,
-  value,
-  ...props
-}: Props) {
-  const [query, setQuery] = useState('');
+export function useCompanyOptions(query: string) {
   const companies = trpc.useQuery([
     'companies.list',
     {
@@ -34,7 +29,27 @@ export default function CompaniesTypeahead({
     },
   ]);
 
-  const { data, isLoading } = companies;
+  const { data, ...restQuery } = companies;
+
+  return {
+    data:
+      data?.map(({ id, name }) => ({
+        id,
+        label: name,
+        value: id,
+      })) ?? [],
+    ...restQuery,
+  };
+}
+
+export default function CompaniesTypeahead({
+  onSelect,
+  value,
+  ...props
+}: Props) {
+  const [query, setQuery] = useState('');
+
+  const { data: companyOptions, isLoading } = useCompanyOptions(query);
 
   return (
     <Typeahead
@@ -42,13 +57,7 @@ export default function CompaniesTypeahead({
       label="Company"
       noResultsMessage="No companies found"
       nullable={true}
-      options={
-        data?.map(({ id, name }) => ({
-          id,
-          label: name,
-          value: id,
-        })) ?? []
-      }
+      options={companyOptions}
       value={value}
       onQueryChange={setQuery}
       onSelect={onSelect}

apps/portal/src/components/shared/CountriesTypeahead.tsx

@@ -1,5 +1,6 @@
 import type { ComponentProps } from 'react';
 import { useState } from 'react';
+import type { Country } from '@prisma/client';
 import type { TypeaheadOption } from '@tih/ui';
 import { Typeahead } from '@tih/ui';
 
@@ -33,6 +34,48 @@ function stringPositionComparator(a: string, b: string, query: string): number {
   );
 }
 
+export function useCompareCountry(query: string) {
+  return (a: Country, b: Country) => {
+    const normalizedQueryString = query.trim().toLocaleLowerCase();
+    if (
+      a.code.toLocaleLowerCase() === normalizedQueryString ||
+      b.code.toLocaleLowerCase() === normalizedQueryString
+    ) {
+      return stringPositionComparator(a.code, b.code, normalizedQueryString);
+    }
+
+    return stringPositionComparator(a.name, b.name, normalizedQueryString);
+  };
+}
+
+export function useCountryOptions(query: string) {
+  const countries = trpc.useQuery([
+    'locations.countries.list',
+    {
+      name: query,
+    },
+  ]);
+
+  const { data, ...restQuery } = countries;
+
+  const compareCountry = useCompareCountry(query);
+
+  const countryOptions = (data ?? [])
+    // Client-side sorting by position of query string appearing
+    // in the country name since we can't do that in Prisma.
+    .sort(compareCountry)
+    .map(({ id, name }) => ({
+      id,
+      label: name,
+      value: id,
+    }));
+
+  return {
+    ...restQuery,
+    data: countryOptions,
+  };
+}
+
 export default function CountriesTypeahead({
   excludedValues,
   label = 'Country',
@@ -41,14 +84,7 @@ export default function CountriesTypeahead({
   ...props
 }: Props) {
   const [query, setQuery] = useState('');
-  const countries = trpc.useQuery([
-    'locations.countries.list',
-    {
-      name: query,
-    },
-  ]);
-
-  const { data, isLoading } = countries;
+  const { data: countryOptions, isLoading } = useCountryOptions(query);
 
   return (
     <Typeahead
@@ -56,34 +92,9 @@ export default function CountriesTypeahead({
       label={label}
       noResultsMessage="No countries found"
       nullable={true}
-      options={(data ?? [])
-        // Client-side sorting by position of query string appearing
-        // in the country name since we can't do that in Prisma.
-        .sort((a, b) => {
-          const normalizedQueryString = query.trim().toLocaleLowerCase();
-          if (
-            a.code.toLocaleLowerCase() === normalizedQueryString ||
-            b.code.toLocaleLowerCase() === normalizedQueryString
-          ) {
-            return stringPositionComparator(
-              a.code,
-              b.code,
-              normalizedQueryString,
-            );
-          }
-
-          return stringPositionComparator(
-            a.name,
-            b.name,
-            normalizedQueryString,
-          );
-        })
-        .map(({ id, name }) => ({
-          id,
-          label: name,
-          value: id,
-        }))
-        .filter((option) => !excludedValues?.has(option.value))}
+      options={countryOptions.filter(
+        (option) => !excludedValues?.has(option.value),
+      )}
       value={value}
       onQueryChange={setQuery}
       onSelect={onSelect}

apps/portal/src/components/shared/JobTitles.ts

@@ -78,5 +78,5 @@ export const JobTitleLabels: JobTitleData = {
 export type JobTitleType = keyof typeof JobTitleLabels;
 
 export function getLabelForJobTitleType(jobTitle: JobTitleType): string {
-  return JobTitleLabels[jobTitle].label;
+  return JobTitleLabels[jobTitle]?.label ?? '';
 }

apps/portal/src/components/shared/JobTitlesTypeahead.tsx

@@ -24,6 +24,23 @@ type Props = BaseProps &
     value?: TypeaheadOption | null;
   }>;
 
+const sortedJobTitleOptions = Object.entries(JobTitleLabels)
+  .map(([slug, { label, ranking }]) => ({
+    id: slug,
+    label,
+    ranking,
+    value: slug,
+  }))
+  .sort((a, b) => b.ranking - a.ranking);
+
+export function useJobTitleOptions(query: string) {
+  const jobTitles = sortedJobTitleOptions.filter(({ label }) =>
+    label.toLocaleLowerCase().includes(query.trim().toLocaleLowerCase()),
+  );
+
+  return jobTitles;
+}
+
 export default function JobTitlesTypeahead({
   excludedValues,
   label: labelProp = 'Job Title',
@@ -33,18 +50,10 @@ export default function JobTitlesTypeahead({
   ...props
 }: Props) {
   const [query, setQuery] = useState('');
-  const options = Object.entries(JobTitleLabels)
-    .map(([slug, { label, ranking }]) => ({
-      id: slug,
-      label,
-      ranking,
-      value: slug,
-    }))
-    .filter(({ label }) =>
-      label.toLocaleLowerCase().includes(query.trim().toLocaleLowerCase()),
-    )
-    .filter((option) => !excludedValues?.has(option.value))
-    .sort((a, b) => b.ranking - a.ranking);
+  const jobTitleOptions = useJobTitleOptions(query);
+  const options = jobTitleOptions.filter(
+    (option) => !excludedValues?.has(option.value),
+  );
 
   return (
     <Typeahead