foldcode.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. (function(mod) {
  2. if (typeof exports == "object" && typeof module == "object") // CommonJS
  3. mod(require("../../lib/codemirror"));
  4. else if (typeof define == "function" && define.amd) // AMD
  5. define(["../../lib/codemirror"], mod);
  6. else // Plain browser env
  7. mod(CodeMirror);
  8. })(function(CodeMirror) {
  9. "use strict";
  10. function doFold(cm, pos, options, force) {
  11. var finder = options && (options.call ? options : options.rangeFinder);
  12. if (!finder) finder = CodeMirror.fold.auto;
  13. if (typeof pos == "number") pos = CodeMirror.Pos(pos, 0);
  14. var minSize = options && options.minFoldSize || 0;
  15. function getRange(allowFolded) {
  16. var range = finder(cm, pos);
  17. if (!range || range.to.line - range.from.line < minSize) return null;
  18. var marks = cm.findMarksAt(range.from);
  19. for (var i = 0; i < marks.length; ++i) {
  20. if (marks[i].__isFold && force !== "fold") {
  21. if (!allowFolded) return null;
  22. range.cleared = true;
  23. marks[i].clear();
  24. }
  25. }
  26. return range;
  27. }
  28. var range = getRange(true);
  29. if (options && options.scanUp) while (!range && pos.line > cm.firstLine()) {
  30. pos = CodeMirror.Pos(pos.line - 1, 0);
  31. range = getRange(false);
  32. }
  33. if (!range || range.cleared || force === "unfold") return;
  34. var myWidget = makeWidget(options);
  35. CodeMirror.on(myWidget, "mousedown", function() { myRange.clear(); });
  36. var myRange = cm.markText(range.from, range.to, {
  37. replacedWith: myWidget,
  38. clearOnEnter: true,
  39. __isFold: true
  40. });
  41. myRange.on("clear", function(from, to) {
  42. CodeMirror.signal(cm, "unfold", cm, from, to);
  43. });
  44. CodeMirror.signal(cm, "fold", cm, range.from, range.to);
  45. }
  46. function makeWidget(options) {
  47. var widget = (options && options.widget) || "\u2194";
  48. if (typeof widget == "string") {
  49. var text = document.createTextNode(widget);
  50. widget = document.createElement("span");
  51. widget.appendChild(text);
  52. widget.className = "CodeMirror-foldmarker";
  53. }
  54. return widget;
  55. }
  56. // Clumsy backwards-compatible interface
  57. CodeMirror.newFoldFunction = function(rangeFinder, widget) {
  58. return function(cm, pos) { doFold(cm, pos, {rangeFinder: rangeFinder, widget: widget}); };
  59. };
  60. // New-style interface
  61. CodeMirror.defineExtension("foldCode", function(pos, options, force) {
  62. doFold(this, pos, options, force);
  63. });
  64. CodeMirror.defineExtension("isFolded", function(pos) {
  65. var marks = this.findMarksAt(pos);
  66. for (var i = 0; i < marks.length; ++i)
  67. if (marks[i].__isFold) return true;
  68. });
  69. CodeMirror.commands.toggleFold = function(cm) {
  70. cm.foldCode(cm.getCursor());
  71. };
  72. CodeMirror.commands.fold = function(cm) {
  73. cm.foldCode(cm.getCursor(), null, "fold");
  74. };
  75. CodeMirror.commands.unfold = function(cm) {
  76. cm.foldCode(cm.getCursor(), null, "unfold");
  77. };
  78. CodeMirror.commands.foldAll = function(cm) {
  79. cm.operation(function() {
  80. for (var i = cm.firstLine(), e = cm.lastLine(); i <= e; i++)
  81. cm.foldCode(CodeMirror.Pos(i, 0), null, "fold");
  82. });
  83. };
  84. CodeMirror.commands.unfoldAll = function(cm) {
  85. cm.operation(function() {
  86. for (var i = cm.firstLine(), e = cm.lastLine(); i <= e; i++)
  87. cm.foldCode(CodeMirror.Pos(i, 0), null, "unfold");
  88. });
  89. };
  90. CodeMirror.registerHelper("fold", "combine", function() {
  91. var funcs = Array.prototype.slice.call(arguments, 0);
  92. return function(cm, start) {
  93. for (var i = 0; i < funcs.length; ++i) {
  94. var found = funcs[i](cm, start);
  95. if (found) return found;
  96. }
  97. };
  98. });
  99. CodeMirror.registerHelper("fold", "auto", function(cm, start) {
  100. var helpers = cm.getHelpers(start, "fold");
  101. for (var i = 0; i < helpers.length; i++) {
  102. var cur = helpers[i](cm, start);
  103. if (cur) return cur;
  104. }
  105. });
  106. });