1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
|
unit ReturnsAfterFinalEnd;
{(*}
(*------------------------------------------------------------------------------
Delphi Code formatter source code
The Original Code is ReturnsAfterFinalEnd.pas, released September 2003.
The Initial Developer of the Original Code is Anthony Steele.
Portions created by Anthony Steele are Copyright (C) 1999-2008 Anthony Steele.
All Rights Reserved.
Contributor(s): Anthony Steele.
The contents of this file are subject to the Mozilla Public License Version 1.1
(the "License"). you may not use this file except in compliance with the License.
You may obtain a copy of the License at http://www.mozilla.org/NPL/
Software distributed under the License is distributed on an "AS IS" basis,
WITHOUT WARRANTY OF ANY KIND, either express or implied.
See the License for the specific language governing rights and limitations
under the License.
Alternatively, the contents of this file may be used under the terms of
the GNU General Public License Version 2 or later (the "GPL")
See http://www.gnu.org/licenses/gpl.html
------------------------------------------------------------------------------*)
{*)}
{$I JcfGlobal.inc}
interface
{ AFS 27 Sept 2003
process to standardise the number of returns
after the final "end." of the unit }
uses SwitchableVisitor;
type
TReturnsAfterFinalEnd = class(TSwitchableVisitor)
private
protected
function EnabledVisitSourceToken(const pcNode: TObject): Boolean; override;
public
constructor Create; override;
end;
implementation
uses
JcfSettings,
SourceToken, FormatFlags, Tokens, ParseTreeNodeType, TokenUtils;
{ TReturnsAfterFinalEnd }
constructor TReturnsAfterFinalEnd.Create;
begin
inherited;
FormatFlags := FormatFlags + [eAddReturn, eRemoveReturn];
end;
function TReturnsAfterFinalEnd.EnabledVisitSourceToken(const pcNode: TObject): Boolean;
var
lcSourceToken: TSourceToken;
lcCurrent, lcPrev: TSourceToken;
liReturnsWanted, liReturnsFound: integer;
begin
Result := False;
lcSourceToken := TSourceToken(pcNode);
if (lcSourceToken.TokenType = ttDot) and
(lcSourceToken.HasParentNode(TopOfFileSection, 1)) then
begin
// count the returns
lcCurrent := lcSourceToken;
liReturnsWanted := FormattingSettings.Returns.NumReturnsAfterFinalEnd;
liReturnsFound := 0;
while lcCurrent <> nil do
begin
if lcCurrent.TokenType = ttReturn then
begin
Inc(liReturnsFound);
lcPrev := lcCurrent.PriorToken;
if (liReturnsFound > liReturnsWanted) and ( not lcPrev.IsSolid) then
begin
{ this returns is surplus - remove it }
BlankToken(lcCurrent);
end;
end;
lcCurrent := lcCurrent.NextToken;
end;
{ need to insert some returns? }
while liReturnsFound < liReturnsWanted do
begin
InsertReturnAfter(lcSourceToken);
Inc(liReturnsFound);
end;
end;
end;
end.
|