core/utils/Tools.ts

import { NextRouter } from 'next/router'
import { inspect as utilInspect } from 'util'
import { createHmac } from 'crypto'
import { Readable } from 'stream'
import cookie from 'cookie'
import * as difflib from 'difflib'

import { BotFlags, ImageOptions, MetrixData, ServerFlags, UserFlags } from '@types'
import Logger from '@utils/Logger'
import { BASE_URLs, KoreanbotsEndPoints, Oauth } from '@utils/Constants'
import Day from './Day'

export function handlePWA(): boolean {
	let displayMode = 'browser'
	const mqStandAlone = '(display-mode: standalone)'
	if (window.navigator.standalone || window.matchMedia(mqStandAlone).matches) {
		displayMode = 'standalone'
	}
	try {
		window.ga('set', 'dimension1', displayMode)
	} catch {
		Logger.warn('[GA] Blocked.')
	}
	return displayMode === 'standalone'
}

export function formatNumber(value: number): string {
	if (!value) return '0'
	const suffixes = ['', '만', '억', '조', '해']
	const suffixNum = Math.floor(('' + value).length / 4)
	let shortValue: number = parseFloat(
		(suffixNum != 0 ? value / Math.pow(10000, suffixNum) : value).toPrecision(2)
	)
	if (shortValue % 1 != 0) {
		shortValue = Number(shortValue.toFixed(1))
	}
	if (suffixNum === 1 && shortValue < 1) return shortValue * 10 + '천'
	else if (shortValue === 1000) return '1천'
	return shortValue + suffixes[suffixNum]
}

function checkFlag(base: number, required: number) {
	return (base & required) === required
}

export function checkUserFlag(base: number, required: number | keyof typeof UserFlags): boolean {
	return checkFlag(base, typeof required === 'number' ? required : UserFlags[required])
}

export function checkBotFlag(base: number, required: number | keyof typeof BotFlags): boolean {
	return checkFlag(base, typeof required === 'number' ? required : BotFlags[required])
}

export function checkServerFlag(
	base: number,
	required: number | keyof typeof ServerFlags
): boolean {
	return checkFlag(base, typeof required === 'number' ? required : ServerFlags[required])
}

export function makeImageURL(root: string, { format = 'png', size = 256 }: ImageOptions): string {
	return `${root}.${format}?size=${size}`
}

export function makeBotURL({
	id,
	vanity,
	flags = 0,
}: {
	flags?: number
	vanity?: string
	id: string
}): string {
	return `/bots/${
		(checkBotFlag(flags, 'trusted') || checkBotFlag(flags, 'partnered')) && vanity ? vanity : id
	}`
}

export function makeServerURL({
	id,
	vanity,
	flags = 0,
}: {
	flags?: number
	vanity?: string
	id: string
}): string {
	return `/servers/${
		(checkServerFlag(flags, 'trusted') || checkServerFlag(flags, 'partnered')) && vanity
			? vanity
			: id
	}`
}

export function makeUserURL({ id }: { id: string }): string {
	return `/users/${id}`
}

export function serialize<T>(data: T): T {
	return JSON.parse(JSON.stringify(data))
}

export function diff(original: string, current: string, header = false, sep = '\n', join?: string) {
	return difflib
		.unifiedDiff(original.split(sep), current.split(sep))
		.slice(header ? 2 : 3)
		.join(join ?? sep)
}

export function objectDiff(
	original: Record<string, string>,
	current: Record<string, string>
): [string, (string | null)[]][] {
	const obj: Record<string, string[]> = {}
	Object.entries(original).forEach((k) => (obj[k[0]] = [k[1]]))
	Object.entries(current).forEach((k) => {
		if (!obj[k[0]]) obj[k[0]] = []
		obj[k[0]][1] = k[1]
	})
	return Object.entries(obj).filter((k) => k[1][0] !== k[1][1])
}

export function makeDiscordCodeblock(content: string, lang?: string): string {
	return `\`\`\`${lang || ''}\n${content.replace(/```/g, '\\`\\`\\`')}\n\`\`\``
}

export function inspect(object: unknown) {
	return utilInspect(object, {
		depth: Infinity,
		maxArrayLength: Infinity,
		maxStringLength: Infinity,
	})
}

export function supportsWebP() {
	const elem = document.createElement('canvas')
	if (elem.getContext && elem.getContext('2d')) {
		// was able or not to get WebP representation
		return elem.toDataURL('image/webp').indexOf('data:image/webp') == 0
	}
	// very old browser like IE 8, canvas not supported
	return false
}

export function systemTheme() {
	try {
		return window.matchMedia('(prefers-color-scheme: dark)')?.matches ? 'dark' : 'light'
	} catch {
		return 'dark'
	}
}

export function checkBrowser(): string {
	const ua = navigator.userAgent
	let tem
	let M =
		ua.match(/(opera|chrome|safari|firefox|msie|trident(?=\/))\/?\s*(\d+(\.\d+)?(\.\d+)?)/i) || []
	if (/trident/i.test(M[1])) {
		tem = /\brv[ :]+(\d+)/g.exec(ua) || []
		return 'IE ' + (tem[1] || '')
	}
	if (M[1] === 'Chrome') {
		tem = ua.match(/\b(OPR|Edge|Whale)\/(\d+)/)
		if (tem != null) return tem.slice(1).join(' ').replace('OPR', 'Opera')
	}
	M = M[2] ? [M[1], M[2]] : [navigator.appName, navigator.appVersion, '-?']
	if ((tem = ua.match(/version\/(\d+)/i)) != null) M.splice(1, 1, tem[1])
	return M.join(' ')
}

export function generateOauthURL(provider: 'discord' | 'github', clientID: string, scope?: string) {
	return Oauth[provider](clientID, scope)
}

export function formData(details: { [key: string]: string | number | boolean }) {
	const formBody = []
	for (const property in details) {
		const encodedKey = encodeURIComponent(property)
		const encodedValue = encodeURIComponent(details[property])
		formBody.push(encodedKey + '=' + encodedValue)
	}
	return formBody.join('&')
}

export function bufferToStream(binary: Buffer) {
	const readableInstanceStream = new Readable({
		read() {
			this.push(binary)
			this.push(null)
		},
	})

	return readableInstanceStream
}

export function parseCookie(req?: { headers: { cookie?: string } }): { [key: string]: string } {
	if (!req) return {}
	return cookie.parse(req.headers.cookie || '')
}

export function redirectTo(router: NextRouter, to: string) {
	router.push(KoreanbotsEndPoints[to] || to)
	return
}

// eslint-disable-next-line @typescript-eslint/no-explicit-any
export function cleanObject<T extends Record<any, any>>(obj: T): T {
	for (const propName in obj) {
		if (obj[propName] !== 0 && !obj[propName]) {
			obj[propName] = null
		}
	}
	return obj
}

export function camoUrl(url: string): string {
	return BASE_URLs.camo + `/${HMAC(url)}/${toHex(url)}`
}

export function HMAC(value: string, secret = process.env.CAMO_SECRET): string | null {
	try {
		return createHmac('sha1', secret).update(value, 'utf8').digest('hex')
	} catch {
		return null
	}
}

export function toHex(value: string): string {
	return Buffer.from(value).toString('hex')
}

export function getRandom<T = unknown>(arr: T[]): T {
	return arr[Math.floor(Math.random() * arr.length)]
}

export function parseDockerhubTag(imageTag: string) {
	return imageTag?.split('/').pop().split(':').pop()
}

export function getYYMMDD(): string {
	return new Date().toISOString().slice(0, 10).split('-').join('')
}

export function convertMetrixToGraph(data: MetrixData[], keyname?: string) {
	return data.map((el) => ({
		x: Day(el.day, 'YYMMDD').toDate(),
		y: el[keyname] || el.count,
	}))
}

export * from './ShowdownExtensions'