src/stc/scintilla/lexers/LexPowerShell.cxx

   1 // Scintilla source code edit control
   2 /** @file LexPowerShell.cxx
   3  ** Lexer for PowerShell scripts.
   4  **/
   5 // Copyright 2008 by Tim Gerundt <tim@gerundt.de>
   6 // The License.txt file describes the conditions under which this software may be distributed.
   7
   8 #include <stdlib.h>
   9 #include <string.h>
  10 #include <stdio.h>
  11 #include <stdarg.h>
  12 #include <assert.h>
  13 #include <ctype.h>
  14
  15 #include "ILexer.h"
  16 #include "Scintilla.h"
  17 #include "SciLexer.h"
  18
  19 #include "WordList.h"
  20 #include "LexAccessor.h"
  21 #include "Accessor.h"
  22 #include "StyleContext.h"
  23 #include "CharacterSet.h"
  24 #include "LexerModule.h"
  25
  26 #ifdef SCI_NAMESPACE
  27 using namespace Scintilla;
  28 #endif
  29
  30 // Extended to accept accented characters
  31 static inline bool IsAWordChar(int ch) {
  32         return ch >= 0x80 || isalnum(ch) || ch == '-' || ch == '_';
  33 }
  34
  35 static void ColourisePowerShellDoc(unsigned int startPos, int length, int initStyle,
  36                            WordList *keywordlists[], Accessor &styler) {
  37
  38         WordList &keywords = *keywordlists[0];
  39         WordList &keywords2 = *keywordlists[1];
  40         WordList &keywords3 = *keywordlists[2];
  41         WordList &keywords4 = *keywordlists[3];
  42         WordList &keywords5 = *keywordlists[4];
  43
  44         styler.StartAt(startPos);
  45
  46         StyleContext sc(startPos, length, initStyle, styler);
  47
  48         for (; sc.More(); sc.Forward()) {
  49
  50                 if (sc.state == SCE_POWERSHELL_COMMENT) {
  51                         if (sc.atLineEnd) {
  52                                 sc.SetState(SCE_POWERSHELL_DEFAULT);
  53                         }
  54                 } else if (sc.state == SCE_POWERSHELL_COMMENTSTREAM) {
  55                         if (sc.ch == '>' && sc.chPrev == '#') {
  56                                 sc.ForwardSetState(SCE_POWERSHELL_DEFAULT);
  57                         }
  58                 } else if (sc.state == SCE_POWERSHELL_STRING) {
  59                         // This is a doubles quotes string
  60                         if (sc.ch == '\"') {
  61                                 sc.ForwardSetState(SCE_POWERSHELL_DEFAULT);
  62                         }
  63                 } else if (sc.state == SCE_POWERSHELL_CHARACTER) {
  64                         // This is a single quote string
  65                         if (sc.ch == '\'') {
  66                                 sc.ForwardSetState(SCE_POWERSHELL_DEFAULT);
  67                         }
  68                 } else if (sc.state == SCE_POWERSHELL_NUMBER) {
  69                         if (!IsADigit(sc.ch)) {
  70                                 sc.SetState(SCE_POWERSHELL_DEFAULT);
  71                         }
  72                 } else if (sc.state == SCE_POWERSHELL_VARIABLE) {
  73                         if (!IsAWordChar(sc.ch)) {
  74                                 sc.SetState(SCE_POWERSHELL_DEFAULT);
  75                         }
  76                 } else if (sc.state == SCE_POWERSHELL_OPERATOR) {
  77                         if (!isoperator(static_cast<char>(sc.ch))) {
  78                                 sc.SetState(SCE_POWERSHELL_DEFAULT);
  79                         }
  80                 } else if (sc.state == SCE_POWERSHELL_IDENTIFIER) {
  81                         if (!IsAWordChar(sc.ch)) {
  82                                 char s[100];
  83                                 sc.GetCurrentLowered(s, sizeof(s));
  84
  85                                 if (keywords.InList(s)) {
  86                                         sc.ChangeState(SCE_POWERSHELL_KEYWORD);
  87                                 } else if (keywords2.InList(s)) {
  88                                         sc.ChangeState(SCE_POWERSHELL_CMDLET);
  89                                 } else if (keywords3.InList(s)) {
  90                                         sc.ChangeState(SCE_POWERSHELL_ALIAS);
  91                                 } else if (keywords4.InList(s)) {
  92                                         sc.ChangeState(SCE_POWERSHELL_FUNCTION);
  93                                 } else if (keywords5.InList(s)) {
  94                                         sc.ChangeState(SCE_POWERSHELL_USER1);
  95                                 }
  96                                 sc.SetState(SCE_POWERSHELL_DEFAULT);
  97                         }
  98                 }
  99
 100                 // Determine if a new state should be entered.
 101                 if (sc.state == SCE_POWERSHELL_DEFAULT) {
 102                         if (sc.ch == '#') {
 103                                 sc.SetState(SCE_POWERSHELL_COMMENT);
 104                         } else if (sc.ch == '<' && sc.chNext == '#') {
 105                                 sc.SetState(SCE_POWERSHELL_COMMENTSTREAM);
 106                         } else if (sc.ch == '\"') {
 107                                 sc.SetState(SCE_POWERSHELL_STRING);
 108                         } else if (sc.ch == '\'') {
 109                                 sc.SetState(SCE_POWERSHELL_CHARACTER);
 110                         } else if (sc.ch == '$') {
 111                                 sc.SetState(SCE_POWERSHELL_VARIABLE);
 112                         } else if (IsADigit(sc.ch) || (sc.ch == '.' && IsADigit(sc.chNext))) {
 113                                 sc.SetState(SCE_POWERSHELL_NUMBER);
 114                         } else if (isoperator(static_cast<char>(sc.ch))) {
 115                                 sc.SetState(SCE_POWERSHELL_OPERATOR);
 116                         } else if (IsAWordChar(sc.ch)) {
 117                                 sc.SetState(SCE_POWERSHELL_IDENTIFIER);
 118                         }
 119                 }
 120         }
 121         sc.Complete();
 122 }
 123
 124 // Store both the current line's fold level and the next lines in the
 125 // level store to make it easy to pick up with each increment
 126 // and to make it possible to fiddle the current level for "} else {".
 127 static void FoldPowerShellDoc(unsigned int startPos, int length, int initStyle,
 128                            WordList *[], Accessor &styler) {
 129         bool foldComment = styler.GetPropertyInt("fold.comment") != 0;
 130         bool foldCompact = styler.GetPropertyInt("fold.compact", 1) != 0;
 131         bool foldAtElse = styler.GetPropertyInt("fold.at.else", 0) != 0;
 132         unsigned int endPos = startPos + length;
 133         int visibleChars = 0;
 134         int lineCurrent = styler.GetLine(startPos);
 135         int levelCurrent = SC_FOLDLEVELBASE;
 136         if (lineCurrent > 0)
 137                 levelCurrent = styler.LevelAt(lineCurrent-1) >> 16;
 138         int levelMinCurrent = levelCurrent;
 139         int levelNext = levelCurrent;
 140         char chNext = styler[startPos];
 141         int styleNext = styler.StyleAt(startPos);
 142         int style = initStyle;
 143         for (unsigned int i = startPos; i < endPos; i++) {
 144                 char ch = chNext;
 145                 chNext = styler.SafeGetCharAt(i + 1);
 146                 int stylePrev = style;
 147                 style = styleNext;
 148                 styleNext = styler.StyleAt(i + 1);
 149                 bool atEOL = (ch == '\r' && chNext != '\n') || (ch == '\n');
 150                 if (style == SCE_POWERSHELL_OPERATOR) {
 151                         if (ch == '{') {
 152                                 // Measure the minimum before a '{' to allow
 153                                 // folding on "} else {"
 154                                 if (levelMinCurrent > levelNext) {
 155                                         levelMinCurrent = levelNext;
 156                                 }
 157                                 levelNext++;
 158                         } else if (ch == '}') {
 159                                 levelNext--;
 160                         }
 161                 } else if (foldComment && style == SCE_POWERSHELL_COMMENTSTREAM) {
 162                         if (stylePrev != SCE_POWERSHELL_COMMENTSTREAM) {
 163                                 levelNext++;
 164                         } else if (styleNext != SCE_POWERSHELL_COMMENTSTREAM) {
 165                                 levelNext--;
 166                         }
 167                 }
 168                 if (!IsASpace(ch))
 169                         visibleChars++;
 170                 if (atEOL || (i == endPos-1)) {
 171                         int levelUse = levelCurrent;
 172                         if (foldAtElse) {
 173                                 levelUse = levelMinCurrent;
 174                         }
 175                         int lev = levelUse | levelNext << 16;
 176                         if (visibleChars == 0 && foldCompact)
 177                                 lev |= SC_FOLDLEVELWHITEFLAG;
 178                         if (levelUse < levelNext)
 179                                 lev |= SC_FOLDLEVELHEADERFLAG;
 180                         if (lev != styler.LevelAt(lineCurrent)) {
 181                                 styler.SetLevel(lineCurrent, lev);
 182                         }
 183                         lineCurrent++;
 184                         levelCurrent = levelNext;
 185                         levelMinCurrent = levelCurrent;
 186                         visibleChars = 0;
 187                 }
 188         }
 189 }
 190
 191 static const char * const powershellWordLists[] = {
 192         "Commands",
 193         "Cmdlets",
 194         "Aliases",
 195         "Functions",
 196         "User1",
 197         0
 198 };
 199
 200 LexerModule lmPowerShell(SCLEX_POWERSHELL, ColourisePowerShellDoc, "powershell", FoldPowerShellDoc, powershellWordLists);
 201