Skip to content

Commit

Permalink
Add PTP as a source
Browse files Browse the repository at this point in the history
  • Loading branch information
Morea committed Oct 15, 2023
1 parent c246ae4 commit 33e7c2b
Show file tree
Hide file tree
Showing 5 changed files with 140 additions and 13 deletions.
67 changes: 60 additions & 7 deletions Find Unique Titles/dist/find.unique.titles.user.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1828,18 +1828,54 @@ var __webpack_async_dependencies__ = __webpack_handle_async_dependencies__([_uti
_utils_cache__WEBPACK_IMPORTED_MODULE_0__ = (__webpack_async_dependencies__.then ? (await __webpack_async_dependencies__)() : __webpack_async_dependencies__)[0];
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
function _awaitAsyncGenerator(value) { return new _OverloadYield(value, 0); }
function _wrapAsyncGenerator(fn) { return function () { return new _AsyncGenerator(fn.apply(this, arguments)); }; }
function _AsyncGenerator(gen) { var front, back; function resume(key, arg) { try { var result = gen[key](arg), value = result.value, overloaded = value instanceof _OverloadYield; Promise.resolve(overloaded ? value.v : value).then(function (arg) { if (overloaded) { var nextKey = "return" === key ? "return" : "next"; if (!value.k || arg.done) return resume(nextKey, arg); arg = gen[nextKey](arg).value; } settle(result.done ? "return" : "normal", arg); }, function (err) { resume("throw", err); }); } catch (err) { settle("throw", err); } } function settle(type, value) { switch (type) { case "return": front.resolve({ value: value, done: !0 }); break; case "throw": front.reject(value); break; default: front.resolve({ value: value, done: !1 }); } (front = front.next) ? resume(front.key, front.arg) : back = null; } this._invoke = function (key, arg) { return new Promise(function (resolve, reject) { var request = { key: key, arg: arg, resolve: resolve, reject: reject, next: null }; back ? back = back.next = request : (front = back = request, resume(key, arg)); }); }, "function" != typeof gen.return && (this.return = void 0); }
_AsyncGenerator.prototype["function" == typeof Symbol && Symbol.asyncIterator || "@@asyncIterator"] = function () { return this; }, _AsyncGenerator.prototype.next = function (arg) { return this._invoke("next", arg); }, _AsyncGenerator.prototype.throw = function (arg) { return this._invoke("throw", arg); }, _AsyncGenerator.prototype.return = function (arg) { return this._invoke("return", arg); };
function _awaitAsyncGenerator(value) { return new _OverloadYield(value, 0); }
function _asyncGeneratorDelegate(inner) { var iter = {}, waiting = !1; function pump(key, value) { return waiting = !0, value = new Promise(function (resolve) { resolve(inner[key](value)); }), { done: !1, value: new _OverloadYield(value, 1) }; } return iter["undefined" != typeof Symbol && Symbol.iterator || "@@iterator"] = function () { return this; }, iter.next = function (value) { return waiting ? (waiting = !1, value) : pump("next", value); }, "function" == typeof inner.throw && (iter.throw = function (value) { if (waiting) throw waiting = !1, value; return pump("throw", value); }), "function" == typeof inner.return && (iter.return = function (value) { return waiting ? (waiting = !1, value) : pump("return", value); }), iter; }
function _OverloadYield(value, kind) { this.v = value, this.k = kind; }
function _asyncIterator(iterable) { var method, async, sync, retry = 2; for ("undefined" != typeof Symbol && (async = Symbol.asyncIterator, sync = Symbol.iterator); retry--;) { if (async && null != (method = iterable[async])) return method.call(iterable); if (sync && null != (method = iterable[sync])) return new AsyncFromSyncIterator(method.call(iterable)); async = "@@asyncIterator", sync = "@@iterator"; } throw new TypeError("Object is not async iterable"); }
function AsyncFromSyncIterator(s) { function AsyncFromSyncIteratorContinuation(r) { if (Object(r) !== r) return Promise.reject(new TypeError(r + " is not an object.")); var done = r.done; return Promise.resolve(r.value).then(function (value) { return { value: value, done: done }; }); } return AsyncFromSyncIterator = function AsyncFromSyncIterator(s) { this.s = s, this.n = s.next; }, AsyncFromSyncIterator.prototype = { s: null, n: null, next: function next() { return AsyncFromSyncIteratorContinuation(this.n.apply(this.s, arguments)); }, return: function _return(value) { var ret = this.s.return; return void 0 === ret ? Promise.resolve({ value: value, done: !0 }) : AsyncFromSyncIteratorContinuation(ret.apply(this.s, arguments)); }, throw: function _throw(value) { var thr = this.s.return; return void 0 === thr ? Promise.reject(value) : AsyncFromSyncIteratorContinuation(thr.apply(this.s, arguments)); } }, new AsyncFromSyncIterator(s); }




function isSupportedCategory(category) {
return category === undefined || category === _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.MOVIE || category === _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.DOCUMENTARY || category === _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.LIVE_PERFORMANCE;
}
var parseTorrents = element => {
var torrents = [];
element.querySelectorAll("tr.basic-movie-list__torrent-row").forEach(element => {
if (element.querySelector(".basic-movie-list__torrent-edition")) {
return;
}
var size = (0,_utils_utils__WEBPACK_IMPORTED_MODULE_1__.parseSize)(element.children[2].textContent);
var title = element.querySelector(".torrent-info-link").textContent;
var resolution = (0,_utils_utils__WEBPACK_IMPORTED_MODULE_1__.parseResolution)(title);
var tags = [];
if (title.includes("Remux")) {
tags.push("Remux");
}
var torrent = {
dom: element,
size,
tags,
resolution
};
torrents.push(torrent);
});
return torrents;
};
var parseCategory = element => {
var categoryTitle = element.querySelector(".basic-movie-list__torrent-edition__main").textContent;
if (categoryTitle.includes("Stand-up Comedy ")) {
return _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.STAND_UP;
} else if (categoryTitle.includes("Live Performance ")) {
return _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.LIVE_PERFORMANCE;
} else {
return _tracker__WEBPACK_IMPORTED_MODULE_2__.Category.MOVIE;
}
};
class PTP {
canBeUsedAsSource() {
return true;
Expand All @@ -1852,9 +1888,20 @@ class PTP {
}
getSearchRequest() {
return _wrapAsyncGenerator(function* () {
yield {
total: 0
};
var _document$querySelect;
var requests = [];
(_document$querySelect = document.querySelectorAll("#torrents-movie-view table.torrent_table > tbody")) === null || _document$querySelect === void 0 ? void 0 : _document$querySelect.forEach(element => {
var imdbId = (0,_utils_utils__WEBPACK_IMPORTED_MODULE_1__.parseImdbIdFromLink)(element.querySelector(".basic-movie-list__movie__ratings-and-tags"));
var request = {
torrents: parseTorrents(element),
dom: element,
imdbId,
query: "",
category: parseCategory(element)
};
requests.push(request);
});
yield* _asyncGeneratorDelegate(_asyncIterator((0,_tracker__WEBPACK_IMPORTED_MODULE_2__.toGenerator)(requests)), _awaitAsyncGenerator);
})();
}
name() {
Expand Down Expand Up @@ -2577,9 +2624,10 @@ var Category = /*#__PURE__*/function (Category) {
Category[Category["ANIME"] = 6] = "ANIME";
Category[Category["MV"] = 7] = "MV";
Category[Category["LIVE_PERFORMANCE"] = 8] = "LIVE_PERFORMANCE";
Category[Category["DOCUMENTARY"] = 9] = "DOCUMENTARY";
Category[Category["GAME"] = 10] = "GAME";
Category[Category["XXX"] = 11] = "XXX";
Category[Category["STAND_UP"] = 9] = "STAND_UP";
Category[Category["DOCUMENTARY"] = 10] = "DOCUMENTARY";
Category[Category["GAME"] = 11] = "GAME";
Category[Category["XXX"] = 12] = "XXX";
return Category;
}({});
var toGenerator = /*#__PURE__*/function () {
Expand Down Expand Up @@ -2771,6 +2819,11 @@ var parseResolution = text => {
for (var resolution of resolutions) {
if (text.includes(resolution)) return resolution;
}
var regex = /\b(\d{3})x(\d{3})\b/;
var match = text.match(regex);
if (match) {
return match[0];
}
return null;
};

Expand Down
1 change: 1 addition & 0 deletions Find Unique Titles/readme.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ You can generate it by running run `npm run dev`.
- Fix HDB not working as a source
- Refactor code to use generator for creating search requests. It is essentially practical when you need to do some fetching to build the request (like fetching another page for IMDB id).
- Add support for CHD and HDSky (thanks to @gawain12)
- PTP can be used a source

## 2023-09-30

Expand Down
76 changes: 71 additions & 5 deletions Find Unique Titles/src/trackers/PTP.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,19 @@ import {
addToMemoryCache,
getFromMemoryCache,
} from "../utils/cache";
import { parseSize } from "../utils/utils";
import { Category, MetaData, Request, Torrent, tracker } from "./tracker";
import {
parseImdbIdFromLink,
parseResolution,
parseSize,
} from "../utils/utils";
import {
Category,
MetaData,
Request,
toGenerator,
Torrent,
tracker,
} from "./tracker";
import tracker_tools from "common";

function isSupportedCategory(category: Category) {
Expand All @@ -16,6 +27,45 @@ function isSupportedCategory(category: Category) {
);
}

const parseTorrents = (element: HTMLElement) => {
const torrents = [];
element
.querySelectorAll("tr.basic-movie-list__torrent-row")
.forEach((element: HTMLElement) => {
if (element.querySelector(".basic-movie-list__torrent-edition")) {
return;
}
const size = parseSize(element.children[2].textContent);
let title = element.querySelector(".torrent-info-link").textContent;
const resolution = parseResolution(title);
const tags = [];
if (title.includes("Remux")) {
tags.push("Remux");
}
const torrent: Torrent = {
dom: element,
size,
tags,
resolution,
};
torrents.push(torrent);
});
return torrents;
};

const parseCategory = (element: HTMLElement): Category => {
const categoryTitle = element.querySelector(
".basic-movie-list__torrent-edition__main"
).textContent;
if (categoryTitle.includes("Stand-up Comedy ")) {
return Category.STAND_UP;
} else if (categoryTitle.includes("Live Performance ")) {
return Category.LIVE_PERFORMANCE;
} else {
return Category.MOVIE;
}
};

export default class PTP implements tracker {
canBeUsedAsSource(): boolean {
return true;
Expand All @@ -30,9 +80,25 @@ export default class PTP implements tracker {
}

async *getSearchRequest(): AsyncGenerator<MetaData | Request, void, void> {
yield {
total: 0,
};
const requests: Array<Request> = [];
document
.querySelectorAll("#torrents-movie-view table.torrent_table > tbody")
?.forEach((element: HTMLElement) => {
const imdbId = parseImdbIdFromLink(
element.querySelector(".basic-movie-list__movie__ratings-and-tags")
);

const request: Request = {
torrents: parseTorrents(element),
dom: element as HTMLElement,
imdbId,
query: "",
category: parseCategory(element),
};
requests.push(request);
});

yield* toGenerator(requests);
}

name(): string {
Expand Down
1 change: 1 addition & 0 deletions Find Unique Titles/src/trackers/tracker.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ export enum Category {
ANIME,
MV,
LIVE_PERFORMANCE,
STAND_UP,
DOCUMENTARY,
GAME,
XXX,
Expand Down
8 changes: 7 additions & 1 deletion Find Unique Titles/src/utils/utils.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ export const parseImdbIdFromLink = (element: HTMLElement) => {
};

export const parseImdbId = (text: string) => {
if (!text) return null
if (!text) return null;
const results = text.match(/(tt\d+)/);
if (!results) {
return null;
Expand All @@ -39,5 +39,11 @@ export const parseResolution = (text: string) => {
for (let resolution of resolutions) {
if (text.includes(resolution)) return resolution;
}
const regex = /\b(\d{3})x(\d{3})\b/;
const match = text.match(regex);

if (match) {
return match[0];
}
return null;
};

0 comments on commit 33e7c2b

Please sign in to comment.