|
| 1 | +/* eslint-disable no-await-in-loop */ |
| 2 | +import { resolve } from 'path'; |
| 3 | +import { |
| 4 | + existsSync, mkdir, readdir, rename, rmdir, |
| 5 | +} from 'fs-extra'; |
| 6 | +import Schema from 'schemastery'; |
| 7 | +import { Context } from '../context'; |
| 8 | + |
| 9 | +export const apply = (ctx: Context) => ctx.addScript( |
| 10 | + 'fixStorage', 'Rename all files to lowercase', |
| 11 | + Schema.object({}), |
| 12 | + async () => { |
| 13 | + const config = await ctx.get('storage').status(); |
| 14 | + console.log(config); |
| 15 | + if (config.type !== 'Local') throw new Error('Only local storage is supported'); |
| 16 | + const dir = (config as any).dir; |
| 17 | + if (!dir) throw new Error('Storage dir not found'); |
| 18 | + const files = await readdir(dir); |
| 19 | + for (const file of files) { |
| 20 | + let inner = await readdir(resolve(dir, file)); |
| 21 | + if (file.toLowerCase() !== file && !existsSync(resolve(dir, file.toLowerCase()))) { |
| 22 | + await mkdir(resolve(dir, file.toLowerCase())); |
| 23 | + console.log('+: ', resolve(dir, file.toLowerCase())); |
| 24 | + } |
| 25 | + for (const f of inner) { |
| 26 | + if (resolve(dir, file, f) === resolve(dir, file.toLowerCase(), f.toLowerCase())) continue; |
| 27 | + await rename(resolve(dir, file, f), resolve(dir, file.toLowerCase(), f.toLowerCase())); |
| 28 | + console.log('R: ', resolve(dir, file, f), '->', resolve(dir, file.toLowerCase(), f.toLowerCase())); |
| 29 | + } |
| 30 | + inner = await readdir(resolve(dir, file)); |
| 31 | + if (!inner.length) { |
| 32 | + await rmdir(resolve(dir, file)); |
| 33 | + console.log('-: ', resolve(dir, file)); |
| 34 | + } |
| 35 | + } |
| 36 | + const coll = ctx.get('db').collection('storage'); |
| 37 | + while (true) { |
| 38 | + let hasItem = false; |
| 39 | + for await (const t of coll.find({ $or: [{ _id: /[A-Z]/ }, { link: /[A-Z]/ }] })) { |
| 40 | + hasItem = true; |
| 41 | + await coll.insertOne({ |
| 42 | + ...t, |
| 43 | + _id: t._id.toLowerCase(), |
| 44 | + ...(t.link ? { link: t.link.toLowerCase() } : {}), |
| 45 | + }); |
| 46 | + await coll.deleteOne({ _id: t._id }); |
| 47 | + console.log('Q: ', t._id, '->', t._id.toLowerCase()); |
| 48 | + } |
| 49 | + if (!hasItem) break; |
| 50 | + } |
| 51 | + return true; |
| 52 | + }, |
| 53 | +); |
0 commit comments