hedgedoc/lib/web/note/controller.ts

import {NextFunction, Response} from "express";
import {NoteUtils} from "./util";
import { ActionController } from "./actions";
import errors from "../../errors";
import config from "../../config";
import logger from "../../logger";
import { User } from "../../models/user";
import { Note } from "../../models/note";

export module NoteController {
  export function publishNoteActions(req: any, res: Response, next: NextFunction) {
    NoteUtils.findNoteOrCreate(req, res, function (note) {
      const action = req.params.action;
      switch (action) {
        case 'download':
          exports.downloadMarkdown(req, res, note);
          break;
        case 'edit':
          res.redirect(config.serverURL + '/' + (note.alias ? note.alias : Note.encodeNoteId(note.id)) + '?both');
          break;
        default:
          res.redirect(config.serverURL + '/s/' + note.shortid);
          break
      }
    })
  }

  export function showPublishNote(req: any, res: Response, next: NextFunction) {
    const include = [{
      model: User,
      as: 'owner'
    }, {
      model: User,
      as: 'lastchangeuser'
    }];
    NoteUtils.findNoteOrCreate(req, res, function (note) {
      // force to use short id
      const shortid = req.params.shortid;
      if ((note.alias && shortid !== note.alias) || (!note.alias && shortid !== note.shortid)) {
        return res.redirect(config.serverURL + '/s/' + (note.alias || note.shortid))
      }
      note.increment('viewcount').then(function (note) {
        if (!note) {
          return errors.errorNotFound(res)
        }
        NoteUtils.getPublishData(req, res, note, (data) => {
          res.set({
            'Cache-Control': 'private' // only cache by client
          });
          return res.render('pretty.ejs', data)
        })
      }).catch(function (err) {
        logger.error(err);
        return errors.errorInternalError(res)
      })
    }, include)
  }

  export function showNote(req: any, res: Response, next: NextFunction) {
    NoteUtils.findNoteOrCreate(req, res, function (note) {
      // force to use note id
      const noteId = req.params.noteId;
      const id = Note.encodeNoteId(note.id);
      if ((note.alias && noteId !== note.alias) || (!note.alias && noteId !== id)) {
        return res.redirect(config.serverURL + '/' + (note.alias || id))
      }
      const body = note.content;
      const extracted = Note.extractMeta(body);
      const meta = Note.parseMeta(extracted.meta);
      let title = Note.decodeTitle(note.title);
      title = Note.generateWebTitle(meta.title || title);
      const opengraph = Note.parseOpengraph(meta, title);
      res.set({
        'Cache-Control': 'private', // only cache by client
        'X-Robots-Tag': 'noindex, nofollow' // prevent crawling
      });
      return res.render('codimd.ejs', {
        title: title,
        opengraph: opengraph
      })
    })
  }

  export function createFromPOST(req: any, res: Response, next: NextFunction) {
    let body = '';
    if (req.body && req.body.length > config.documentMaxLength) {
      return errors.errorTooLong(res)
    } else if (req.body) {
      body = req.body
    }
    body = body.replace(/[\r]/g, '');
    return NoteUtils.newNote(req, res, body)
  }

  export function doAction(req: any, res: Response, next: NextFunction) {
    const noteId = req.params.noteId;
    NoteUtils.findNoteOrCreate(req, res, (note) => {
      const action = req.params.action;
      // TODO: Don't switch on action, choose action in Router and use separate functions
      switch (action) {
        case 'publish':
        case 'pretty': // pretty deprecated
          res.redirect(config.serverURL + '/s/' + (note.alias || note.shortid));
          break;
        case 'slide':
          res.redirect(config.serverURL + '/p/' + (note.alias || note.shortid));
          break;
        case 'download':
          exports.downloadMarkdown(req, res, note);
          break;
        case 'info':
          ActionController.getInfo(req, res, note);
          break;
        case 'gist':
          ActionController.createGist(req, res, note);
          break;
        case 'revision':
          ActionController.getRevision(req, res, note);
          break;
        default:
          return res.redirect(config.serverURL + '/' + noteId)
      }
    })
  }

  export function downloadMarkdown(req: Request, res: Response, note: any) {
    const body = note.content;
    let filename = Note.decodeTitle(note.title);
    filename = encodeURIComponent(filename);
    res.set({
      'Access-Control-Allow-Origin': '*', // allow CORS as API
      'Access-Control-Allow-Headers': 'Range',
      'Access-Control-Expose-Headers': 'Cache-Control, Content-Encoding, Content-Range',
      'Content-Type': 'text/markdown; charset=UTF-8',
      'Cache-Control': 'private',
      'Content-disposition': 'attachment; filename=' + filename + '.md',
      'X-Robots-Tag': 'noindex, nofollow' // prevent crawling
    });
    res.send(body)
  }
}