All files / app/assets/javascripts/ide/lib/common model.js

93.87% Statements 46/49
87.5% Branches 21/24
94.44% Functions 17/18
93.47% Lines 43/46

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137                  144x 144x 144x 144x 144x   144x                       144x 26x                 144x   144x 144x 144x   144x 144x 144x       246x       130x       441x       290x       33x       3x       2x       79x       11x                       167x       72x 72x 72x       161x   350x   4x 4x   161x   161x   24x             89x   89x 25x     89x   89x 89x 89x   89x      
import { editor as monacoEditor, Uri } from 'monaco-editor';
import { insertFinalNewline } from '~/lib/utils/text_utility';
import eventHub from '../../eventhub';
import { trimTrailingWhitespace } from '../../utils';
import { defaultModelOptions } from '../editor_options';
import Disposable from './disposable';
 
export default class Model {
  constructor(file, head = null) {
    this.disposable = new Disposable();
    this.file = file;
    this.head = head;
    this.content = file.content !== '' || file.deleted ? file.content : file.raw;
    this.options = { ...defaultModelOptions };
 
    this.disposable.add(
      (this.originalModel = monacoEditor.createModel(
        head ? head.content : this.file.raw,
        undefined,
        new Uri('gitlab', false, `original/${this.path}`),
      )),
      (this.model = monacoEditor.createModel(
        this.content,
        undefined,
        new Uri('gitlab', false, this.path),
      )),
    );
    if (this.file.mrChange) {
      this.disposable.add(
        (this.baseModel = monacoEditor.createModel(
          this.file.baseRaw,
          undefined,
          new Uri('gitlab', false, `target/${this.path}`),
        )),
      );
    }
 
    this.events = new Set();
 
    this.updateContent = this.updateContent.bind(this);
    this.updateNewContent = this.updateNewContent.bind(this);
    this.dispose = this.dispose.bind(this);
 
    eventHub.$on(`editor.update.model.dispose.${this.file.key}`, this.dispose);
    eventHub.$on(`editor.update.model.content.${this.file.key}`, this.updateContent);
    eventHub.$on(`editor.update.model.new.content.${this.file.key}`, this.updateNewContent);
  }
 
  get url() {
    return this.model.uri.toString();
  }
 
  get language() {
    return this.model.getLanguageId();
  }
 
  get path() {
    return this.file.key;
  }
 
  getModel() {
    return this.model;
  }
 
  getOriginalModel() {
    return this.originalModel;
  }
 
  getBaseModel() {
    return this.baseModel;
  }
 
  setValue(value) {
    this.getModel().setValue(value);
  }
 
  onChange(cb) {
    this.events.add(this.disposable.add(this.model.onDidChangeContent((e) => cb(this, e))));
  }
 
  onDispose(cb) {
    this.events.add(cb);
  }
 
  updateContent({ content, changed }) {
    this.getOriginalModel().setValue(content);
 
    if (!changed) {
      this.getModel().setValue(content);
    }
  }
 
  updateNewContent(content) {
    this.getModel().setValue(content);
  }
 
  updateOptions(obj = {}) {
    Object.assign(this.options, obj);
    this.model.updateOptions(obj);
    this.applyCustomOptions();
  }
 
  applyCustomOptions() {
    this.updateNewContent(
      Object.entries(this.options).reduce((content, [key, value]) => {
        switch (key) {
          case 'endOfLine':
            this.model.pushEOL(value);
            return this.model.getValue();
          case 'insertFinalNewline':
            return value ? insertFinalNewline(content) : content;
          case 'trimTrailingWhitespace':
            return value ? trimTrailingWhitespace(content) : content;
          default:
            return content;
        }
      }, this.model.getValue()),
    );
  }
 
  dispose() {
    if (!this.model.isDisposed()) this.applyCustomOptions();
 
    this.events.forEach((cb) => {
      if (typeof cb === 'function') cb();
    });
 
    this.events.clear();
 
    eventHub.$off(`editor.update.model.dispose.${this.file.key}`, this.dispose);
    eventHub.$off(`editor.update.model.content.${this.file.key}`, this.updateContent);
    eventHub.$off(`editor.update.model.new.content.${this.file.key}`, this.updateNewContent);
 
    this.disposable.dispose();
  }
}