blob: 1ddbff4e0b0bfa62c388707b2b9a5abb0f371e70 (
plain)
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
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System.Globalization;
using System.Linq;
using System.Web.Mvc.Properties;
using System.Web.Razor.Generator;
using System.Web.Razor.Parser;
using System.Web.Razor.Parser.SyntaxTree;
using System.Web.Razor.Text;
using System.Web.Razor.Tokenizer.Symbols;
namespace System.Web.Mvc.Razor
{
public class MvcVBRazorCodeParser : VBCodeParser
{
internal const string ModelTypeKeyword = "ModelType";
private const string GenericTypeFormatString = "{0}(Of {1})";
private SourceLocation? _endInheritsLocation;
private bool _modelStatementFound;
public MvcVBRazorCodeParser()
{
MapDirective(ModelTypeKeyword, ModelTypeDirective);
}
protected override bool InheritsStatement()
{
// Verify we're on the right keyword and accept
Assert(VBKeyword.Inherits);
VBSymbol inherits = CurrentSymbol;
NextToken();
_endInheritsLocation = CurrentLocation;
PutCurrentBack();
PutBack(inherits);
EnsureCurrent();
bool result = base.InheritsStatement();
CheckForInheritsAndModelStatements();
return result;
}
private void CheckForInheritsAndModelStatements()
{
if (_modelStatementFound && _endInheritsLocation.HasValue)
{
Context.OnError(_endInheritsLocation.Value, String.Format(CultureInfo.CurrentCulture, MvcResources.MvcRazorCodeParser_CannotHaveModelAndInheritsKeyword, ModelTypeKeyword));
}
}
protected virtual bool ModelTypeDirective()
{
AssertDirective(ModelTypeKeyword);
Span.CodeGenerator = SpanCodeGenerator.Null;
Context.CurrentBlock.Type = BlockType.Directive;
AcceptAndMoveNext();
SourceLocation endModelLocation = CurrentLocation;
if (At(VBSymbolType.WhiteSpace))
{
Span.EditHandler.AcceptedCharacters = AcceptedCharacters.None;
}
AcceptWhile(VBSymbolType.WhiteSpace);
Output(SpanKind.MetaCode);
if (_modelStatementFound)
{
Context.OnError(endModelLocation, String.Format(CultureInfo.CurrentCulture, MvcResources.MvcRazorCodeParser_OnlyOneModelStatementIsAllowed, ModelTypeKeyword));
}
_modelStatementFound = true;
if (EndOfFile || At(VBSymbolType.WhiteSpace) || At(VBSymbolType.NewLine))
{
Context.OnError(endModelLocation, MvcResources.MvcRazorCodeParser_ModelKeywordMustBeFollowedByTypeName, ModelTypeKeyword);
}
// Just accept to a newline
AcceptUntil(VBSymbolType.NewLine);
if (!Context.DesignTimeMode)
{
// We want the newline to be treated as code, but it causes issues at design-time.
Optional(VBSymbolType.NewLine);
}
string baseType = String.Concat(Span.Symbols.Select(s => s.Content)).Trim();
Span.CodeGenerator = new SetModelTypeCodeGenerator(baseType, GenericTypeFormatString);
CheckForInheritsAndModelStatements();
Output(SpanKind.Code);
return false;
}
}
}
|