// CodeMirror, copyright (c) by Marijn Haverbeke and others
// Distributed under an MIT license: http://codemirror.net/LICENSE

(function() {
  var mode = CodeMirror.getMode({indentUnit: 2}, "swift");
  function MT(name) { test.mode(name, mode, Array.prototype.slice.call(arguments, 1)); }

  // Ensure all number types are properly represented.
  MT("numbers",
     "[keyword var] [def a] [operator =] [number 17]",
     "[keyword var] [def b] [operator =] [number -0.5]",
     "[keyword var] [def c] [operator =] [number 0.3456e-4]",
     "[keyword var] [def d] [operator =] [number 345e2]",
     "[keyword var] [def e] [operator =] [number 0o7324]",
     "[keyword var] [def f] [operator =] [number 0b10010]",
     "[keyword var] [def g] [operator =] [number -0x35ade]",
     "[keyword var] [def h] [operator =] [number 0xaea.ep-13]",
     "[keyword var] [def i] [operator =] [number 0x13ep6]");

  // Variable/class/etc definition.
  MT("definition",
     "[keyword var] [def a] [operator =] [number 5]",
     "[keyword let] [def b][punctuation :] [variable-2 Int] [operator =] [number 10]",
     "[keyword class] [def C] [punctuation {] [punctuation }]",
     "[keyword struct] [def D] [punctuation {] [punctuation }]",
     "[keyword enum] [def E] [punctuation {] [punctuation }]",
     "[keyword extension] [def F] [punctuation {] [punctuation }]",
     "[keyword protocol] [def G] [punctuation {] [punctuation }]",
     "[keyword func] [def h][punctuation ()] [punctuation {] [punctuation }]",
     "[keyword import] [def Foundation]",
     "[keyword typealias] [def NewString] [operator =] [variable-2 String]",
     "[keyword associatedtype] [def I]",
     "[keyword for] [def j] [keyword in] [number 0][punctuation ..][operator <][number 3] [punctuation {] [punctuation }]");

  // Strings and string interpolation.
  MT("strings",
     "[keyword var] [def a][punctuation :] [variable-2 String] [operator =] [string \"test\"]",
     "[keyword var] [def b][punctuation :] [variable-2 String] [operator =] [string \"\\(][variable a][string )\"]");

  // Comments.
  MT("comments",
     "[comment // This is a comment]",
     "[comment /* This is another comment */]",
     "[keyword var] [def a] [operator =] [number 5] [comment // Third comment]");

  // Atoms.
  MT("atoms",
     "[keyword class] [def FooClass] [punctuation {]",
     "  [keyword let] [def fooBool][punctuation :] [variable-2 Bool][operator ?]",
     "  [keyword let] [def fooInt][punctuation :] [variable-2 Int][operator ?]",
     "  [keyword func] [keyword init][punctuation (][variable fooBool][punctuation :] [variable-2 Bool][punctuation ,] [variable barBool][punctuation :] [variable-2 Bool][punctuation )] [punctuation {]",
     "    [atom super][property .init][punctuation ()]",
     "    [atom self][property .fooBool] [operator =] [variable fooBool]",
     "    [variable fooInt] [operator =] [atom nil]",
     "    [keyword if] [variable barBool] [operator ==] [atom true] [punctuation {]",
     "      [variable print][punctuation (][string \"True!\"][punctuation )]",
     "    [punctuation }] [keyword else] [keyword if] [variable barBool] [operator ==] [atom false] [punctuation {]",
     "      [keyword for] [atom _] [keyword in] [number 0][punctuation ...][number 5] [punctuation {]",
     "        [variable print][punctuation (][string \"False!\"][punctuation )]",
     "      [punctuation }]",
     "    [punctuation }]",
     "  [punctuation }]",
     "[punctuation }]");

  // Types.
  MT("types",
     "[keyword var] [def a] [operator =] [variable-2 Array][operator <][variable-2 Int][operator >]",
     "[keyword var] [def b] [operator =] [variable-2 Set][operator <][variable-2 Bool][operator >]",
     "[keyword var] [def c] [operator =] [variable-2 Dictionary][operator <][variable-2 String][punctuation ,][variable-2 Character][operator >]",
     "[keyword var] [def d][punctuation :] [variable-2 Int64][operator ?] [operator =] [variable-2 Optional][punctuation (][number 8][punctuation )]",
     "[keyword func] [def e][punctuation ()] [operator ->] [variable-2 Void] [punctuation {]",
     "  [keyword var] [def e1][punctuation :] [variable-2 Float] [operator =] [number 1.2]",
     "[punctuation }]",
     "[keyword func] [def f][punctuation ()] [operator ->] [variable-2 Never] [punctuation {]",
     "  [keyword var] [def f1][punctuation :] [variable-2 Double] [operator =] [number 2.4]",
     "[punctuation }]");

  // Operators.
  MT("operators",
     "[keyword var] [def a] [operator =] [number 1] [operator +] [number 2]",
     "[keyword var] [def b] [operator =] [number 1] [operator -] [number 2]",
     "[keyword var] [def c] [operator =] [number 1] [operator *] [number 2]",
     "[keyword var] [def d] [operator =] [number 1] [operator /] [number 2]",
     "[keyword var] [def e] [operator =] [number 1] [operator %] [number 2]",
     "[keyword var] [def f] [operator =] [number 1] [operator |] [number 2]",
     "[keyword var] [def g] [operator =] [number 1] [operator &] [number 2]",
     "[keyword var] [def h] [operator =] [number 1] [operator <<] [number 2]",
     "[keyword var] [def i] [operator =] [number 1] [operator >>] [number 2]",
     "[keyword var] [def j] [operator =] [number 1] [operator ^] [number 2]",
     "[keyword var] [def k] [operator =] [operator ~][number 1]",
     "[keyword var] [def l] [operator =] [variable foo] [operator ?] [number 1] [punctuation :] [number 2]",
     "[keyword var] [def m][punctuation :] [variable-2 Int] [operator =] [variable-2 Optional][punctuation (][number 8][punctuation )][operator !]");

  // Punctuation.
  MT("punctuation",
     "[keyword let] [def a] [operator =] [number 1][punctuation ;] [keyword let] [def b] [operator =] [number 2]",
     "[keyword let] [def testArr][punctuation :] [punctuation [[][variable-2 Int][punctuation ]]] [operator =] [punctuation [[][variable a][punctuation ,] [variable b][punctuation ]]]",
     "[keyword for] [def i] [keyword in] [number 0][punctuation ..][operator <][variable testArr][property .count] [punctuation {]",
     "  [variable print][punctuation (][variable testArr][punctuation [[][variable i][punctuation ]])]",
     "[punctuation }]");

  // Identifiers.
  MT("identifiers",
     "[keyword let] [def abc] [operator =] [number 1]",
     "[keyword let] [def ABC] [operator =] [number 2]",
     "[keyword let] [def _123] [operator =] [number 3]",
     "[keyword let] [def _$1$2$3] [operator =] [number 4]",
     "[keyword let] [def A1$_c32_$_] [operator =] [number 5]",
     "[keyword let] [def `var`] [operator =] [punctuation [[][number 1][punctuation ,] [number 2][punctuation ,] [number 3][punctuation ]]]",
     "[keyword let] [def square$] [operator =] [variable `var`][property .map] [punctuation {][variable $0] [operator *] [variable $0][punctuation }]",
     "$$ [number 1][variable a] $[atom _] [variable _$] [variable __] `[variable a] [variable b]`");

  // Properties.
  MT("properties",
     "[variable print][punctuation (][variable foo][property .abc][punctuation )]",
     "[variable print][punctuation (][variable foo][property .ABC][punctuation )]",
     "[variable print][punctuation (][variable foo][property ._123][punctuation )]",
     "[variable print][punctuation (][variable foo][property ._$1$2$3][punctuation )]",
     "[variable print][punctuation (][variable foo][property .A1$_c32_$_][punctuation )]",
     "[variable print][punctuation (][variable foo][property .`var`][punctuation )]",
     "[variable print][punctuation (][variable foo][property .__][punctuation )]");

  // Instructions or other things that start with #.
  MT("instructions",
     "[keyword if] [builtin #available][punctuation (][variable iOS] [number 9][punctuation ,] [operator *][punctuation )] [punctuation {}]",
     "[variable print][punctuation (][builtin #file][punctuation ,] [builtin #function][punctuation )]",
     "[variable print][punctuation (][builtin #line][punctuation ,] [builtin #column][punctuation )]",
     "[builtin #if] [atom true]",
     "[keyword import] [def A]",
     "[builtin #elseif] [atom false]",
     "[keyword import] [def B]",
     "[builtin #endif]",
     "[builtin #sourceLocation][punctuation (][variable file][punctuation :] [string \"file.swift\"][punctuation ,] [variable line][punctuation :] [number 2][punctuation )]");

  // Attributes; things that start with @.
  MT("attributes",
     "[attribute @objc][punctuation (][variable objcFoo][punctuation :)]",
     "[attribute @available][punctuation (][variable iOS][punctuation )]");

  // Property/number edge case.
  MT("property_number",
     "[variable print][punctuation (][variable foo][property ._123][punctuation )]",
     "[variable print][punctuation (]")

  // TODO: correctly identify when multiple variables are being declared
  // by use of a comma-separated list.
  // TODO: correctly identify when variables are being declared in a tuple.
  // TODO: identify protocols as types when used before an extension?
})();