@mieweb/wikigdrive
Version:
Google Drive to MarkDown synchronization
176 lines (175 loc) • 8.93 kB
JavaScript
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
return c > 3 && r && Object.defineProperty(target, key, r), r;
};
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';
export class GoogleDriveController extends Controller {
constructor(subPath, filesService) {
super(subPath);
Object.defineProperty(this, "filesService", {
enumerable: true,
configurable: true,
writable: true,
value: 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;
}
}
}
__decorate([
RouteGet('/:driveId/share')
], GoogleDriveController.prototype, "getShare", null);
__decorate([
RouteGet('/:driveId/upload')
], GoogleDriveController.prototype, "getUpload", null);
__decorate([
RouteGet('/:driveId/:fileId'),
RouteResponse('stream'),
RouteErrorHandler(new ShareErrorHandler())
], GoogleDriveController.prototype, "getDocs", null);