83 lines
2 KiB
TypeScript
83 lines
2 KiB
TypeScript
/*
|
|
* SPDX-FileCopyrightText: syuilo and other misskey contributors
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
import * as fs from 'node:fs';
|
|
import * as Path from 'node:path';
|
|
import { fileURLToPath } from 'node:url';
|
|
import { dirname } from 'node:path';
|
|
import { Inject, Injectable } from '@nestjs/common';
|
|
import { DI } from '@/di-symbols.js';
|
|
import type { Config } from '@/config.js';
|
|
import { bindThis } from '@/decorators.js';
|
|
|
|
const _filename = fileURLToPath(import.meta.url);
|
|
const _dirname = dirname(_filename);
|
|
|
|
const path = Path.resolve(_dirname, '../../../../files');
|
|
const cachePath = Path.resolve(_dirname, '../../../../cache');
|
|
|
|
@Injectable()
|
|
export class InternalStorageService {
|
|
constructor(
|
|
@Inject(DI.config)
|
|
private config: Config,
|
|
) {
|
|
}
|
|
|
|
@bindThis
|
|
public resolvePath(key: string) {
|
|
return Path.resolve(path, key);
|
|
}
|
|
|
|
@bindThis
|
|
public resolveCachePath(key: string) {
|
|
return Path.resolve(cachePath, key);
|
|
}
|
|
|
|
@bindThis
|
|
public existsCache(key: string) {
|
|
return fs.existsSync(this.resolveCachePath(key));
|
|
}
|
|
|
|
@bindThis
|
|
public read(key: string) {
|
|
return fs.createReadStream(this.resolvePath(key));
|
|
}
|
|
|
|
@bindThis
|
|
public readCache(key: string) {
|
|
return fs.createReadStream(this.resolveCachePath(key));
|
|
}
|
|
|
|
@bindThis
|
|
public saveFromPath(key: string, srcPath: string) {
|
|
fs.mkdirSync(path, { recursive: true });
|
|
fs.copyFileSync(srcPath, this.resolvePath(key));
|
|
return `${this.config.url}/files/${key}`;
|
|
}
|
|
|
|
@bindThis
|
|
public saveFromBuffer(key: string, data: Buffer) {
|
|
fs.mkdirSync(path, { recursive: true });
|
|
fs.writeFileSync(this.resolvePath(key), data);
|
|
return `${this.config.url}/files/${key}`;
|
|
}
|
|
|
|
@bindThis
|
|
public saveCacheFromBuffer(key: string, data: Buffer) {
|
|
fs.mkdirSync(cachePath, { recursive: true });
|
|
fs.writeFileSync(this.resolveCachePath(key), data);
|
|
}
|
|
|
|
@bindThis
|
|
public del(key: string) {
|
|
fs.unlink(this.resolvePath(key), () => {});
|
|
}
|
|
|
|
@bindThis
|
|
public delCache(key: string) {
|
|
fs.unlink(this.resolveCachePath(key), () => {});
|
|
}
|
|
}
|