const indexRoute = require('./index'); const loginRoute = require('./login'); const logoutRoute = require('./logout'); const migrationRoute = require('./migration'); const setupRoute = require('./setup'); const multer = require('multer')(); // API routes const treeApiRoute = require('./api/tree'); const notesApiRoute = require('./api/notes'); const treeChangesApiRoute = require('./api/tree_changes'); const cloningApiRoute = require('./api/cloning'); const noteRevisionsApiRoute = require('./api/note_revisions'); const recentChangesApiRoute = require('./api/recent_changes'); const settingsApiRoute = require('./api/settings'); const passwordApiRoute = require('./api/password'); const migrationApiRoute = require('./api/migration'); const syncApiRoute = require('./api/sync'); const loginApiRoute = require('./api/login'); const eventLogRoute = require('./api/event_log'); const recentNotesRoute = require('./api/recent_notes'); const appInfoRoute = require('./api/app_info'); const exportRoute = require('./api/export'); const importRoute = require('./api/import'); const setupApiRoute = require('./api/setup'); const sqlRoute = require('./api/sql'); const anonymizationRoute = require('./api/anonymization'); const cleanupRoute = require('./api/cleanup'); const imageRoute = require('./api/image'); const labelsRoute = require('./api/labels'); const scriptRoute = require('./api/script'); const senderRoute = require('./api/sender'); const filesRoute = require('./api/file_upload'); const searchRoute = require('./api/search'); const log = require('../services/log'); const express = require('express'); const router = express.Router(); const auth = require('../services/auth'); const cls = require('../services/cls'); const sql = require('../services/sql'); function apiResultHandler(res, result) { // if it's an array and first element is integer then we consider this to be [statusCode, response] format if (Array.isArray(result) && result.length > 0 && Number.isInteger(result[0])) { const [statusCode, response] = result; res.status(statusCode).send(response); if (statusCode !== 200) { log.info(`${method} ${path} returned ${statusCode} with response ${JSON.stringify(response)}`); } } else if (result === undefined) { res.status(200).send(); } else { res.status(200).send(result); } } function apiRoute(method, path, routeHandler) { route(method, path, [auth.checkApiAuth], routeHandler, apiResultHandler); } // API routes requiring HTTP protocol. This means we ignore route return value and make an electron auth exception function httpApiRoute(method, path, routeHandler) { route(method, path, [auth.checkApiAuth, multer.single('upload')], routeHandler); } function route(method, path, middleware, routeHandler, resultHandler) { router[method](path, ...middleware, async (req, res, next) => { try { const result = await cls.init(async () => { cls.namespace.set('sourceId', req.headers.source_id); return await sql.doInTransaction(async () => { return await routeHandler(req, res, next); }); }); if (resultHandler) { resultHandler(res, result); } } catch (e) { log.info(`${method} ${path} threw exception: ` + e.stack); res.sendStatus(500); } }); } const GET = 'get', POST = 'post', PUT = 'put', DELETE = 'delete'; const uploadMiddleware = multer.single('upload'); function register(app) { app.use('/', indexRoute); app.use('/login', loginRoute); app.use('/logout', logoutRoute); app.use('/migration', migrationRoute); app.use('/setup', setupRoute); apiRoute(GET, '/api/tree', treeApiRoute.getTree); apiRoute(PUT, '/api/tree/:branchId/set-prefix', treeApiRoute.setPrefix); apiRoute(PUT, '/api/tree/:branchId/move-to/:parentNoteId', treeChangesApiRoute.moveBranchToParent); apiRoute(PUT, '/api/tree/:branchId/move-before/:beforeBranchId', treeChangesApiRoute.moveBranchBeforeNote); apiRoute(PUT, '/api/tree/:branchId/move-after/:afterBranchId', treeChangesApiRoute.moveBranchAfterNote); apiRoute(PUT, '/api/tree/:branchId/expanded/:expanded', treeChangesApiRoute.setExpanded); apiRoute(DELETE, '/api/tree/:branchId', treeChangesApiRoute.deleteBranch); apiRoute(GET, '/api/notes/:noteId', notesApiRoute.getNote); apiRoute(PUT, '/api/notes/:noteId', notesApiRoute.updateNote); apiRoute(POST, '/api/notes/:parentNoteId/children', notesApiRoute.createNote); apiRoute(PUT, '/api/notes/:noteId/sort', notesApiRoute.sortNotes); apiRoute(PUT, '/api/notes/:noteId/protect-sub-tree/:isProtected', notesApiRoute.protectBranch); apiRoute(PUT, /\/api\/notes\/(.*)\/type\/(.*)\/mime\/(.*)/, notesApiRoute.setNoteTypeMime); apiRoute(PUT, '/api/notes/:childNoteId/clone-to/:parentNoteId', cloningApiRoute.cloneNoteToParent); apiRoute(PUT, '/api/notes/:noteId/clone-after/:afterBranchId', cloningApiRoute.cloneNoteAfter); apiRoute(GET, '/api/notes/:noteId/labels', labelsRoute.getNoteLabels); apiRoute(PUT, '/api/notes/:noteId/labels', labelsRoute.updateNoteLabels); apiRoute(GET, '/api/labels/names', labelsRoute.getAllLabelNames); apiRoute(GET, '/api/labels/values/:labelName', labelsRoute.getValuesForLabel); apiRoute(GET, '/api/note-revisions/:noteId', noteRevisionsApiRoute.getNoteRevisions); apiRoute(GET, '/api/recent-changes', recentChangesApiRoute.getRecentChanges); apiRoute(GET, '/api/settings', settingsApiRoute.getAllowedSettings); apiRoute(GET, '/api/settings/all', settingsApiRoute.getAllSettings); apiRoute(POST, '/api/settings', settingsApiRoute.updateSetting); apiRoute(POST, '/api/password/change', passwordApiRoute.changePassword); apiRoute(GET, '/api/sync/check', syncApiRoute.checkSync); apiRoute(POST, '/api/sync/now', syncApiRoute.syncNow); apiRoute(POST, '/api/sync/fill-sync-rows', syncApiRoute.fillSyncRows); apiRoute(POST, '/api/sync/force-full-sync', syncApiRoute.forceFullSync); apiRoute(POST, '/api/sync/force-note-sync/:noteId', syncApiRoute.forceNoteSync); apiRoute(GET, '/api/sync/changed', syncApiRoute.getChanged); apiRoute(GET, '/api/sync/notes/:noteId', syncApiRoute.getNote); apiRoute(GET, '/api/sync/branches/:branchId', syncApiRoute.getBranch); apiRoute(GET, '/api/sync/note_revisions/:noteRevisionId', syncApiRoute.getNoteRevision); apiRoute(GET, '/api/sync/options/:name', syncApiRoute.getOption); apiRoute(GET, '/api/sync/note_reordering/:parentNoteId', syncApiRoute.getNoteReordering); apiRoute(GET, '/api/sync/recent_notes/:branchId', syncApiRoute.getRecentNote); apiRoute(GET, '/api/sync/images/:imageId', syncApiRoute.getImage); apiRoute(GET, '/api/sync/note_images/:noteImageId', syncApiRoute.getNoteImage); apiRoute(GET, '/api/sync/labels/:labelId', syncApiRoute.getLabel); apiRoute(GET, '/api/sync/api_tokens/:apiTokenId', syncApiRoute.getApiToken); apiRoute(PUT, '/api/sync/notes', syncApiRoute.updateNote); apiRoute(PUT, '/api/sync/note_revisions', syncApiRoute.updateNoteRevision); apiRoute(PUT, '/api/sync/note_reordering', syncApiRoute.updateNoteReordering); apiRoute(PUT, '/api/sync/options', syncApiRoute.updateOption); apiRoute(PUT, '/api/sync/recent_notes', syncApiRoute.updateRecentNote); apiRoute(PUT, '/api/sync/images', syncApiRoute.updateImage); apiRoute(PUT, '/api/sync/note_images', syncApiRoute.updateNoteImage); apiRoute(PUT, '/api/sync/labels', syncApiRoute.updateLabel); apiRoute(PUT, '/api/sync/api_tokens', syncApiRoute.updateApiToken); apiRoute(GET, '/api/event-log', eventLogRoute.getEventLog); apiRoute(GET, '/api/recent-notes', recentNotesRoute.getRecentNotes); apiRoute(PUT, '/api/recent-notes/:branchId/:notePath', recentNotesRoute.addRecentNote); apiRoute(GET, '/api/app-info', appInfoRoute.getAppInfo); httpApiRoute(GET, '/api/export/:noteId', exportRoute.exportNote); httpApiRoute(POST, '/api/import/:parentNoteId', importRoute.importTar); route(POST, '/api/setup', [auth.checkAppNotInitialized], setupApiRoute.setup, apiResultHandler); apiRoute(POST, '/api/sql/execute', sqlRoute.execute); apiRoute(POST, '/api/anonymization/anonymize', anonymizationRoute.anonymize); apiRoute(POST, '/api/cleanup/cleanup-soft-deleted-items', cleanupRoute.cleanupSoftDeletedItems); apiRoute(POST, '/api/cleanup/cleanup-unused-images', cleanupRoute.cleanupUnusedImages); apiRoute(POST, '/api/cleanup/vacuum-database', cleanupRoute.vacuumDatabase); httpApiRoute(GET, '/api/images/:imageId/:filename', imageRoute.returnImage); httpApiRoute(POST, '/api/images', imageRoute.uploadImage); apiRoute(POST, '/api/script/exec', scriptRoute.exec); apiRoute(POST, '/api/script/run/:noteId', scriptRoute.run); apiRoute(GET, '/api/script/startup', scriptRoute.getStartupBundles); apiRoute(GET, '/api/script/bundle/:noteId', scriptRoute.getBundle); route(POST, '/api/sender/login', [], senderRoute.login, apiResultHandler); route(POST, '/api/sender/image', [auth.checkSenderToken], senderRoute.uploadImage, apiResultHandler); route(POST, '/api/sender/note', [auth.checkSenderToken], senderRoute.saveNote, apiResultHandler); route(POST, '/api/files/upload/:parentNoteId', [auth.checkApiAuthOrElectron, uploadMiddleware], filesRoute.uploadFile, apiResultHandler); route(GET, '/api/files/download/:noteId', [auth.checkApiAuthOrElectron], filesRoute.downloadFile); apiRoute(GET, '/api/search/:searchString', searchRoute.searchNotes); apiRoute(POST, '/api/search/:searchString', searchRoute.saveSearchToNote); app.use('', router); app.use('/api/migration', migrationApiRoute); app.use('/api/login', loginApiRoute); } module.exports = { register };