Skip to content
86 changes: 86 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,86 @@
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();

const creditsMap: Record<string, Credit[]> = {};

credits.forEach((credit) => {
const mediaItemId = credit.getMediaItemId();
if (!creditsMap[mediaItemId]) {
creditsMap[mediaItemId] = [];
}
creditsMap[mediaItemId].push(credit);
});

mediaItems.forEach((mediaItem) => {
const mediaItemId = mediaItem.getId();
const associatedCredits = creditsMap[mediaItemId] || [];

associatedCredits.forEach((credit) => mediaItem.addCredit(credit));
});

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;
}
}
2 changes: 2 additions & 0 deletions lesson_10/libraries/src/loaders/loaders.module.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import { Module } from '@nestjs/common';
import { AngelicaCastilloLoader } from './angelica_castillo_loader.js';
import { AnthonyMaysLoader } from './anthony_mays_loader.js';
import { ChelseaOgbonniaLoader } from './chelsea_ogbonnia_loader.js';
import { JamesCapparellLoader } from './james_capparell_loader.js';
import { NileJacksonLoader } from './nile_jackson_loader.js';
import { XavierCruzLoader } from './xavier_cruz_loader.js';
Expand All @@ -10,6 +11,7 @@ export const Loaders = Symbol.for('Loaders');
// Add your quiz provider here.
const LOADER_PROVIDERS = [
AnthonyMaysLoader,
ChelseaOgbonniaLoader,
AngelicaCastilloLoader,
JamesCapparellLoader,
NileJacksonLoader,
Expand Down