From 0bce1fb3fad6972dfaba72c6837c07016a473556 Mon Sep 17 00:00:00 2001 From: jbey251 Date: Fri, 28 Mar 2025 13:40:23 +0000 Subject: [PATCH] FEATURE: Target the media_items.csv file to load data. --- .../libraries/src/loaders/JBey_loader.ts | 46 +++++++++++++++++++ .../libraries/src/loaders/loaders.module.ts | 3 +- 2 files changed, 48 insertions(+), 1 deletion(-) create mode 100644 lesson_10/libraries/src/loaders/JBey_loader.ts diff --git a/lesson_10/libraries/src/loaders/JBey_loader.ts b/lesson_10/libraries/src/loaders/JBey_loader.ts new file mode 100644 index 000000000..4ee8ce17c --- /dev/null +++ b/lesson_10/libraries/src/loaders/JBey_loader.ts @@ -0,0 +1,46 @@ +import csv from 'csv-parser'; +import fs from 'fs'; +import { Credit, MediaItem } from '../models/index.js'; +import { Loader } from './loader.js'; + +export class JBeyLoader implements Loader { + getLoaderName(): string { + return 'JBey'; + } + + async loadData(): Promise { + const credits = await this.loadCredits(); + const mediaItems = await this.loadMediaItems(); + + console.log( + `Loaded ${credits.length} credits and ${mediaItems.length} media items`, + ); + + return [...mediaItems.values()]; + } + + async loadMediaItems(): Promise { + // TODO: Implement this method. + const mediaItems = []; + const readable = fs + .createReadStream('data/media_items.csv', 'utf-8') + .pipe(csv()); + for await (const row of readable) { + const { id, title, type, year } = row; + mediaItems.push(new MediaItem(id, title, type, year, [])); + } + return mediaItems; + } + + async loadCredits(): Promise { + const credits = []; + const readable = fs + .createReadStream('data/credits.csv', 'utf-8') + .pipe(csv()); + for await (const row of readable) { + const { media_item_id, role, name } = row; + credits.push(new Credit(media_item_id, name, role)); + } + return credits; + } +} diff --git a/lesson_10/libraries/src/loaders/loaders.module.ts b/lesson_10/libraries/src/loaders/loaders.module.ts index 22e32e2cd..a16da7c95 100644 --- a/lesson_10/libraries/src/loaders/loaders.module.ts +++ b/lesson_10/libraries/src/loaders/loaders.module.ts @@ -1,10 +1,11 @@ import { Module } from '@nestjs/common'; import { AnthonyMaysLoader } from './anthony_mays_loader.js'; +import { JBeyLoader } from './JBey_loader.js'; export const Loaders = Symbol.for('Loaders'); -const LOADER_PROVIDERS = [AnthonyMaysLoader]; +const LOADER_PROVIDERS = [AnthonyMaysLoader, JBeyLoader]; @Module({ providers: [