feat: extend line filtering to support nfk

orgs/atb.json

@@ -1,7 +1,7 @@
 {
   "$schema": "./schema-validation.json",
   "orgId": "atb",
-  "orgLineIdPrefix": "ATB:Line:2_",
+  "authorityId": "ATB:Authority:2",
 
   "supportEmail": "kundeservice@atb.no",
 

orgs/fram.json

@@ -2,7 +2,7 @@
   "$schema": "./schema-validation.json",
 
   "orgId": "fram",
-  "orgLineIdPrefix": "MOR:Line:",
+  "authorityId": "MOR:Authority:MOR",
 
   "supportEmail": "fram@mrfylke.no",
 

orgs/nfk.json

@@ -1,7 +1,7 @@
 {
   "$schema": "./schema-validation.json",
   "orgId": "nfk",
-  "orgLineIdPrefix": "NOR:Line:12_",
+  "authorityId": "NOR:Authority:12",
 
   "supportEmail": "sentrumsterminalen@boreal.no",
 

orgs/schema-validation.json

@@ -28,7 +28,7 @@
           "enum": ["atb", "fram", "nfk", "troms"],
           "type": "string"
         },
-        "orgLineIdPrefix": {
+        "authorityId": {
           "type": "string"
         },
         "$schema": {

orgs/troms.json

@@ -1,7 +1,7 @@
 {
   "$schema": "./schema-validation.json",
   "orgId": "troms",
-  "orgLineIdPrefix": "TRO:Line:1_",
+  "authorityId": "TRO:Authority:1",
 
   "supportEmail": "",
 

src/modules/org-data/index.tsx

@@ -17,7 +17,7 @@ export type TranslatableUrl = { default: string } & Partial<{
 }>;
 export type OrgData = {
   orgId: WEBSHOP_ORGS;
-  orgLineIdPrefix: string;
+  authorityId: string;
   supportEmail: string;
 
   fylkeskommune?: {

src/page-modules/assistant/__tests__/assistant.test.tsx

@@ -69,6 +69,9 @@ describe('assistant page', function () {
       async singleTrip() {
         return {} as any;
       },
+      async lines() {
+        return {} as any;
+      },
       client: null as any,
     };
 
@@ -312,6 +315,9 @@ describe('assistant page', function () {
       async singleTrip() {
         return {} as any;
       },
+      async lines() {
+        return {} as any;
+      },
       client: null as any,
     };
 

src/page-modules/assistant/client/journey-planner/index.ts

@@ -10,12 +10,14 @@ import useSWRInfinite from 'swr/infinite';
 import { createTripQuery, tripQueryToQueryString } from '../../utils';
 import { useEffect, useState } from 'react';
 import { formatLocalTimeToCET } from '@atb/utils/date';
+import { LineData } from '../../server/journey-planner/validators';
 
 const MAX_NUMBER_OF_INITIAL_SEARCH_ATTEMPTS = 3;
 const INITIAL_NUMBER_OF_WANTED_TRIP_PATTERNS = 6;
 
 export type TripApiReturnType = TripData;
 export type NonTransitTripApiReturnType = NonTransitTripData;
+export type LinesApiReturnType = LineData;
 
 function createKeyGetterOfQuery(query: TripQuery) {
   return (pageIndex: number, previousPageData: TripApiReturnType) => {

src/page-modules/assistant/line-filter/index.tsx

@@ -1,39 +1,61 @@
 import { Typo } from '@atb/components/typography';
 import { PageText, useTranslation } from '@atb/translations';
-import { ChangeEvent, useState } from 'react';
+import { ChangeEvent, useEffect, useState } from 'react';
 import style from './line-filter.module.css';
 import { getOrgData } from '@atb/modules/org-data';
 import LabeledInput from '@atb/components/labled-input';
+import useSWR from 'swr';
 type LineFilterProps = {
   filterState: string[] | null;
   onChange: (lineFilter: string[] | null) => void;
 };
 
+const fetcher = (url: string) => fetch(url).then((res) => res.json());
+
 export default function LineFilter({ filterState, onChange }: LineFilterProps) {
   const { t } = useTranslation();
 
-  const { orgLineIdPrefix } = getOrgData();
+  const { authorityId } = getOrgData();
 
-  const [localFilterState, setLocalFilterState] = useState(
-    filterState?.map((line) => line.replace(orgLineIdPrefix, '')).join(', ') ??
-      '',
+  const { data, error, isLoading } = useSWR(
+    `api/assistant/lines/${authorityId}`,
+    fetcher,
   );
 
+  const [localFilterState, setLocalFilterState] = useState('');
   const onChangeWrapper = (event: ChangeEvent<HTMLInputElement>) => {
+    if (error || isLoading) return;
     const lineFilter = event.target.value;
     setLocalFilterState(lineFilter);
-
     if (!lineFilter) {
       onChange(null);
     } else {
-      const linesWithPrefix = lineFilter
+      const publicCodes = lineFilter
         .split(',')
-        .map((line) => `${orgLineIdPrefix}${line.trim()}`);
+        .map((publicCode) => `${publicCode.trim()}`);
 
-      onChange(linesWithPrefix);
+      const lines = publicCodes
+        .map((publicCode) => data.publicCodeLineList[publicCode])
+        .filter((line) => line !== undefined);
+
+      onChange(lines);
     }
   };
 
+  useEffect(() => {
+    const mapFromLineIdToPublicCode = () => {
+      if (!data) return;
+      const lines =
+        filterState
+          ?.map((line) => data.linePublicCodeList[line])
+          .filter((line, index, lines) => lines.indexOf(line) === index)
+          .join(', ') ?? '';
+      return lines;
+    };
+    const lines = mapFromLineIdToPublicCode();
+    if (lines) setLocalFilterState(lines);
+  }, [data]);
+
   return (
     <div className={style.container}>
       <Typo.h3 textType="body__primary--bold" className={style.heading}>

src/page-modules/assistant/server/journey-planner/index.ts

@@ -13,6 +13,7 @@ import {
   TripsQueryVariables,
 } from './journey-gql/trip.generated';
 import {
+  LineData,
   TripData,
   TripPatternWithDetails,
   nonTransitSchema,
@@ -21,6 +22,7 @@ import {
 } from './validators';
 import type {
   FromToTripQuery,
+  LineInput,
   NonTransitData,
   NonTransitTripData,
   NonTransitTripInput,
@@ -59,6 +61,12 @@ import {
   addAssistantTripToCache,
   getAssistantTripIfCached,
 } from '../trip-cache';
+import {
+  LineFragment,
+  LinesDocument,
+  LinesQuery,
+  LinesQueryVariables,
+} from './journey-gql/lines.generated';
 
 const DEFAULT_JOURNEY_CONFIG = {
   numTripPatterns: 8, // The maximum number of trip patterns to return.
@@ -73,6 +81,7 @@ export type JourneyPlannerApi = {
   trip(input: TripInput): Promise<TripData>;
   nonTransitTrips(input: NonTransitTripInput): Promise<NonTransitTripData>;
   singleTrip(input: string): Promise<TripPatternWithDetails | undefined>;
+  lines(input: LineInput): Promise<LineData>;
 };
 
 export function createJourneyApi(
@@ -156,6 +165,20 @@ export function createJourneyApi(
       return nonTransits;
     },
 
+    async lines(input) {
+      const result = await client.query<LinesQuery, LinesQueryVariables>({
+        query: LinesDocument,
+        variables: { authorities: input.authorities },
+      });
+
+      if (result.error || result.errors) {
+        throw result.error || result.errors;
+      }
+
+      const publicCodeLineList = createPublicCodeLineList(result.data.lines);
+      const linePublicCodeList = createLinePublicCodeList(result.data.lines);
+      return { publicCodeLineList, linePublicCodeList };
+    },
     async trip(input) {
       const journeyModes = {
         accessMode: StreetMode.Foot,
@@ -784,3 +807,25 @@ async function getSortedViaTrips(
   };
   return sortedData;
 }
+
+function createPublicCodeLineList(lines: LineFragment[]) {
+  const publicCodeLineList: Record<string, string[]> = {};
+  lines.forEach((line) => {
+    if (line.publicCode) {
+      if (publicCodeLineList.hasOwnProperty(line.publicCode)) {
+        publicCodeLineList[line.publicCode].push(line.id);
+      } else {
+        publicCodeLineList[line.publicCode] = [line.id];
+      }
+    }
+  });
+  return publicCodeLineList;
+}
+
+function createLinePublicCodeList(lines: LineFragment[]) {
+  const linePublicCodeList: Record<string, string> = {};
+  lines.forEach((line) => {
+    if (line.publicCode) linePublicCodeList[line.id] = line.publicCode;
+  });
+  return linePublicCodeList;
+}

src/page-modules/assistant/server/journey-planner/journey-gql/lines.gql

@@ -0,0 +1,10 @@
+query Lines($authorities: [String]) {
+  lines(authorities: $authorities) {
+    ...line
+  }
+}
+
+fragment line on Line {
+  id
+  publicCode
+}

src/page-modules/assistant/server/journey-planner/validators.ts

@@ -146,6 +146,11 @@ export const tripPatternWithDetailsSchema = z.object({
   ),
 });
 
+export const linesSchema = z.object({
+  publicCodeLineList: z.record(z.array(z.string())),
+  linePublicCodeList: z.record(z.string()),
+});
+
 export type TripData = z.infer<typeof tripSchema>;
 export type NonTransitData = z.infer<typeof nonTransitSchema>;
 export type TripPattern = z.infer<typeof tripPatternSchema>;
@@ -154,3 +159,4 @@ export type Quay = z.infer<typeof quaySchema>;
 export type TripPatternWithDetails = z.infer<
   typeof tripPatternWithDetailsSchema
 >;
+export type LineData = z.infer<typeof linesSchema>;

src/page-modules/assistant/types.ts

@@ -82,3 +82,7 @@ export type NonTransitTripData = {
   footTrip?: NonTransitData;
   bikeRentalTrip?: NonTransitData;
 };
+
+export type LineInput = {
+  authorities: string[];
+};

src/pages/api/assistant/lines/[authorityId].ts

@@ -0,0 +1,22 @@
+import { tryResult } from '@atb/modules/api-server';
+import { LinesApiReturnType } from '@atb/page-modules/assistant/client';
+import { handlerWithAssistantClient } from '@atb/page-modules/assistant/server';
+
+export default handlerWithAssistantClient<LinesApiReturnType>({
+  async GET(req, res, { client, ok }) {
+    const { authorityId } = req.query;
+    const authorities: string[] = [];
+
+    if (typeof authorityId === 'string') {
+      authorities.push(authorityId);
+    }
+
+    return tryResult(req, res, async () => {
+      return ok(
+        await client.lines({
+          authorities: authorities,
+        }),
+      );
+    });
+  },
+});