10 changed files with 64 additions and 208 deletions
Split View
Diff Options
-
+2 -0services/seeker/package.json
-
+0 -23services/seeker/src/Application.ts
-
+6 -2services/seeker/src/index.ts
-
+4 -5services/seeker/src/services/IpcInterface.ts
-
+0 -136services/seeker/src/services/MovieSearch.ts
-
+5 -6services/seeker/src/services/PostProcessor/PostProcessor.ts
-
+28 -25services/seeker/src/services/Supervisor.ts
-
+1 -2services/seeker/src/services/TorrentIpc.ts
-
+17 -7services/seeker/src/services/TorrentManager.ts
-
+1 -2services/seeker/src/services/index.ts
+ 2
- 0
services/seeker/package.json
View File
+ 0
- 23
services/seeker/src/Application.ts
View File
| @ -1,23 +0,0 @@ | |||
| import * as services from "./services"; | |||
| import { Microservice } from "@autoplex/microservice"; | |||
| /** | |||
| * The main application class | |||
| */ | |||
| export default class Application extends Microservice | |||
| { | |||
| /** | |||
| * Return the current application instance | |||
| */ | |||
| public static instance() { return super.instance() } | |||
| /** | |||
| * Create a new application instance | |||
| */ | |||
| public constructor() { | |||
| super(); | |||
| for (let ServiceClass of Object.values(services)) { | |||
| this.installService(ServiceClass); | |||
| } | |||
| } | |||
| } | |||
+ 6
- 2
services/seeker/src/index.ts
View File
| @ -1,7 +1,11 @@ | |||
| import Application from "./Application"; | |||
| import { Microservice } from "@autoplex/microservice"; | |||
| import * as services from "./services"; | |||
| // Create a new application instance | |||
| let app = new Application(); | |||
| let app = new Microservice(); | |||
| // Install the internal services | |||
| app.installServices(Object.values(services)); | |||
| // Start the application | |||
| app.exec().then(process.exit); | |||
+ 4
- 5
services/seeker/src/services/IpcInterface.ts
View File
+ 0
- 136
services/seeker/src/services/MovieSearch.ts
View File
| @ -1,136 +0,0 @@ | |||
| import Application from "../Application"; | |||
| import { MovieTicket } from "@autoplex/database"; | |||
| import * as providerClasses from "../torrents"; | |||
| import Provider, { MediaType } from "../torrents/providers/Provider"; | |||
| import Torrent from "../torrents/Torrent"; | |||
| import { rankTorrents } from "../torrents/ranking"; | |||
| import { InternalService } from "@autoplex/microservice"; | |||
| import Supervisor from "./Supervisor"; | |||
| export default class MovieSearch extends InternalService | |||
| { | |||
| /** | |||
| * The queue of current movie tickts requiring torrents | |||
| */ | |||
| protected movieQueue!: MovieTicket[]; | |||
| /** | |||
| * Available providers that support movies | |||
| */ | |||
| protected providers!: Provider[]; | |||
| /** | |||
| * Indicate if the service is currently searching for movies in the queue | |||
| */ | |||
| protected isSearchingForMovies: boolean; | |||
| /** | |||
| * Store a reference to the supervisor service | |||
| */ | |||
| protected supervisor!: Supervisor; | |||
| /** | |||
| * Create a new instance of the movie search service | |||
| */ | |||
| public constructor(app: Application) { | |||
| super(app); | |||
| this.movieQueue = []; | |||
| this.isSearchingForMovies = false; | |||
| } | |||
| /** | |||
| * The service name | |||
| */ | |||
| public readonly NAME = "Movie Search"; | |||
| /** | |||
| * Boot the movie search service | |||
| */ | |||
| public async boot() { | |||
| let providers = Object.values(providerClasses); | |||
| this.providers = providers.filter(provider => provider.PROVIDES & MediaType.Movies) | |||
| .map(ProviderClass => new ProviderClass()); | |||
| } | |||
| /** | |||
| * Connect to the supervisor instance when everything is booted | |||
| */ | |||
| public start() { | |||
| this.supervisor = this.app.service<Supervisor>("Supervisor"); | |||
| } | |||
| /** | |||
| * Shutdown the service | |||
| */ | |||
| public async shutdown() { | |||
| } | |||
| // Interface Methods --------------------------------------------------------------------------- | |||
| /** | |||
| * Enqueue a movie to the search queue | |||
| */ | |||
| public enqueueMovie(movie: MovieTicket) { | |||
| this.movieQueue.push(movie); | |||
| this.searchMovies(); | |||
| } | |||
| // Movie Searching ----------------------------------------------------------------------------- | |||
| /** | |||
| * Search for movies currently in the queue | |||
| */ | |||
| protected async searchMovies() { | |||
| if (this.isSearchingForMovies) { | |||
| return; | |||
| } | |||
| this.isSearchingForMovies = true; | |||
| while (this.movieQueue.length > 0) { | |||
| let movie = this.movieQueue.splice(0, 1)[0]; | |||
| let link = await this.searchMovie(movie); | |||
| if (link === null) { | |||
| continue; | |||
| } | |||
| this.supervisor.onMovieTorrentFound(movie, link); | |||
| } | |||
| this.isSearchingForMovies = false; | |||
| } | |||
| /** | |||
| * Search for a movie | |||
| */ | |||
| protected async searchMovie(movie: MovieTicket) { | |||
| // Search by IMDb | |||
| let torrents = await this.searchImdb(movie); | |||
| if (torrents.length == 0) { | |||
| return null; | |||
| } | |||
| // Determine the preferred torrents | |||
| let preferredTorrents = rankTorrents(torrents); | |||
| if (preferredTorrents.length == 0) { | |||
| preferredTorrents = torrents; | |||
| } | |||
| // Return the selected torrent | |||
| this.log("Found movie torrent for", movie.title); | |||
| return await preferredTorrents[0].downloadLink(); | |||
| } | |||
| /** | |||
| * Search for a movie by its IMDb ID | |||
| */ | |||
| protected async searchImdb(movie: MovieTicket): Promise<Torrent[]> { | |||
| if (movie.imdbId == null) { | |||
| return []; | |||
| } | |||
| let results = await Promise.all(this.providers.map(provider => provider.searchMovie(movie))); | |||
| return (<Torrent[]>[]).concat(...results); | |||
| } | |||
| /** | |||
| * Pick the best torrent from the list | |||
| */ | |||
| protected pickBestTorrent() { | |||
| } | |||
| } | |||