Skip to content
69 changes: 69 additions & 0 deletions lesson_10/libraries/src/loaders/chelsea_ogbonnia_loader.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
import csv from 'csv-parser';
import fs from 'fs';
import { Credit, MediaItem, MediaType } from '../models/index.js';
import { Loader } from './loader.js';

export class ChelseaOgbonniaLoader implements Loader {
getLoaderName(): string {
return 'chesleaogbonnia';
}

async loadData(): Promise<MediaItem[]> {
const credits = await this.loadCredits();
const mediaItems = await this.loadMediaItems();

console.log(
`Loaded ${credits.length} credits and ${mediaItems.length} media items`,
);

return mediaItems;
}

async loadMediaItems(): Promise<MediaItem[]> {
const mediaItems: MediaItem[] = [];
const readable = fs
.createReadStream('data/media_items.csv', 'utf-8')
.pipe(csv());

for await (const row of readable) {
const { id, type, title, year } = row;
let mediaType: MediaType | undefined;
switch (type) {
case 'movie':
mediaType = MediaType.Movie;
break;
case 'tv_show':
mediaType = MediaType.TVShow;
break;
case 'documentary':
mediaType = MediaType.Documentary;
break;
default:
throw new Error(`Unknown media type: ${type}`);
}
const releaseYear = Number(year);
const mediaItem = new MediaItem(
String(id),
title,
mediaType,
releaseYear,
[],
);
mediaItems.push(mediaItem);
}
return mediaItems;
}

async loadCredits(): Promise<Credit[]> {
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;
const credit = new Credit(mediaItemId, name, role);
credits.push(credit);
}
return credits;
}
}
6 changes: 3 additions & 3 deletions lesson_10/libraries/src/loaders/loaders.module.ts
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
import { Module } from '@nestjs/common';
import { AnthonyMaysLoader } from './anthony_mays_loader.js';
import { XavierCruzLoader } from './xavier_cruz_loader.js';
import { ChelseaOgbonniaLoader } from './chelsea_ogbonnia_loader.js';

export const Loaders = Symbol.for('Loaders');

// Add your quiz provider here.
const LOADER_PROVIDERS = [AnthonyMaysLoader, XavierCruzLoader];
const LOADER_PROVIDERS = [AnthonyMaysLoader, ChelseaOgbonniaLoader];

@Module({
providers: [
Expand All @@ -18,4 +18,4 @@ const LOADER_PROVIDERS = [AnthonyMaysLoader, XavierCruzLoader];
],
exports: [Loaders],
})
export class LoadersModule {}
export class LoadersModule {}