filestash/client/pages/viewerpage/editor/orgmode.js

import 'codemirror/addon/mode/simple';

CodeMirror.__mode = 'orgmode';

CodeMirror.defineSimpleMode("orgmode", {
    start: [
        {regex: /^(^\*{1,6}\s)(TODO|DOING|WAITING){0,1}(CANCEL|DEFERRED|DONE){0,1}(.*)$/, token: ["header org-level-star", "header org-todo", "header org-done", "header"]},
        {regex: /(^\+[^\/]*\+)/, token: ["strikethrough"]},
        {regex: /(^\*[^\/]*\*)/, token: ["strong"]},
        {regex: /(^\/[^\/]*\/)/, token: ["em"]},
        {regex: /(^\_[^\/]*\_)/, token: ["link"]},
        {regex: /(^\~[^\/]*\~)/, token: ["comment"]},
        {regex: /(^\=[^\/]*\=)/, token: ["comment"]},
        // special syntax
        //{regex: /(^[\*]+)(\s[TODO|NEXT|DONE|DEFERRED|REJECTED|WAITING]{2,})?(.*)/, token: ['comment', 'qualifier', 'header']}, // headline
        {regex: /\[\[[^\[\]]*\]\[[^\[\]]*\]\]/, token: "url"}, // links
        {regex: /\[[xX\s]?\]/, token: 'qualifier'}, // checkbox
        {regex: /\#\+BEGIN_[A-Z]*/, token: "comment", next: "env"}, // comments
        {regex: /:?[A-Z_]+\:.*/, token: "comment"}, // property drawers
        {regex: /(\#\+[A-Z_]*)(\:.*)/, token: ["keyword", 'qualifier']}, // environments
        {regex: /(CLOCK\:|SHEDULED\:)(\s.+)/, token: ["comment", "keyword"]}
    ],
    env: [
        {regex: /.*?\#\+END_[A-Z]*/, token: "comment", next: "start"},
        {regex: /.*/, token: "comment"}
    ]
});
CodeMirror.registerHelper("fold", "orgmode", function(cm, start) {
    // init
    const levelToMatch = headerLevel(start.line);

    // no folding needed
    if(levelToMatch === null) return;

    // find folding limits
    const lastLine = cm.lastLine();
    let end = start.line;
    while(end < lastLine){
        end += 1;
        let level = headerLevel(end);
        if(level && level <= levelToMatch) {
            end = end - 1;
            break;
        };
    }

    return {
        from: CodeMirror.Pos(start.line, cm.getLine(start.line).length),
        to: CodeMirror.Pos(end, cm.getLine(end).length)
    };

    function headerLevel(lineNo) {
        var line = cm.getLine(lineNo);
        var match = /^\*+/.exec(line);
        if(match && match.length === 1 && /header/.test(cm.getTokenTypeAt(CodeMirror.Pos(lineNo, 0)))){
            return match[0].length;
        }
        return null;
    }
});
CodeMirror.registerGlobalHelper("fold", "drawer", function(mode) {
    return mode.name === 'orgmode' ? true : false;
}, function(cm, start) {
    const drawer = isBeginningOfADrawer(start.line);
    if(drawer === false) return;

    // find folding limits
    const lastLine = cm.lastLine();
    let end = start.line;
    while(end < lastLine){
        end += 1;
        if(isEndOfADrawer(end)){
            break
        }
    }
    return {
        from: CodeMirror.Pos(start.line, cm.getLine(start.line).length),
        to: CodeMirror.Pos(end, cm.getLine(end).length)
    };

    function isBeginningOfADrawer(lineNo) {
        var line = cm.getLine(lineNo);
        var match = /^\:.*\:$/.exec(line);
        if(match && match.length === 1 && match[0] !== ':END:'){
            return true;
        }
        return false;
    }
    function isEndOfADrawer(lineNo){
        var line = cm.getLine(lineNo);
        return line.trim() === ':END:' ? true : false;
    }
});

CodeMirror.afterInit = function(editor){
    let state = {
        stab: 'OVERVIEW'
    };
    editor.setOption("extraKeys", {
        "Tab": function(cm) {
            let pos = cm.getCursor();
            isFold(cm, pos) ? unfold(cm, pos) : fold(cm, pos);
        },
        "Shift-Tab": function(cm){
            if(state.stab === "SHOW_ALL"){
                // fold everything that can be fold
                state.stab = 'OVERVIEW';
                cm.operation(function() {
                    for (var i = cm.firstLine(), e = cm.lastLine(); i <= e; i++){
                        fold(cm, CodeMirror.Pos(i, 0));
                    }
                });
            }else{
                // unfold all headers
                state.stab = 'SHOW_ALL';
                cm.operation(function() {
                    for (var i = cm.firstLine(), e = cm.lastLine(); i <= e; i++){
                        if(/header/.test(cm.getTokenTypeAt(CodeMirror.Pos(i, 0))) === true){
                            unfold(cm, CodeMirror.Pos(i, 0))
                        }
                    }
                });
            }
        }
    });

    function fold(cm, start){
        cm.foldCode(start, null, "fold");
    }
    function unfold(cm, start){
        cm.foldCode(start, null, "unfold");
    }
    function isFold(cm, start){
        const line = start.line;
        const marks = cm.findMarks(CodeMirror.Pos(line, 0), CodeMirror.Pos(line + 1, 0));
        for (let i = 0; i < marks.length; ++i)
            if (marks[i].__isFold && marks[i].find().from.line == line) return marks[i];
        return false;
    }
    editor.on('touchstart', function(cm, e){
        setTimeout(() => {
            isFold(cm, cm.getCursor()) ? unfold(cm, cm.getCursor()) : fold(cm, cm.getCursor())
        }, 150);
    });
    // fold everything except headers by default
    editor.operation(function() {
        for (var i = 0; i < editor.lineCount() ; i++) {
            if(/header/.test(editor.getTokenTypeAt(CodeMirror.Pos(i, 0))) === false){
                fold(editor, CodeMirror.Pos(i, 0));
            }
        }
    });

    function collapseWidget(){
        let $widget = document.createElement('span');
        $widget.appendChild(document.createTextNode('colapse'));
        return $widget;
    }
    function expandWidget(){
        let $widget = document.createElement('span');
        $widget.appendChild(document.createTextNode('expand'));
        return $widget;
    }
}

export default CodeMirror;