-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
13f4040
commit bc44476
Showing
2 changed files
with
136 additions
and
81 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,127 @@ | ||
| import fetch, {Headers, RequestInit, Response} from "node-fetch"; | ||
| import { photonVersion } from "./version"; | ||
| import { appId } from "../static/scripts/utils/consts"; | ||
|
|
||
|
|
||
| export class RedditAuth { | ||
| private accessToken: string = ""; | ||
| private accessTokenExpires: number = 0; | ||
| rateLimitUsed: number = 0; | ||
| rateLimitRemaining: number = 100; | ||
| rateLimitResetAt: number = Date.now()/1000 + 100; | ||
|
|
||
| async getAccessTokenWithRefresh(): Promise<string> { | ||
| const now = Date.now(); | ||
| if (this.accessToken && this.accessTokenExpires > now) | ||
| return this.accessToken; | ||
| this.accessToken = await this.getAccessToken(); | ||
| this.accessTokenExpires = now + 1000 * 60 * 60; | ||
| return this.accessToken; | ||
| } | ||
|
|
||
| async oauthRequest<T>(endpoint: string, params: { [q: string]: string }, method: string = "GET"): Promise<T> { | ||
| const fetchOptions: RequestInit = { | ||
| method: method, | ||
| headers: { | ||
| "Authorization": "Bearer " + await this.getAccessTokenWithRefresh(), | ||
| }, | ||
| }; | ||
| params["raw_json"] = "1"; | ||
| if (method !== "GET") { | ||
| fetchOptions.body = JSON.stringify(params); | ||
| } | ||
| const url = new URL(`https://oauth.reddit.com${endpoint}`); | ||
| for (const key in params) { | ||
| url.searchParams.append(key, params[key]); | ||
| } | ||
| const response = await fetch(url.toString(), fetchOptions); | ||
| await this.rateLimitCheck(response.headers); | ||
| const text = await response.text(); | ||
| return JSON.parse(text); | ||
| } | ||
|
|
||
| async resolvePath(path: string): Promise<{ url: string, path: string, redirects: number }> { | ||
| let response: Response; | ||
| let url: string; | ||
| let previousUrl: string|undefined; | ||
| let isRedirect = false; | ||
| let i = 0; | ||
| const supportedRedditUrl = /^https:\/\/(\w+\.)?reddit.com\/(r|u|user)\/[^/#?]+([?/#]|$)(?!s\/)/i; // not a /s/ url | ||
| do { | ||
| response = await fetch(`https://oauth.reddit.com${path}`, { | ||
| method: "HEAD", | ||
| redirect: "manual", | ||
| headers: { | ||
| "Authorization": "Bearer " + await this.getAccessTokenWithRefresh(), | ||
| "User-Agent": `web_backend:photon-reddit.com:v${photonVersion} (by /u/RaiderBDev)` | ||
| } | ||
| }); | ||
| let newUrl = response.headers.get("location"); | ||
| if (!newUrl) | ||
| break; | ||
| if (!newUrl.startsWith("http")) | ||
| newUrl = new URL(newUrl, response.url).href; | ||
| url = newUrl; | ||
| if (url === previousUrl) | ||
| break; | ||
| previousUrl = url; | ||
| isRedirect = response.status >= 300 && response.status < 400; | ||
| i++; | ||
| } | ||
| while (isRedirect && !supportedRedditUrl.test(url) && i < 5); | ||
|
|
||
| if (!supportedRedditUrl.test(url)) { | ||
| throw new Error("Failed to resolve url"); | ||
| } | ||
| const urlObj = new URL(url); | ||
| urlObj.search = ""; | ||
| urlObj.hash = ""; | ||
|
|
||
| return { | ||
| url: urlObj.toString(), | ||
| path: urlObj.pathname, | ||
| redirects: i | ||
| }; | ||
| } | ||
|
|
||
| async verifyIsWorking(): Promise<void> { | ||
| const response = await this.oauthRequest<any>("/api/v1/me", {}); | ||
| if (!response || response.error) | ||
| throw new Error("Invalid response"); | ||
| } | ||
|
|
||
| async getAccessToken(): Promise<string> { | ||
| const response = await fetch("https://www.reddit.com/api/v1/access_token", { | ||
| method: "POST", | ||
| headers: { | ||
| "Content-Type": "application/x-www-form-urlencoded", | ||
| "Authorization": "Basic " + Buffer.from(appId + ":").toString("base64"), | ||
| "User-Agent": `web_backend:photon-reddit.com:v${photonVersion} (by /u/RaiderBDev)` | ||
| }, | ||
| body: new URLSearchParams({ | ||
| grant_type: "https://oauth.reddit.com/grants/installed_client", | ||
| device_id: "DO_NOT_TRACK_THIS_DEVICE" | ||
| }) | ||
| }); | ||
| const json: any = await response.json(); | ||
| const accessToken = json["access_token"]; | ||
| if (typeof accessToken !== "string" || accessToken.length === 0) | ||
| throw new Error("Invalid access token"); | ||
| return accessToken; | ||
| } | ||
|
|
||
| private async rateLimitCheck(headers: Headers): Promise<void> { | ||
| const rlReqUsed = parseInt(headers.get("x-ratelimit-used")!); | ||
| const rlReqRemaining = parseInt(headers.get("x-ratelimit-remaining")!); | ||
| const rlTimeRemaining = parseInt(headers.get("x-ratelimit-reset")!); | ||
| if (isNaN(rlReqUsed) || isNaN(rlReqRemaining) || isNaN(rlTimeRemaining)) | ||
| return; | ||
| this.rateLimitUsed = rlReqUsed; | ||
| this.rateLimitRemaining = rlReqRemaining; | ||
| const rateLimitResetAt = Date.now()/1000 + rlTimeRemaining; | ||
| if (rateLimitResetAt > this.rateLimitResetAt) | ||
| this.rateLimitResetAt = rateLimitResetAt; | ||
| if (rlReqRemaining < 15) | ||
| console.log(`Rate limit: ${rlReqRemaining} requests remaining, ${rlTimeRemaining} seconds until reset`); | ||
| } | ||
| } |