Files
wr.do/lib/utils.ts
2025-05-26 17:40:01 +08:00

369 lines
9.7 KiB
TypeScript

import crypto from "crypto";
import { Metadata } from "next";
import { clsx, type ClassValue } from "clsx";
import ms from "ms";
import { twMerge } from "tailwind-merge";
import { siteConfig } from "@/config/site";
import { TIME_RANGES } from "./enums";
export function cn(...inputs: ClassValue[]) {
return twMerge(clsx(inputs));
}
export function constructMetadata({
title = siteConfig.name,
description = siteConfig.description,
image = siteConfig.ogImage,
icons = "/favicon.ico",
noIndex = false,
}: {
title?: string;
description?: string;
image?: string;
icons?: string;
noIndex?: boolean;
} = {}): Metadata {
return {
title,
description,
keywords: [
"Cloudflare",
"DNS",
"DNS Records",
"Short Link",
"Open API",
"Screenshot API",
],
authors: [
{
name: "oiov",
},
],
creator: "oiov",
openGraph: {
type: "website",
locale: "en_US",
url: siteConfig.url,
title,
description,
siteName: title,
},
twitter: {
card: "summary_large_image",
title,
description,
images: [image],
creator: "@yesmoree",
},
icons,
metadataBase: new URL(siteConfig.url),
manifest: `${siteConfig.url}/site.webmanifest`,
...(noIndex && {
robots: {
index: false,
follow: false,
},
}),
};
}
export function formatDate(input: string | number): string {
const date = new Date(input);
const locale = navigator.language || "en-US";
return date.toLocaleDateString(locale, {
second: "numeric",
minute: "numeric",
hour: "numeric",
weekday: "long",
month: "long",
day: "numeric",
year: "numeric",
});
}
export function formatTime(input: string | number): string {
const date = new Date(input);
const locale = navigator.language || "en-US";
return date.toLocaleTimeString(locale, {
// second: "numeric",
minute: "numeric",
hour: "numeric",
});
}
// Utils from precedent.dev
export const timeAgo = (timestamp: Date, timeOnly?: boolean): string => {
if (!timestamp) return "never";
return `${ms(Date.now() - new Date(timestamp).getTime())}${
timeOnly ? "" : " ago"
}`;
};
export const expirationTime = (
expiration: string,
updatedAt?: Date,
timeOnly?: boolean,
): string => {
if (!expiration || !updatedAt) return "Invalid data";
if (expiration === "-1") return "Never";
const expirationSeconds = parseInt(expiration, 10);
if (isNaN(expirationSeconds)) return "Invalid expiration format";
const now = Date.now();
const updatedAtTimestamp = new Date(updatedAt).getTime();
const expirationMilliseconds = expirationSeconds * 1000;
const expirationTime = updatedAtTimestamp + expirationMilliseconds;
const remainingTime = expirationTime - now;
if (remainingTime <= 0) return "Expired";
const remainingTimeString = ms(remainingTime, { long: true });
if (timeOnly) {
return remainingTimeString;
}
return `${remainingTimeString}`;
};
export async function fetcher<JSON = any>(
input: RequestInfo,
init?: RequestInit,
): Promise<JSON> {
const res = await fetch(input, init);
if (!res.ok) {
const json = await res.json();
if (json.error) {
const error = new Error(json.error) as Error & {
status: number;
};
error.status = res.status;
throw error;
} else {
throw new Error("An unexpected error occurred");
}
}
return res.json();
}
export const isValidEmail = (email: string) =>
/^[^\s@]+@[^\s@]+\.[^\s@]+$/.test(email);
export function nFormatter(num: number, digits: number = 2) {
if (!num) return "0";
const lookup = [
{ value: 1, symbol: "" },
{ value: 1e3, symbol: "K" },
{ value: 1e6, symbol: "M" },
{ value: 1e9, symbol: "G" },
{ value: 1e12, symbol: "T" },
{ value: 1e15, symbol: "P" },
{ value: 1e18, symbol: "E" },
];
const rx = /\.0+$|(\.[0-9]*[1-9])0+$/;
var item = lookup
.slice()
.reverse()
.find(function (item) {
return num >= item.value;
});
return item
? (num / item.value).toFixed(digits || 1).replace(rx, "$1") + item.symbol
: "0";
}
export function capitalize(str: string) {
if (!str || typeof str !== "string") return str;
return str.charAt(0).toUpperCase() + str.slice(1);
}
export const truncate = (str: string, length: number) => {
if (!str || str.length <= length) return str;
return `${str.slice(0, length)}...`;
};
export const getBlurDataURL = async (url: string | null) => {
if (!url) {
return "data:image/webp;base64,AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
}
if (url.startsWith("/_static/")) {
url = `${siteConfig.url}${url}`;
}
try {
const response = await fetch(
`https://wsrv.nl/?url=${url}&w=50&h=50&blur=5`,
);
const buffer = await response.arrayBuffer();
const base64 = Buffer.from(buffer).toString("base64");
return `data:image/png;base64,${base64}`;
} catch (error) {
return "data:image/webp;base64,AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA=";
}
};
export const placeholderBlurhash =
"data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAAAXNSR0IArs4c6QAAAoJJREFUWEfFl4lu4zAMRO3cx/9/au6reMaOdkxTTl0grQFCRoqaT+SQotq2bV9N8rRt28xms87m83l553eZ/9vr9Wpkz+ezkT0ej+6dv1X81AFw7M4FBACPVn2c1Z3zLgDeJwHgeLFYdAARYioAEAKJEG2WAjl3gCwNYymQQ9b7/V4spmIAwO6Wy2VnAMikBWlDURBELf8CuN1uHQSrPwMAHK5WqwFELQ01AIXdAa7XawfAb3p6AOwK5+v1ugAoEq4FRSFLgavfQ49jAGQpAE5wjgGCeRrGdBArwHOPcwFcLpcGU1X0IsBuN5tNgYhaiFFwHTiAwq8I+O5xfj6fOz38K+X/fYAdb7fbAgFAjIJ6Aav3AYlQ6nfnDoDz0+lUxNiLALvf7XaDNGQ6GANQBKR85V27B4D3QQRw7hGIYlQKWGM79hSweyCUe1blXhEAogfABwHAXAcqSYkxCtHLUK3XBajSc4Dj8dilAeiSAgD2+30BAEKV4GKcAuDqB4TdYwBgPQByCgApUBoE4EJUGvxUjF3Q69/zLw3g/HA45ABKgdIQu+JPIyDnisCfAxAFNFM0EFNQ64gfS0EUoQP8ighrZSjn3oziZEQpauyKbfjbZchHUL/3AS/Dd30gAkxuRACgfO+EWQW8qwI1o+wseNuKcQiESjALvwNoMI0TcRzD4lFcPYwIM+JTF5x6HOs8yI7jeB5oKhpMRFH9UwaSCDB2Jmg4rc6E2TT0biIaG0rQhNqyhpHBcayTTSXH6vcDL7/sdqRK8LkwTsU499E8vRcAojHcZ4AxABdilgrp4lsXk8oVqgwh7+6H3phqd8J0Kk4vbx/+sZqCD/vNLya/5dT9fAH8g1WdNGgwbQAAAABJRU5ErkJggg==";
export function generateSecret(length: number = 16): string {
const buffer = crypto.randomBytes(length);
return buffer.toString("hex");
}
export function generateUrlSuffix(length: number = 6): string {
const characters =
"ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
const charactersLength = characters.length;
let result = "";
const randomValues = crypto.randomBytes(length);
for (let i = 0; i < length; i++) {
result += characters[randomValues[i] % charactersLength];
}
return result;
}
export function isLink(str: string): boolean {
try {
const url = new URL(str);
return true;
} catch (_) {
return false;
}
}
export function removeUrlSuffix(url: string): string {
return url.startsWith("http") ? url.split("//")[1] : url;
}
export function toCamelCase(str: string) {
return str
.split("-")
.map((word) => word.charAt(0).toUpperCase() + word.slice(1))
.join("");
}
export const getStartDate = (range: string): Date | undefined => {
if (!range || !(range in TIME_RANGES)) return undefined;
return new Date(Date.now() - TIME_RANGES[range]);
};
export function htmlToText(html: string): string {
if (typeof window === "undefined") return html; // 服务端渲染时返回原始内容
const parser = new DOMParser();
const doc = parser.parseFromString(html, "text/html");
return doc.body.textContent || "";
}
export function formatFileSize(bytes: number): string {
if (bytes === 0) return "0 B";
const k = 1024;
const sizes = ["B", "KB", "MB", "GB"];
const i = Math.floor(Math.log(bytes) / Math.log(k));
return `${(bytes / Math.pow(k, i)).toFixed(1)} ${sizes[i]}`;
}
export function downloadFile(url: string, filename: string): Promise<void> {
return new Promise((resolve, reject) => {
const a = document.createElement("a");
a.href = url;
a.download = filename;
a.setAttribute("rel", "noopener noreferrer");
a.setAttribute("target", "_blank");
try {
document.body.appendChild(a);
a.click();
document.body.removeChild(a);
resolve();
} catch (error) {
reject(error);
}
});
}
export async function downloadFileFromUrl(
url: string,
filename: string,
): Promise<void> {
try {
// 获取文件内容
const response = await fetch(url, {
// credentials: "include", // 如果需要带上cookie
// mode: "cors", // 处理跨域
});
if (!response.ok) {
throw new Error(`HTTP error! status: ${response.status}`);
}
// 获取 Blob
const blob = await response.blob();
// 创建下载 URL
const downloadUrl = URL.createObjectURL(blob);
// 创建下载链接
const a = document.createElement("a");
a.href = downloadUrl;
a.download = filename;
// 执行下载
document.body.appendChild(a);
a.click();
document.body.removeChild(a);
// 清理
URL.revokeObjectURL(downloadUrl);
} catch (error) {
console.error("下载失败:", error);
throw error;
}
}
export const getSearchParams = (url: string) => {
// Create a params object
let params = {} as Record<string, string>;
new URL(url).searchParams.forEach(function (val, key) {
params[key] = val;
});
return params;
};
export const isValidUrl = (url: string) => {
try {
new URL(url);
return true;
} catch (e) {
return false;
}
};
export const getUrlFromString = (str: string) => {
if (isValidUrl(str)) return str;
try {
if (str.includes(".") && !str.includes(" ")) {
return new URL(`https://${str}`).toString();
}
} catch (_) {}
return str;
};
export function extractHost(url: string): string {
const regex = /^(?:https?:\/\/)?([^\/?:#]+)/i;
const match = url.match(regex);
return match ? match[1] : "";
}