feat: extend files with offset field

isolomak · Aug 4, 2023 · 1c4df09 · 1c4df09
1 parent af815ac
commit 1c4df09
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 13 deletions.
diff --git a/src/parse/TorrentParser.ts b/src/parse/TorrentParser.ts
@@ -4,12 +4,6 @@ import { IDotTorrent, IDotTorrentFileInfo } from '../types';
 
 export default class TorrentParser {
 
-	/**
-	 * Constructor
-	 */
-	// eslint-disable-next-line @typescript-eslint/no-useless-constructor
-	constructor() {}
-
 	/**
 	 * Parse torrent file
 	 */
@@ -153,8 +147,19 @@ export default class TorrentParser {
 			}
 		}
 
-		return Array.from(fileMap.entries())
+		const files = Array.from(fileMap.entries())
 			.map(([ key, value ]) => ({ path: key, length: value }));
+
+		return files
+			.map((item, index) => {
+				let offset = 0;
+
+				files
+					.filter((_, internalIndex) => internalIndex < index)
+					.forEach(({ length }) => { offset += length; });
+
+				return { ...item, offset };
+			});
 	}
 
 	/**
@@ -163,7 +168,7 @@ export default class TorrentParser {
 	private static _name(decodedData: BencodeDictionary): string | null {
 		const torrentInfo = TorrentParser._info(decodedData);
 
-		if (!torrentInfo || !torrentInfo.name) {
+		if (!torrentInfo?.name) {
 			return null;
 		}
 
@@ -238,7 +243,7 @@ export default class TorrentParser {
 			return null;
 		}
 
-		return files.reduce((acc, next) => acc += next.length, 0);
+		return files.reduce((acc, next) => acc + next.length, 0);
 	}
 
 	/**

diff --git a/src/types.ts b/src/types.ts
@@ -26,6 +26,7 @@ export interface ICreateTorrentParams {
 export interface IDotTorrentFileInfo {
 	length: number;
 	path: string;
+	offset: number;
 }
 
 /**

diff --git a/test/parseFile.test.ts b/test/parseFile.test.ts
@@ -18,8 +18,8 @@ describe('Parse file tests', () => {
 			createdAt: 1585998070,
 			encoding: 'UTF-8',
 			files: [
-				{ length: 16, path: 'file_1.txt' },
-				{ length: 16, path: 'file_2.txt' }
+				{ length: 16, path: 'file_1.txt', offset: 0 },
+				{ length: 16, path: 'file_2.txt', offset: 16 }
 			],
 			infoHash: '46bf9dd414eae4cd74fa4d129d2f23553a21bca7',
 			name: 'test_torrent',
@@ -47,8 +47,8 @@ describe('Parse file tests', () => {
 			createdAt: 1585998070,
 			encoding: 'UTF-8',
 			files: [
-				{ length: 16, path: 'file_1.txt' },
-				{ length: 16, path: 'file_2.txt' }
+				{ length: 16, path: 'file_1.txt', offset: 0 },
+				{ length: 16, path: 'file_2.txt', offset: 16 }
 			],
 			infoHash: '46bf9dd414eae4cd74fa4d129d2f23553a21bca7',
 			name: 'test_torrent',