@astii/monaco-copilot/esm/format.js

monaco-editor code copilot

function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
var FormatterCompletion = /*#__PURE__*/function () {
  function FormatterCompletion(model, position, range) {
    _classCallCheck(this, FormatterCompletion);
    _defineProperty(this, "originalCompletion", void 0);
    _defineProperty(this, "formattedCompletion", void 0);
    this.model = model;
    this.position = position;
    this.range = range;
  }
  _createClass(FormatterCompletion, [{
    key: "ignoreBlankLines",
    value: function ignoreBlankLines() {
      if (this.formattedCompletion.trimStart() === "" && this.originalCompletion !== "\n") {
        this.formattedCompletion = this.formattedCompletion.trim();
      }
      return this;
    }
  }, {
    key: "removeInvalidLineBreaks",
    value: function removeInvalidLineBreaks() {
      this.formattedCompletion = this.formattedCompletion.trimEnd();
      return this;
    }
  }, {
    key: "trimStart",
    value: function trimStart() {
      var firstNonSpaceIndex = this.formattedCompletion.search(/\S/);
      if (firstNonSpaceIndex > this.position.column - 1) {
        this.formattedCompletion = this.formattedCompletion.slice(firstNonSpaceIndex);
      }
      return this;
    }
  }, {
    key: "computeRange",
    value: function computeRange() {
      var lineCount = (this.formattedCompletion.match(/\n/g) || []).length;
      var lastLineColumnCount = this.getLastLineColumnCount();
      var charAfterCursor = this.getCharAfterCursor();
      return {
        startLineNumber: this.position.lineNumber,
        startColumn: this.position.column,
        endLineNumber: this.position.lineNumber + lineCount,
        endColumn: !this.formattedCompletion.includes(charAfterCursor) ? this.position.column : this.position.lineNumber === this.range.startLineNumber && lineCount === 0 ? this.position.column + lastLineColumnCount : lastLineColumnCount
      };
    }
  }, {
    key: "getCharAfterCursor",
    value: function getCharAfterCursor() {
      var line = this.model.getLineContent(this.position.lineNumber);
      return line[this.position.column - 1];
    }
  }, {
    key: "getLastLineColumnCount",
    value: function getLastLineColumnCount() {
      var lines = this.formattedCompletion.split("\n");
      return lines[lines.length - 1].length;
    }
  }, {
    key: "format",
    value: function format(code) {
      this.formattedCompletion = code;
      this.originalCompletion = code;
      this.ignoreBlankLines().removeInvalidLineBreaks().trimStart();
      return this.formattedCompletion;
    }
  }]);
  return FormatterCompletion;
}();
export var getFormatter = function getFormatter(model, position, range) {
  var formatter = new FormatterCompletion(model, position, range);
  return function (completions) {
    return completions.map(function (it) {
      return {
        insertText: formatter.format(it.code),
        range: range
      };
    });
  };
};