"

 COPYRIGHT (c) 2010 by Jan Vrany, SWING Research Group. CTU in Prague

	      All Rights Reserved

Permission is hereby granted, free of charge, to any person

obtaining a copy of this software and associated documentation

files (the 'Software'), to deal in the Software without

restriction, including without limitation the rights to use,

copy, modify, merge, publish, distribute, sublicense, and/or sell

copies of the Software, and to permit persons to whom the

Software is furnished to do so, subject to the following

conditions:

The above copyright notice and this permission notice shall be

included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,

EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

OTHER DEALINGS IN THE SOFTWARE.

"

"{ Package: 'jv:smallsense' }"

"{ NameSpace: SmallSense }"

SyntaxHighlighter subclass:#SmalltalkSyntaxHighlighter

	instanceVariableNames:'elements lastVariableElements lastSelectorElement

		ignoreBadIdentifier'

	classVariableNames:''

	poolDictionaries:''

	category:'SmallSense-Smalltalk'

!

!SmalltalkSyntaxHighlighter class methodsFor:'documentation'!

copyright

"

 COPYRIGHT (c) 2010 by Jan Vrany, SWING Research Group. CTU in Prague

	      All Rights Reserved

Permission is hereby granted, free of charge, to any person

obtaining a copy of this software and associated documentation

files (the 'Software'), to deal in the Software without

restriction, including without limitation the rights to use,

copy, modify, merge, publish, distribute, sublicense, and/or sell

copies of the Software, and to permit persons to whom the

Software is furnished to do so, subject to the following

conditions:

The above copyright notice and this permission notice shall be

included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND,

EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES

OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND

NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT

HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,

WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR

OTHER DEALINGS IN THE SOFTWARE.

"

! !

!SmalltalkSyntaxHighlighter class methodsFor:'highlighting'!

format:aString parsingWith:aBlock in:aClass elementsInto:elements

    "common code for formatStatementList, formatExpression, ...

     format (recolor) whatever is parsed in aBlock (within the context of a given class).

     Return the text containing font changes and color information.

     As a side effect, put syntax elements into the passed in elements container

     (for element-highlighting in codeView2)"

    |parser tree text endPos|

    aString isNil ifTrue:[^ nil].

    parser := self for:(ReadStream on:aString string) in:aClass.

    parser elements: elements.

    parser ignoreErrors:true.

    parser ignoreWarnings:true.

    parser sourceText:(text := aString string asText).

    "/ use an array here - this can be changed much faster using #at:put:

    text emphasisCollection:(Array new:aString size).

    parser nextToken.

    tree := aBlock value:parser.

    "/ now, convert the emphasis-array to a runArray

    text emphasisCollection:(text emphasis asRunArray).

    tree == #Error ifTrue:[

        "/ mhmh - which is better ...

        "/ alternative1: color rest after error in red

"/        text 

"/            emphasizeFrom:(parser sourceStream position) 

"/            to:text size 

"/            with:(#color->Color red).

        "/ alternative2: take original emphasis for rest

        endPos := parser sourceStream position + 1.

        endPos >= text size ifTrue:[

            ^ text

        ].

        ^ ((text copyTo:endPos) , (aString copyFrom:(endPos+1))).

        "/ alternative3: no emphasis for rest.

"/        ^ text "/ aString

    ].

    ^ text

    "

     self

        formatStatementList:'(1 + 2) max:5. 1 + 2' 

        in:UndefinedObject

        elementsInto:(OrderedCollection new).

    "

!

formatClassDefinition:aString in:aClass elementsInto: elements

    "format (recolor) a class definition expression in a given class.

     Return the text containing font changes and color information."

    ^ self formatExpression:aString in:aClass elementsInto: elements

    "Created: / 10-04-2011 / 18:18:00 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

formatExpression:aString in:aClass elementsInto: elements

    "format (recolor) an expression in a given class.

     Return the text containing font changes and color information."

    |parser tree text endPos|

    aString isNil ifTrue:[^ nil].

    parser := self for:(ReadStream on:aString string) in:aClass.

    parser elements: elements.

    parser ignoreErrors:true.

    parser ignoreWarnings:true.

    parser sourceText:(text := aString string asText).

    "/ use an array here - this can be changed much faster using #at:put:

    text emphasisCollection:(Array new:aString size).

    parser nextToken.

    tree := parser "expression"statementList.

    "/ now, convert the emphasis-array to a runArray

    text emphasisCollection:(text emphasis asRunArray).

    tree == #Error ifTrue:[

	"/ mhmh - which is better ...

	"/ alternative1: color rest after error in red

"/        text

"/            emphasizeFrom:(parser sourceStream position)

"/            to:text size

"/            with:(#color->Color red).

	"/ alternative2: take original emphasis for rest

	endPos := parser sourceStream position1Based.

	endPos >= text size ifTrue:[

	    ^ text

	].

	^ ((text copyTo:endPos) , (aString copyFrom:(endPos+1))).

	"/ alternative3: no emphasis for rest.

"/        ^ text "/ aString

    ].

    ^ text

    "

     self

	formatExpression:'(1 + 2) max:5'

	in:UndefinedObject

    "

    "Created: / 25-07-2010 / 08:56:05 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 25-07-2010 / 10:57:00 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

formatMethod:aString in:aClass using:preferencesOrNil elementsInto: elements

    "format (recolor) a method in a given class.

     Return the text containing font changes and color information."

    "/ obsolete interface

    ^ self

	formatMethod:nil

	source:aString

	in:aClass

	using:preferencesOrNil elementsInto:elements

    "Created: / 25-07-2010 / 08:56:29 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified (comment): / 05-07-2011 / 11:07:50 / cg"

!

formatMethod:aMethodOrNil source:aString in:aClass using:preferencesOrNil elementsInto: elements

    "format (recolor) a method in a given class.

     Return the text containing font changes and color information."

    |highlighter tree text endPos eColor|

    aString isNil ifTrue:[^ nil].

    Error handle:[:ex |

        ex creator isHandled ifTrue:[

            ex reject.

        ].

        (self parseErrorSignal handles:ex) ifFalse:[

            "Parse error may happen when re-formatting incomplete code while editing"

            ('SyntaxHighlighter [info]: error during highlight: ' , ex description) infoPrintCR.

            "/ ex suspendedContext fullPrintAll.

        ].

        ^ aString

    ] do:[

        highlighter := self for:(ReadStream on:aString string) in:aClass.

        highlighter elements: elements.

        preferencesOrNil notNil ifTrue:[highlighter preferences:preferencesOrNil].

        "/ highlighter ignoreErrors:true.

        highlighter ignoreWarnings:true.

        highlighter sourceText:(text := aString string asText).

        "/ use an array here - this can be changed much faster using #at:put:

        text emphasisCollection:(Array new:aString size).

        tree := highlighter parseMethod.

        "/ now, convert the emphasis-array to a runArray

        text emphasisCollection:(text emphasis asRunArray).

        tree == #Error ifTrue:[

            eColor := UserPreferences current errorColor.

            eColor notNil ifTrue:[

                "/ mhmh - which is better ...

                "/ alternative1: color rest after error in red

                text

                    emphasizeFrom:(highlighter sourceStream position + 1) 

                    to:text size

                    with:(#color->eColor).

            ] ifFalse:[

                "/ alternative2: take original emphasis for rest

                endPos := highlighter sourceStream position + 1.

                endPos >= text size ifTrue:[

                    ^ Array with: text with: highlighter elements

                ].

                ^ ((text copyTo:endPos) , (aString copyFrom:(endPos+1)))

            ].

            "/ alternative3: no emphasis for rest.

        ].

        ^text

    ]

    "

     self

        formatMethod:'foo

    ^ self bar:''hello''.

    ' , (Character doubleQuote asString) , 'some comment' , (Character doubleQuote asString) , '

'

        in:UndefinedObject

    "

    "Modified: / 22-08-2006 / 13:32:04 / cg"

    "Created: / 05-07-2011 / 10:39:21 / cg"

    "Modified: / 28-05-2013 / 22:45:02 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!SmalltalkSyntaxHighlighter methodsFor:'accessing'!

elements

    ^ elements

!

elements:aParseTreeIndex

    "the element collection, to collect variables, selectors etc. into"

    elements := aParseTreeIndex.

    "Modified (comment): / 21-08-2011 / 09:13:31 / cg"

!

tree: aParseNode

    super tree: aParseNode.

    elements tree: aParseNode

    "Created: / 16-02-2012 / 09:56:17 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!SmalltalkSyntaxHighlighter methodsFor:'initialization'!

initialize

    super initialize.

    elements := ParseTreeIndex new.

    lastVariableElements := Dictionary new.

    "Created: / 14-02-2010 / 13:08:27 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 21-08-2011 / 09:37:35 / cg"

    "Modified: / 16-02-2012 / 09:59:58 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!SmalltalkSyntaxHighlighter methodsFor:'parsing-expressions'!

_binaryExpressionFor:receiverArg

    "parse a binary-expression; return a node-tree, nil or #Error"

    |receiver expr arg sel pos1 pos2 lno|

    receiver := receiverArg.

    (receiver == #Error) ifTrue:[^ #Error].

    "special kludge: since Scanner cannot know if -digit is a binary

     expression or a negative constant, handle cases here"

    [(tokenType == #BinaryOperator)

     or:[(tokenType == $|)

     or:[(tokenType == $^ and:[parserFlags allowCaretAsBinop])

	 or:[((tokenType == #Integer) or:[tokenType == #Float])

	     and:[tokenValue < 0]]]]

    ] whileTrue:[

	"/ kludge alarm: in a function-call argList, #, is not a binarySelector

	inFunctionCallArgument == true ifTrue:[

	    ((tokenType == #BinaryOperator) and:[tokenName = ',']) ifTrue:[

		^ receiver

	    ].

	].

	pos1 := tokenPosition.

	lno := tokenLineNr.

	"/ kludge alarm: bar, caret and minus are not scanned as binop

	(tokenType == $|) ifTrue:[

	    sel := '|'.

	    sel := self selectorCheck:sel for:receiver position:tokenPosition to:tokenPosition.

	    self nextToken.

	] ifFalse:[

	    (tokenType == $^) ifTrue:[

		sel := '^'.

		sel := self selectorCheck:sel for:receiver position:tokenPosition to:tokenPosition.

		self nextToken.

	    ] ifFalse:[

		(tokenType == #BinaryOperator) ifTrue:[

		    sel := tokenName.

		    sel := self selectorCheck:sel for:receiver position:tokenPosition to:(tokenPosition + tokenName size - 1).

		    self nextToken

		] ifFalse:[

		    sel := '-'.

		    token := tokenValue := tokenValue negated.

		    tokenPosition := tokenPosition + 1. "/ to skip the sign

		]

	    ].

	].

	pos2 := pos1 + sel size - 1.

	self markSelector:sel from:pos1 to:pos2 receiverNode:receiver.

	lastSelectorElement := nil.

	arg := self unaryExpression.

	(arg == #Error) ifTrue:[^ #Error].

	expr := BinaryNode receiver:receiver selector:sel arg:arg fold:foldConstants.

	expr isErrorNode ifTrue:[

	    self parseError:(expr errorString) position:pos1 to:tokenPosition.

	    errorFlag := false. "ok, user wants it - so he'll get it"

	    expr := BinaryNode receiver:receiver selector:sel arg:arg fold:nil.

	].

	expr lineNumber:lno.

	expr selectorPosition:pos1.

	self checkPlausibilityOf:expr from:pos1 to:pos2.

	parseForCode ifFalse:[

	    self rememberSelectorUsed:sel receiver:receiver

	].

	receiver := expr.   "/ for next message

    ].

    ^ receiver

    "Modified: / 09-01-1998 / 19:05:18 / stefan"

    "Modified: / 14-02-2010 / 17:54:53 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 19-01-2012 / 10:46:49 / cg"

    "Created: / 16-02-2012 / 21:54:28 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

_keywordExpressionFor:receiverArg

    "parse a keyword-expression; return a node-tree, nil or #Error.

     keywordExpression ::= binaryexpression

			   | { KEYWORD-PART binaryExpression }

    "

    |expr receiver sel arg args posR1 posR2 pos1 pos2 lno positions constVal|

    receiver := receiverArg.

    posR1 := tokenPosition.

    (tokenType == #Keyword) ifFalse:[^ receiver].

    pos1 := posR2 := tokenPosition.

    pos2 := tokenPosition + tokenName size - 1.

    positions := OrderedCollection with:(pos1 to:pos2).

    sel := tokenName.

    lno := tokenLineNr.

    self nextToken.

    arg := self binaryExpression.

    (arg == #Error) ifTrue:[^ #Error].

    args := Array with:arg.

    [tokenType == #Keyword] whileTrue:[

	sel := sel , tokenName.

	pos2 := tokenPosition + tokenName size - 1.

	positions add:(tokenPosition to:pos2).

	self nextToken.

	arg := self binaryExpression.

	(arg == #Error) ifTrue:[^ #Error].

	args := args copyWith:arg.

    ].

    positions do:[:p |

	self markSelector:sel from:p start to:p stop receiverNode:receiver.

    ].

    lastSelectorElement := nil.

    sel := self selectorCheck:sel for:receiver positions:positions.

    ignoreWarnings ifFalse:[

	(Class definitionSelectors includes:sel) ifTrue:[

	    (receiver isVariable and:[receiver isUndeclared]) ifTrue:[

		"this is not an error - the undefined class may be loaded after this code!!"

		self warning:('as yet undefined superclass: ' , receiver name) position:pos1 to:pos2.

	    ].

	].

    ].

    expr := MessageNode receiver:receiver selector:sel args:args fold:foldConstants.

    expr isErrorNode ifTrue:[

	self parseError:(expr errorString) position:pos1 to:pos2.

	errorFlag := false. "ok, user wants it - so he'll get it"

	expr := MessageNode receiver:receiver selector:sel args:args fold:nil.

    ].

    expr lineNumber:lno.

    self checkPlausibilityOf:expr from:pos1 to:pos2.

    parseForCode ifFalse:[

	self rememberSelectorUsed:sel receiver:receiver

    ].

"/        (contextToEvaluateIn isNil and:[selfValue isNil]) ifTrue:[    "/ do not check this for doits

"/            receiver isSuper ifTrue:[

"/                sel ~= selector ifTrue:[

"/                    self warnCommonMistake:'possible bad super message (selector should be same as in current method) ?'

"/                                  position:posR1 to:posR2-1

"/                ].

"/            ].

"/        ].

"/

    (sel = #ifTrue: or:[sel = #ifFalse: or:[sel = #ifTrue:ifFalse: or:[sel = #ifFalse:ifTrue:]]]) ifTrue:[

	(expr receiver withConstantValueDo:[:val | constVal := val]) ifTrue:[

	    |indexOfArgNotExecuted|

	    "/ receiver evaluates to a constant

	    constVal == true ifTrue:[

		(sel startsWith: #ifFalse:) ifTrue:[

		    indexOfArgNotExecuted := 1.

		] ifFalse:[

		    indexOfArgNotExecuted := 2.

		]

	    ].

	    constVal == false ifTrue:[

		(sel startsWith: #ifTrue:) ifTrue:[

		    indexOfArgNotExecuted := 1.

		] ifFalse:[

		    indexOfArgNotExecuted := 2.

		]

	    ].

	    indexOfArgNotExecuted == 2 ifTrue:[

		args size == 1 ifTrue:[ indexOfArgNotExecuted := nil]

	    ].

	    indexOfArgNotExecuted notNil ifTrue:[

		|argIsNotExecuted|

		"/ self warning:'receiver is constant; arg',indexOfArgNotExecuted printString,' is never executed' position:pos1 to:tokenPosition.

		argIsNotExecuted := expr args at:indexOfArgNotExecuted.

		argIsNotExecuted isBlockNode ifTrue:[

		    self markCommentFrom:argIsNotExecuted startPosition to:argIsNotExecuted endPosition.

		].

	    ].

	].

    ].

    (ignoreErrors or:[ignoreWarnings]) ifFalse:[

	(sel = #and: or:[sel = #or:]) ifTrue:[

	    expr arg1 isBlock ifFalse:[

		(expr arg1 isVariable

		and:[ (expr arg1 name asLowercase includesString:'block')]) ifFalse:[

		    self warnCommonMistake:'(possible common mistake) missing block brackets ?'

			      position:pos2+1 to:tokenPosition-1

		]

	    ].

	    ^ expr.

	].

	(sel = #whileTrue: or:[sel = #whileFalse:]) ifTrue:[

	    expr receiver isBlock ifFalse:[

		(expr receiver isVariable

		and:[ (expr receiver name asLowercase includesString:'block')]) ifFalse:[

		    self warnCommonMistake:'(possible common mistake) missing block brackets ?'

			      position:pos1 to:pos2

		]

	    ].

	    ^ expr.

	].

	(sel = #ifTrue: or:[sel = #ifFalse:]) ifTrue:[

	    expr receiver isMessage ifTrue:[

		(expr receiver selector = #whileTrue or:[expr receiver selector = #whileFalse]) ifTrue:[

		    self warnCommonMistake:'strange receiver expression'

			      position:pos1 to:pos2

		].

	    ].

	    ^ expr

	].

    ].

    ^ expr.

    "Modified: / 14-02-2010 / 17:58:40 / Jan Vrany <jan.vrany@fit.cvut.cz>"