mirror of
https://github.com/Nezumi-2711/onedrive-vercel-index.git
synced 2026-09-22 20:01:46 +00:00
refactor new file download functions to new file
This commit is contained in:
spencerwooo
@@ -1,141 +1,40 @@
|
||||
import axios from 'axios'
|
||||
import sha256 from 'crypto-js/sha256'
|
||||
import useSWR, { cache, Key, useSWRInfinite } from 'swr'
|
||||
import toast from 'react-hot-toast'
|
||||
import JSZip from 'jszip'
|
||||
|
||||
import siteConfig from '../config/site.json'
|
||||
import { fetcher } from './fetchWithSWR'
|
||||
import { getStoredToken } from './protectedRouteHandler'
|
||||
|
||||
/**
|
||||
* Extract the current web page's base url
|
||||
* @returns base url of the page
|
||||
*/
|
||||
export const getBaseUrl = () => {
|
||||
if (typeof window !== 'undefined') {
|
||||
return window.location.origin
|
||||
}
|
||||
return ''
|
||||
}
|
||||
// Blob download helper
|
||||
export function downloadBlob({ blob, name }: { blob: Blob; name: string }) {
|
||||
// Prepare for download
|
||||
const el = document.createElement('a')
|
||||
el.style.display = 'none'
|
||||
document.body.appendChild(el)
|
||||
|
||||
// Common axios fetch function for use with useSWR
|
||||
const fetcher = (url: string, token?: string) => {
|
||||
return token
|
||||
? axios
|
||||
.get(url, {
|
||||
headers: { 'od-protected-token': token },
|
||||
})
|
||||
.then(res => res.data)
|
||||
: axios.get(url).then(res => res.data)
|
||||
}
|
||||
/**
|
||||
* Use stale SWR instead of revalidating on each request. Not ideal for this scenario but have to do
|
||||
* if fetching serverside props from component instead of pages.
|
||||
* @param url request url
|
||||
* @returns useSWR instance
|
||||
*/
|
||||
export const useStaleSWR = (url: Key, path: string = '') => {
|
||||
const revalidationOptions = {
|
||||
revalidateOnMount: !(cache.has(`arg@"${url}"@null`) || cache.has(url)),
|
||||
revalidateOnFocus: false,
|
||||
revalidateOnReconnect: true,
|
||||
}
|
||||
|
||||
const hashedToken = getStoredToken(path)
|
||||
return useSWR([url, hashedToken], fetcher, revalidationOptions)
|
||||
}
|
||||
|
||||
/**
|
||||
* Paging with useSWRInfinite + protected token support
|
||||
* @param path Current query directory path
|
||||
* @returns useSWRInfinite API
|
||||
*/
|
||||
export const useProtectedSWRInfinite = (path: string = '') => {
|
||||
const hashedToken = getStoredToken(path)
|
||||
|
||||
/**
|
||||
* Next page infinite loading for useSWR
|
||||
* @param pageIdx The index of this paging collection
|
||||
* @param prevPageData Previous page information
|
||||
* @param path Directory path
|
||||
* @returns API to the next page
|
||||
*/
|
||||
const getNextKey = (pageIndex, previousPageData) => {
|
||||
// Reached the end of the collection
|
||||
if (previousPageData && !previousPageData.folder) return null
|
||||
|
||||
// First page with no prevPageData
|
||||
if (pageIndex === 0) return [`/api?path=${path}`, hashedToken]
|
||||
|
||||
// Add nextPage token to API endpoint
|
||||
return [`/api?path=${path}&next=${previousPageData.next}`, hashedToken]
|
||||
}
|
||||
|
||||
const revalidationOptions = {
|
||||
revalidateOnMount: !(cache.has(`arg@"/api?path=${path}"@null`) || cache.has(`/api?path=${path}`)),
|
||||
revalidateOnFocus: false,
|
||||
revalidateOnReconnect: true,
|
||||
}
|
||||
return useSWRInfinite(getNextKey, fetcher, revalidationOptions)
|
||||
}
|
||||
|
||||
// Hash password token with SHA256
|
||||
const encryptToken = (token: string) => {
|
||||
return sha256(token).toString()
|
||||
}
|
||||
// Fetch stored token from localStorage and encrypt with SHA256
|
||||
const getStoredToken = (path: string) => {
|
||||
const storedToken =
|
||||
typeof window !== 'undefined' ? JSON.parse(localStorage.getItem(matchProtectedRoute(path)) as string) : ''
|
||||
return storedToken ? encryptToken(storedToken) : null
|
||||
}
|
||||
/**
|
||||
* Compares the hash of .password and od-protected-token header
|
||||
* @param odTokenHeader od-protected-token header (sha256 hashed token)
|
||||
* @param dotPassword non-hashed .password file
|
||||
* @returns whether the two hashes are the same
|
||||
*/
|
||||
export const compareHashedToken = (odTokenHeader: string, dotPassword: string) => {
|
||||
return encryptToken(dotPassword.trim()) === odTokenHeader
|
||||
}
|
||||
|
||||
/**
|
||||
* Match the specified route against a list of predefined routes
|
||||
* @param route directory path
|
||||
* @returns whether the directory is protected
|
||||
*/
|
||||
export const matchProtectedRoute = (route: string) => {
|
||||
const protectedRoutes: string[] = siteConfig.protectedRoutes
|
||||
let authTokenPath = ''
|
||||
|
||||
for (const r of protectedRoutes) {
|
||||
// protected route array could be empty
|
||||
if (r) {
|
||||
if (
|
||||
route.startsWith(
|
||||
r
|
||||
.split('/')
|
||||
.map(p => encodeURIComponent(p))
|
||||
.join('/')
|
||||
)
|
||||
) {
|
||||
authTokenPath = r
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return authTokenPath
|
||||
// Download zip file
|
||||
const bUrl = window.URL.createObjectURL(blob)
|
||||
el.href = bUrl
|
||||
el.download = name
|
||||
el.click()
|
||||
window.URL.revokeObjectURL(bUrl)
|
||||
el.remove()
|
||||
}
|
||||
|
||||
/**
|
||||
* Download multiple files after compressing them into a zip
|
||||
* @param toastId Toast ID to be used for toast notification
|
||||
* @param files Files to be downloaded
|
||||
* @param folder Optional folder name to hold files, otherwise flatten files in the zip
|
||||
*/
|
||||
export const downloadMultipleFiles = async (
|
||||
toastId: string,
|
||||
files: { name: string; url: string }[],
|
||||
export async function downloadMultipleFiles({
|
||||
toastId,
|
||||
files,
|
||||
folder,
|
||||
}: {
|
||||
toastId: string
|
||||
files: { name: string; url: string }[]
|
||||
folder?: string
|
||||
) => {
|
||||
}): Promise<void> {
|
||||
const zip = new JSZip()
|
||||
const dir = folder ? zip.folder(folder)! : zip
|
||||
|
||||
@@ -155,23 +54,74 @@ export const downloadMultipleFiles = async (
|
||||
id: toastId,
|
||||
})
|
||||
})
|
||||
downloadBlob(b, folder ? folder + '.zip' : 'download.zip')
|
||||
downloadBlob({ blob: b, name: folder ? folder + '.zip' : 'download.zip' })
|
||||
}
|
||||
|
||||
// Blob download helper
|
||||
const downloadBlob = (b: Blob, name: string) => {
|
||||
// Prepare for download
|
||||
const el = document.createElement('a')
|
||||
el.style.display = 'none'
|
||||
document.body.appendChild(el)
|
||||
/**
|
||||
* Download hierarchical tree-like files after compressing them into a zip
|
||||
* @param toastId Toast ID to be used for toast notification
|
||||
* @param files Files to be downloaded. Array of file and folder items excluding root folder.
|
||||
* Folder items MUST be in front of its children items in the array.
|
||||
* Use async generator because generation of the array may be slow.
|
||||
* When waiting for its generation, we can meanwhile download bodies of already got items.
|
||||
* Only folder items can have url undefined.
|
||||
* @param basePath Root dir path of files to be downloaded
|
||||
* @param folder Optional folder name to hold files, otherwise flatten files in the zip
|
||||
*/
|
||||
|
||||
// Download zip file
|
||||
const bUrl = window.URL.createObjectURL(b)
|
||||
el.href = bUrl
|
||||
el.download = name
|
||||
el.click()
|
||||
window.URL.revokeObjectURL(bUrl)
|
||||
el.remove()
|
||||
export async function downloadTreelikeMultipleFiles({
|
||||
toastId,
|
||||
files,
|
||||
basePath,
|
||||
folder,
|
||||
}: {
|
||||
toastId: string
|
||||
files: AsyncGenerator<{
|
||||
name: string
|
||||
url?: string
|
||||
path: string
|
||||
isFolder: boolean
|
||||
}>
|
||||
basePath: string
|
||||
folder?: string
|
||||
}): Promise<void> {
|
||||
const zip = new JSZip()
|
||||
const root = folder ? zip.folder(folder)! : zip
|
||||
const map = [{ path: basePath, dir: root }]
|
||||
|
||||
// Add selected file blobs to zip according to its path
|
||||
for await (const { name, url, path, isFolder } of files) {
|
||||
// Search parent dir in map
|
||||
const i = map
|
||||
.slice()
|
||||
.reverse()
|
||||
.findIndex(
|
||||
({ path: parent }) =>
|
||||
path.substring(0, parent.length) === parent && path.substring(parent.length + 1).indexOf('/') === -1
|
||||
)
|
||||
if (i === -1) {
|
||||
throw new Error('File array does not satisfy requirement')
|
||||
}
|
||||
|
||||
// Add file or folder to zip
|
||||
const dir = map[map.length - 1 - i].dir
|
||||
if (isFolder) {
|
||||
map.push({ path, dir: dir.folder(name)! })
|
||||
} else {
|
||||
dir.file(
|
||||
name,
|
||||
fetch(url!).then(r => r.blob())
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Create zip file and download it
|
||||
const b = await zip.generateAsync({ type: 'blob' }, metadata => {
|
||||
toast.loading(`Downloading ${metadata.percent.toFixed(0)}%. Refresh to cancel...`, {
|
||||
id: toastId,
|
||||
})
|
||||
})
|
||||
downloadBlob({ blob: b, name: folder ? folder + '.zip' : 'download.zip' })
|
||||
}
|
||||
|
||||
/**
|
||||
@@ -218,63 +168,3 @@ export async function* traverseFolder(path: string): AsyncGenerator<
|
||||
folderPaths = items.filter(i => i.isFolder).map(i => i.path)
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Download hierarchical tree-like files after compressing them into a zip
|
||||
* @param files Files to be downloaded. Array of file and folder items excluding root folder.
|
||||
* Folder items MUST be in front of its children items in the array.
|
||||
* Use async generator because generation of the array may be slow.
|
||||
* When waiting for its generation, we can meanwhile download bodies of already got items.
|
||||
* Only folder items can have url undefined.
|
||||
* @param basePath Root dir path of files to be downloaded
|
||||
* @param folder Optional folder name to hold files, otherwise flatten files in the zip
|
||||
*/
|
||||
export const downloadTreelikeMultipleFiles = async (
|
||||
toastId: string,
|
||||
files: AsyncGenerator<{
|
||||
name: string
|
||||
url?: string
|
||||
path: string
|
||||
isFolder: boolean
|
||||
}>,
|
||||
basePath: string,
|
||||
folder?: string
|
||||
) => {
|
||||
const zip = new JSZip()
|
||||
const root = folder ? zip.folder(folder)! : zip
|
||||
const map = [{ path: basePath, dir: root }]
|
||||
|
||||
// Add selected file blobs to zip according to its path
|
||||
for await (const { name, url, path, isFolder } of files) {
|
||||
// Search parent dir in map
|
||||
const i = map
|
||||
.slice()
|
||||
.reverse()
|
||||
.findIndex(
|
||||
({ path: parent }) =>
|
||||
path.substring(0, parent.length) === parent && path.substring(parent.length + 1).indexOf('/') === -1
|
||||
)
|
||||
if (i === -1) {
|
||||
throw new Error('File array does not satisfy requirement')
|
||||
}
|
||||
|
||||
// Add file or folder to zip
|
||||
const dir = map[map.length - 1 - i].dir
|
||||
if (isFolder) {
|
||||
map.push({ path, dir: dir.folder(name)! })
|
||||
} else {
|
||||
dir.file(
|
||||
name,
|
||||
fetch(url!).then(r => r.blob())
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
// Create zip file and download it
|
||||
const b = await zip.generateAsync({ type: 'blob' }, metadata => {
|
||||
toast.loading(`Downloading ${metadata.percent.toFixed(0)}%. Refresh to cancel...`, {
|
||||
id: toastId,
|
||||
})
|
||||
})
|
||||
downloadBlob(b, folder ? folder + '.zip' : 'download.zip')
|
||||
}
|
||||
@@ -0,0 +1,65 @@
|
||||
import axios from 'axios'
|
||||
import useSWR, { cache, Key, useSWRInfinite } from 'swr'
|
||||
|
||||
import { getStoredToken } from './protectedRouteHandler'
|
||||
|
||||
// Common axios fetch function for use with useSWR
|
||||
export function fetcher(url: string, token?: string): Promise<any> {
|
||||
return token
|
||||
? axios
|
||||
.get(url, {
|
||||
headers: { 'od-protected-token': token },
|
||||
})
|
||||
.then(res => res.data)
|
||||
: axios.get(url).then(res => res.data)
|
||||
}
|
||||
/**
|
||||
* Use stale SWR instead of revalidating on each request. Not ideal for this scenario but have to do
|
||||
* if fetching serverside props from component instead of pages.
|
||||
* @param url request url
|
||||
* @returns useSWR instance
|
||||
*/
|
||||
export function useStaleSWR({ url, path = '' }: { url: Key; path?: string }) {
|
||||
const revalidationOptions = {
|
||||
revalidateOnMount: !(cache.has(`arg@"${url}"@null`) || cache.has(url)),
|
||||
revalidateOnFocus: false,
|
||||
revalidateOnReconnect: true,
|
||||
}
|
||||
|
||||
const hashedToken = getStoredToken(path)
|
||||
return useSWR([url, hashedToken], fetcher, revalidationOptions)
|
||||
}
|
||||
|
||||
/**
|
||||
* Paging with useSWRInfinite + protected token support
|
||||
* @param path Current query directory path
|
||||
* @returns useSWRInfinite API
|
||||
*/
|
||||
export function useProtectedSWRInfinite(path: string = '') {
|
||||
const hashedToken = getStoredToken(path)
|
||||
|
||||
/**
|
||||
* Next page infinite loading for useSWR
|
||||
* @param pageIdx The index of this paging collection
|
||||
* @param prevPageData Previous page information
|
||||
* @param path Directory path
|
||||
* @returns API to the next page
|
||||
*/
|
||||
function getNextKey(pageIndex, previousPageData): (string | null)[] | null {
|
||||
// Reached the end of the collection
|
||||
if (previousPageData && !previousPageData.folder) return null
|
||||
|
||||
// First page with no prevPageData
|
||||
if (pageIndex === 0) return [`/api?path=${path}`, hashedToken]
|
||||
|
||||
// Add nextPage token to API endpoint
|
||||
return [`/api?path=${path}&next=${previousPageData.next}`, hashedToken]
|
||||
}
|
||||
|
||||
const revalidationOptions = {
|
||||
revalidateOnMount: !(cache.has(`arg@"/api?path=${path}"@null`) || cache.has(`/api?path=${path}`)),
|
||||
revalidateOnFocus: false,
|
||||
revalidateOnReconnect: true,
|
||||
}
|
||||
return useSWRInfinite(getNextKey, fetcher, revalidationOptions)
|
||||
}
|
||||
@@ -0,0 +1,10 @@
|
||||
/**
|
||||
* Extract the current web page's base url
|
||||
* @returns base url of the page
|
||||
*/
|
||||
export function getBaseUrl(): string {
|
||||
if (typeof window !== 'undefined') {
|
||||
return window.location.origin
|
||||
}
|
||||
return ''
|
||||
}
|
||||
@@ -83,15 +83,15 @@ const extensions = {
|
||||
* @param key The index key
|
||||
* @returns Whether or not the key exists inside the object
|
||||
*/
|
||||
export const hasKey = <O>(obj: O, key: PropertyKey): key is keyof O => {
|
||||
export function hasKey<O>(obj: O, key: PropertyKey): key is keyof O {
|
||||
return key in obj
|
||||
}
|
||||
|
||||
export const getExtension = (fileName: string) => {
|
||||
export function getExtension(fileName: string): string {
|
||||
return fileName.slice(((fileName.lastIndexOf('.') - 1) >>> 0) + 2).toLowerCase()
|
||||
}
|
||||
|
||||
export const getFileIcon = (fileName: string) => {
|
||||
export function getFileIcon(fileName: string): [IconPrefix, IconName] {
|
||||
const extension = getExtension(fileName)
|
||||
return hasKey(extensions, extension) ? extensions[extension] : icons.file
|
||||
}
|
||||
|
||||
@@ -0,0 +1,58 @@
|
||||
import sha256 from 'crypto-js/sha256'
|
||||
import siteConfig from '../config/site.json'
|
||||
|
||||
// Hash password token with SHA256
|
||||
function encryptToken(token: string): string {
|
||||
return sha256(token).toString()
|
||||
}
|
||||
|
||||
// Fetch stored token from localStorage and encrypt with SHA256
|
||||
export function getStoredToken(path: string): string | null {
|
||||
const storedToken =
|
||||
typeof window !== 'undefined' ? JSON.parse(localStorage.getItem(matchProtectedRoute(path)) as string) : ''
|
||||
return storedToken ? encryptToken(storedToken) : null
|
||||
}
|
||||
|
||||
/**
|
||||
* Compares the hash of .password and od-protected-token header
|
||||
* @param odTokenHeader od-protected-token header (sha256 hashed token)
|
||||
* @param dotPassword non-hashed .password file
|
||||
* @returns whether the two hashes are the same
|
||||
*/
|
||||
export function compareHashedToken({
|
||||
odTokenHeader,
|
||||
dotPassword,
|
||||
}: {
|
||||
odTokenHeader: string
|
||||
dotPassword: string
|
||||
}): boolean {
|
||||
return encryptToken(dotPassword.trim()) === odTokenHeader
|
||||
}
|
||||
/**
|
||||
* Match the specified route against a list of predefined routes
|
||||
* @param route directory path
|
||||
* @returns whether the directory is protected
|
||||
*/
|
||||
|
||||
export function matchProtectedRoute(route: string): string {
|
||||
const protectedRoutes: string[] = siteConfig.protectedRoutes
|
||||
let authTokenPath = ''
|
||||
|
||||
for (const r of protectedRoutes) {
|
||||
// protected route array could be empty
|
||||
if (r) {
|
||||
if (
|
||||
route.startsWith(
|
||||
r
|
||||
.split('/')
|
||||
.map(p => encodeURIComponent(p))
|
||||
.join('/')
|
||||
)
|
||||
) {
|
||||
authTokenPath = r
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return authTokenPath
|
||||
}
|
||||
Reference in New Issue
Block a user