Spaces:
Running
Running
File size: 2,736 Bytes
6b3405c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
import { PreviewServer, ViteDevServer } from "vite";
import { RateLimiterMemory } from "rate-limiter-flexible";
import { argon2Verify } from "hash-wasm";
import { incrementSearchesSinceLastRestart } from "./searchesSinceLastRestart";
import { rankSearchResults } from "./rankSearchResults";
import { getSearchToken } from "./searchToken";
import { fetchSearXNG } from "./fetchSearXNG";
import { isVerifiedToken, addVerifiedToken } from "./verifiedTokens";
export function searchEndpointServerHook<
T extends ViteDevServer | PreviewServer,
>(server: T) {
const rateLimiter = new RateLimiterMemory({
points: 2,
duration: 10,
});
server.middlewares.use(async (request, response, next) => {
if (!request.url.startsWith("/search")) return next();
const { searchParams } = new URL(
request.url,
`http://${request.headers.host}`,
);
const limitParam = searchParams.get("limit");
const limit =
limitParam && Number(limitParam) > 0 ? Number(limitParam) : undefined;
const query = searchParams.get("q");
if (!query) {
response.statusCode = 400;
response.end("Missing the query parameter.");
return;
}
const token = searchParams.get("token");
if (!isVerifiedToken(token)) {
let isValidToken = false;
try {
isValidToken = await argon2Verify({
password: getSearchToken(),
hash: token,
});
} catch (error) {
void error;
}
if (isValidToken) {
addVerifiedToken(token);
} else {
response.statusCode = 401;
response.end("Unauthorized.");
return;
}
}
try {
await rateLimiter.consume(token);
} catch {
response.statusCode = 429;
response.end("Too many requests.");
return;
}
const { textResults, imageResults } = await fetchSearXNG(query, limit);
incrementSearchesSinceLastRestart();
if (textResults.length === 0 && imageResults.length === 0) {
response.setHeader("Content-Type", "application/json");
response.end(JSON.stringify({ textResults: [], imageResults: [] }));
return;
}
try {
const rankedTextResults = await rankSearchResults(query, textResults);
response.setHeader("Content-Type", "application/json");
response.end(
JSON.stringify({ textResults: rankedTextResults, imageResults }),
);
} catch (error) {
const errorMessage =
error instanceof Error ? error.message : String(error);
console.error(`Error ranking search results: ${errorMessage}`);
response.setHeader("Content-Type", "application/json");
response.end(JSON.stringify({ textResults, imageResults }));
}
});
}
|