@mieweb/wikigdrive
Version:
Google Drive to MarkDown synchronization
213 lines (212 loc) • 12.9 kB
JavaScript
var __runInitializers = (this && this.__runInitializers) || function (thisArg, initializers, value) {
var useValue = arguments.length > 2;
for (var i = 0; i < initializers.length; i++) {
value = useValue ? initializers[i].call(thisArg, value) : initializers[i].call(thisArg);
}
return useValue ? value : void 0;
};
var __esDecorate = (this && this.__esDecorate) || function (ctor, descriptorIn, decorators, contextIn, initializers, extraInitializers) {
function accept(f) { if (f !== void 0 && typeof f !== "function") throw new TypeError("Function expected"); return f; }
var kind = contextIn.kind, key = kind === "getter" ? "get" : kind === "setter" ? "set" : "value";
var target = !descriptorIn && ctor ? contextIn["static"] ? ctor : ctor.prototype : null;
var descriptor = descriptorIn || (target ? Object.getOwnPropertyDescriptor(target, contextIn.name) : {});
var _, done = false;
for (var i = decorators.length - 1; i >= 0; i--) {
var context = {};
for (var p in contextIn) context[p] = p === "access" ? {} : contextIn[p];
for (var p in contextIn.access) context.access[p] = contextIn.access[p];
context.addInitializer = function (f) { if (done) throw new TypeError("Cannot add initializers after decoration has completed"); extraInitializers.push(accept(f || null)); };
var result = (0, decorators[i])(kind === "accessor" ? { get: descriptor.get, set: descriptor.set } : descriptor[key], context);
if (kind === "accessor") {
if (result === void 0) continue;
if (result === null || typeof result !== "object") throw new TypeError("Object expected");
if (_ = accept(result.get)) descriptor.get = _;
if (_ = accept(result.set)) descriptor.set = _;
if (_ = accept(result.init)) initializers.unshift(_);
}
else if (_ = accept(result)) {
if (kind === "field") initializers.unshift(_);
else descriptor[key] = _;
}
}
if (target) Object.defineProperty(target, contextIn.name, descriptor);
done = true;
};
import process from 'node:process';
import { Controller, RouteErrorHandler, RouteGet, RouteResponse } from './Controller.js';
import { convertToPreviewUrl, getCachedChanges, outputDirectory, ShareErrorHandler } from './FolderController.js';
import { UserConfigService } from '../../google_folder/UserConfigService.js';
import { MarkdownTreeProcessor } from '../../transform/MarkdownTreeProcessor.js';
import { getContentFileService } from '../../transform/utils.js';
import { GoogleTreeProcessor } from '../../google_folder/GoogleTreeProcessor.js';
import { UserAuthClient } from '../../../google/AuthClient.js';
import { filterParams } from '../../../google/driveFetch.js';
import { GoogleDriveService } from '../../../google/GoogleDriveService.js';
import { redirError } from '../auth.js';
import { LocalLog } from '../../transform/LocalLog.js';
import { MimeTypes } from '../../../model/GoogleFile.js';
let GoogleDriveController = (() => {
var _a;
let _classSuper = Controller;
let _instanceExtraInitializers = [];
let _getShare_decorators;
let _getUpload_decorators;
let _getDocs_decorators;
return _a = class GoogleDriveController extends _classSuper {
constructor(subPath, filesService) {
super(subPath);
Object.defineProperty(this, "filesService", {
enumerable: true,
configurable: true,
writable: true,
value: (__runInitializers(this, _instanceExtraInitializers), filesService)
});
}
async getShare(ctx) {
const driveId = await ctx.routeParamPath('driveId');
const serverUrl = process.env.AUTH_DOMAIN || process.env.DOMAIN;
const state = new URLSearchParams(filterParams({
shareDrive: 1,
driveId: driveId !== 'none' ? (driveId || '') : '',
instance: process.env.AUTH_INSTANCE
})).toString();
const authClient = new UserAuthClient(process.env.GOOGLE_AUTH_CLIENT_ID, process.env.GOOGLE_AUTH_CLIENT_SECRET);
const shareUrl = await authClient.getWebDriveShareUrl(`${serverUrl}/auth`, state);
if (process.env.VERSION === 'dev') {
console.debug(shareUrl);
}
return { shareUrl };
}
async getUpload(ctx) {
const driveId = await ctx.routeParamPath('driveId');
const serverUrl = process.env.AUTH_DOMAIN || process.env.DOMAIN;
const state = new URLSearchParams(filterParams({
uploadDrive: 1,
driveId: driveId !== 'none' ? (driveId || '') : '',
instance: process.env.AUTH_INSTANCE
})).toString();
const authClient = new UserAuthClient(process.env.GOOGLE_AUTH_CLIENT_ID, process.env.GOOGLE_AUTH_CLIENT_SECRET);
const shareUrl = await authClient.getUploadDriveUrl(`${serverUrl}/auth`, state);
if (process.env.VERSION === 'dev') {
console.debug(shareUrl);
}
return { shareUrl };
}
async getDocs(ctx) {
const driveId = await ctx.routeParamPath('driveId');
const fileId = await ctx.routeParamPath('fileId');
const user = await ctx.routeParamUser();
if (!user?.google_access_token) {
throw redirError(ctx.req, 'Not authenticated');
}
const googleFileSystem = await this.filesService.getSubFileService(driveId, '/');
const userConfigService = new UserConfigService(googleFileSystem);
await userConfigService.load();
const transformedFileSystem = await this.filesService.getSubFileService(driveId + '_transform', '');
const contentFileService = await getContentFileService(transformedFileSystem, userConfigService);
const localLog = new LocalLog(contentFileService);
await localLog.load();
const logRow = localLog.findLastFile(fileId);
const markdownTreeProcessor = new MarkdownTreeProcessor(contentFileService);
await markdownTreeProcessor.load();
const [foundTreeItem] = await markdownTreeProcessor.findByPath(logRow?.filePath);
const contentDir = (userConfigService.config.transform_subdir || '').startsWith('/') ? userConfigService.config.transform_subdir : undefined;
ctx.res.setHeader('wgd-content-dir', contentDir || '');
ctx.res.setHeader('wgd-google-id', fileId);
if (!foundTreeItem) {
const googleDriveService = new GoogleDriveService(ctx.logger, null);
const auth = {
async getAccessToken() {
return user.google_access_token;
}
};
try {
const file = await googleDriveService.getFile(auth, fileId);
if (file) {
ctx.res.setHeader('wgd-google-parent-id', file.parentId || '');
ctx.res.setHeader('wgd-google-version', file.version || '');
ctx.res.setHeader('wgd-google-modified-time', file.modifiedTime || '');
ctx.res.setHeader('wgd-mime-type', file.mimeType || '');
ctx.res.setHeader('wgd-last-author', file.lastAuthor || '');
ctx.res.setHeader('Content-type', file.mimeType);
ctx.res.send('Not synced');
return;
}
}
catch (err) {
if (err.status === 401) {
throw redirError(ctx.req, err.message);
}
}
ctx.res.status(404).send({ message: 'No local' });
return;
}
const treeItem = convertToPreviewUrl(userConfigService.config.preview_rewrite_rule, driveId)(foundTreeItem);
ctx.res.setHeader('wgd-google-parent-id', treeItem.parentId || '');
ctx.res.setHeader('wgd-google-version', treeItem.version || '');
ctx.res.setHeader('wgd-google-modified-time', treeItem.modifiedTime || '');
ctx.res.setHeader('wgd-path', treeItem.path || '');
ctx.res.setHeader('wgd-file-name', treeItem.fileName || '');
ctx.res.setHeader('wgd-mime-type', treeItem.mimeType || '');
ctx.res.setHeader('wgd-preview-url', treeItem.previewUrl || '');
ctx.res.setHeader('wgd-last-author', treeItem.lastAuthor || '');
if (!contentDir) {
ctx.res.status(404).send('Content subdirectory must be set and start with /');
return;
}
const treeProcessor = new GoogleTreeProcessor(googleFileSystem);
await treeProcessor.load();
const [leaf] = await treeProcessor.findById(foundTreeItem.id);
if (leaf) {
ctx.res.setHeader('wgd-synced-version', leaf.version || '');
ctx.res.setHeader('wgd-synced-modified-time', leaf.modifiedTime || '');
}
const changes = await getCachedChanges(ctx.logger, transformedFileSystem, contentFileService, googleFileSystem);
const change = changes.find(change => change.path === (contentDir.substring(1) + treeItem.path).replace(/^\//, ''));
if (change) {
if (change.state.isNew) {
treeItem['status'] = 'N';
}
else if (change.state.isModified) {
treeItem['status'] = 'M';
}
else if (change.state.isDeleted) {
treeItem['status'] = 'D';
}
ctx.res.setHeader('wgd-git-attachments', String(change.attachments) || '0');
}
ctx.res.setHeader('wgd-git-status', treeItem['status'] || '');
const filePath = contentDir + treeItem.path;
if (!await transformedFileSystem.exists(filePath)) {
ctx.res.status(404).send('Not exist in transformedFileSystem');
return;
}
if (await transformedFileSystem.isDirectory(filePath)) {
const treeItems = outputDirectory(treeItem);
ctx.res.setHeader('content-type', MimeTypes.FOLDER_MIME);
ctx.res.send(JSON.stringify(treeItems));
return;
}
else {
if (treeItem.mimeType) {
ctx.res.setHeader('Content-type', treeItem.mimeType);
}
const buffer = await transformedFileSystem.readBuffer(filePath);
ctx.res.send(buffer);
return;
}
}
},
(() => {
const _metadata = typeof Symbol === "function" && Symbol.metadata ? Object.create(_classSuper[Symbol.metadata] ?? null) : void 0;
_getShare_decorators = [RouteGet('/:driveId/share')];
_getUpload_decorators = [RouteGet('/:driveId/upload')];
_getDocs_decorators = [RouteGet('/:driveId/:fileId'), RouteResponse('stream'), RouteErrorHandler(new ShareErrorHandler())];
__esDecorate(_a, null, _getShare_decorators, { kind: "method", name: "getShare", static: false, private: false, access: { has: obj => "getShare" in obj, get: obj => obj.getShare }, metadata: _metadata }, null, _instanceExtraInitializers);
__esDecorate(_a, null, _getUpload_decorators, { kind: "method", name: "getUpload", static: false, private: false, access: { has: obj => "getUpload" in obj, get: obj => obj.getUpload }, metadata: _metadata }, null, _instanceExtraInitializers);
__esDecorate(_a, null, _getDocs_decorators, { kind: "method", name: "getDocs", static: false, private: false, access: { has: obj => "getDocs" in obj, get: obj => obj.getDocs }, metadata: _metadata }, null, _instanceExtraInitializers);
if (_metadata) Object.defineProperty(_a, Symbol.metadata, { enumerable: true, configurable: true, writable: true, value: _metadata });
})(),
_a;
})();
export { GoogleDriveController };