diff --git a/lesson_10/libraries/src/loaders/loaders.module.ts b/lesson_10/libraries/src/loaders/loaders.module.ts index abf6a6b70..47964c087 100644 --- a/lesson_10/libraries/src/loaders/loaders.module.ts +++ b/lesson_10/libraries/src/loaders/loaders.module.ts @@ -7,6 +7,7 @@ import { JamesCapparellLoader } from './james_capparell_loader.js'; import { JosephCaballeroLoader } from './joseph_caballero_loader.js'; import { NileJacksonLoader } from './nile_jackson_loader.js'; import { PabloLimonParedesLoader } from './pablo_limon_paredes_loader.js'; +import { TommyTranLoader } from './tommy_tran_loader.js'; import { XavierCruzLoader } from './xavier_cruz_loader.js'; import { ZionBuchananLoader } from './zion_buchanan_loader.js'; @@ -22,6 +23,7 @@ const LOADER_PROVIDERS = [ JosephCaballeroLoader, NileJacksonLoader, PabloLimonParedesLoader, + TommyTranLoader, XavierCruzLoader, ZionBuchananLoader, ]; diff --git a/lesson_10/libraries/src/loaders/tommy_tran_loader.ts b/lesson_10/libraries/src/loaders/tommy_tran_loader.ts new file mode 100644 index 000000000..593107d33 --- /dev/null +++ b/lesson_10/libraries/src/loaders/tommy_tran_loader.ts @@ -0,0 +1,45 @@ +import csv from 'csv-parser'; +import fs from 'fs'; +import { Credit, MediaItem } from '../models/index.js'; +import { Loader } from './loader.js'; + +export class TommyTranLoader implements Loader { + getLoaderName(): string { + return 'tommytran'; + } + + 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 { + const mediaitems: MediaItem[] = []; + 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, [])); // help from xavier to instatiate the class + } + 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: mediaItemId, role, name } = row; + credits.push(new Credit(mediaItemId, name, role)); + } + return credits; + } +}