coomer-downloader 3.1.0 → 3.2.0

package/src/utils/file.ts

@@ -0,0 +1,75 @@
+import os from 'node:os';
+import path from 'node:path';
+import type { MediaType } from '../types';
+import { filterString, testMediaType } from './filters';
+
+interface ICoomerFile {
+  name: string;
+  url: string;
+  filepath?: string;
+  size?: number;
+  downloaded?: number;
+  content?: string;
+}
+
+export class CoomerFile {
+  public state: 'downloading' | 'pause' = 'pause';
+
+  constructor(
+    public name: string,
+    public url: string,
+    public filepath?: string,
+    public size?: number,
+    public downloaded?: number,
+    public content?: string,
+  ) {}
+
+  get textContent() {
+    const text = `${this.name || ''} ${this.content || ''}`.toLowerCase();
+    return text;
+  }
+
+  public static from(f: ICoomerFile) {
+    return new CoomerFile(f.name, f.url, f.filepath, f.size, f.downloaded, f.content);
+  }
+}
+
+export class CoomerFileList {
+  public dirPath?: string;
+  public dirName?: string;
+
+  constructor(public files: CoomerFile[] = []) {}
+
+  public setDirPath(dir: string, dirName?: string) {
+    dirName = dirName || (this.dirName as string);
+
+    if (dir === './') {
+      this.dirPath = path.resolve(dir, dirName);
+    } else {
+      this.dirPath = path.join(os.homedir(), path.join(dir, dirName));
+    }
+
+    this.files.forEach((file) => {
+      file.filepath = path.join(this.dirPath as string, file.name);
+    });
+
+    return this;
+  }
+
+  public filterByText(include: string, exclude: string) {
+    this.files = this.files.filter((f) => filterString(f.textContent, include, exclude));
+    return this;
+  }
+
+  public filterByMediaType(media?: string) {
+    if (media) {
+      this.files = this.files.filter((f) => testMediaType(f.name, media as MediaType));
+    }
+    return this;
+  }
+
+  public skip(n: number) {
+    this.files = this.files.slice(n);
+    return this;
+  }
+}

package/src/utils/filters.ts

@@ -1,12 +1,16 @@
-import type { File, MediaType } from '../types';
+import type { MediaType } from '../types';
 
-export const isImage = (name: string) => /\.(jpg|jpeg|png|gif|bmp|tiff|webp|avif)$/i.test(name);
+export function isImage(name: string) {
+  return /\.(jpg|jpeg|png|gif|bmp|tiff|webp|avif)$/i.test(name);
+}
 
-export const isVideo = (name: string) =>
-  /\.(mp4|m4v|avi|mov|mkv|webm|flv|wmv|mpeg|mpg|3gp)$/i.test(name);
+export function isVideo(name: string) {
+  return /\.(mp4|m4v|avi|mov|mkv|webm|flv|wmv|mpeg|mpg|3gp)$/i.test(name);
+}
 
-export const testMediaType = (name: string, type: MediaType) =>
-  type === 'all' ? true : type === 'image' ? isImage(name) : isVideo(name);
+export function testMediaType(name: string, type: MediaType) {
+  return type === 'all' ? true : type === 'image' ? isImage(name) : isVideo(name);
+}
 
 function includesAllWords(str: string, words: string[]) {
   if (!words.length) return true;
@@ -25,13 +29,6 @@ function parseQuery(query: string) {
     .filter((_) => _);
 }
 
-function filterString(text: string, include: string, exclude: string): boolean {
+export function filterString(text: string, include: string, exclude: string): boolean {
   return includesAllWords(text, parseQuery(include)) && includesNoWords(text, parseQuery(exclude));
 }
-
-export function filterKeywords(files: File[], include: string, exclude: string) {
-  return files.filter((f) => {
-    const text = `${f.name || ''} ${f.content || ''}`.toLowerCase();
-    return filterString(text, include, exclude);
-  });
-}

package/src/utils/index.ts

@@ -1,6 +1,6 @@
-export { downloadFiles } from './downloader';
-export { getFileSize, mkdir } from './files';
-export { filterKeywords, isImage, isVideo, testMediaType } from './filters';
+export { Downloader } from './downloader';
+export { isImage, isVideo, testMediaType } from './filters';
+export { getFileSize, mkdir } from './io';
 export { createMultibar } from './multibar';
 export {
   fetchByteRange,

package/src/utils/multibar.ts

@@ -1,5 +1,5 @@
 import { MultiBar, type Options, type SingleBar } from 'cli-progress';
-import { subject } from './downloader';
+import type { Downloader } from './downloader';
 import { b2mb, formatNameStdout } from './strings';
 
 const config: Options = {
@@ -10,14 +10,14 @@ const config: Options = {
   format: '{percentage}% | {filename} | {value}/{total}{size}',
 };
 
-export function createMultibar() {
+export function createMultibar(downloader: Downloader) {
   const multibar = new MultiBar(config);
   let bar: SingleBar;
   let minibar: SingleBar;
   let filename: string;
   let index = 0;
 
-  subject.subscribe({
+  downloader.subject.subscribe({
     next: ({ type, filesCount, file }) => {
       switch (type) {
         case 'FILES_DOWNLOADING_START':

package/src/utils/strings.ts

@@ -2,15 +2,6 @@ export function b2mb(bytes: number) {
   return Number.parseFloat((bytes / 1048576).toFixed(2));
 }
 
-export function sanitizeString(str: string) {
-  return (
-    str
-      .match(/(\w| |-)/g)
-      ?.join('')
-      .replace(/ +/g, ' ') || ''
-  );
-}
-
 export function formatNameStdout(pathname: string) {
   const name = pathname.split('/').pop() || '';
   const consoleWidth = process.stdout.columns;