feature(): Add ClassTooBig inspection to vb.net

hhariri · Sep 22, 2016 · 3b7dc4d · 3b7dc4d
1 parent 5d76318
commit 3b7dc4d
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 2 deletions.
diff --git a/CleanCode/src/CleanCode/CleanCode.csproj b/CleanCode/src/CleanCode/CleanCode.csproj
@@ -164,6 +164,7 @@
     <Compile Include="CleanCodeHighlightingGroupIds.cs" />
     <Compile Include="Features\ChainedReferences\MaximumChainedReferencesHighlighting.cs" />
     <Compile Include="Features\ChainedReferences\ChainedReferencesCheck.cs" />
+    <Compile Include="Features\ClassTooBig\ClassTooBigCheckVb.cs" />
     <Compile Include="Features\ClassTooBig\ClassTooBigHighlighting.cs" />
     <Compile Include="Features\ExcessiveIndentation\ExcessiveIndentationCheck.cs" />
     <Compile Include="Features\ComplexExpression\ComplexConditionExpressionHighlighting.cs" />

diff --git a/CleanCode/src/CleanCode/Features/ClassTooBig/ClassTooBigCheckVb.cs b/CleanCode/src/CleanCode/Features/ClassTooBig/ClassTooBigCheckVb.cs
@@ -0,0 +1,31 @@
+using CleanCode.Resources;
+using CleanCode.Settings;
+using JetBrains.Application.Settings;
+using JetBrains.ReSharper.Daemon.Stages.Dispatcher;
+using JetBrains.ReSharper.Feature.Services.Daemon;
+using JetBrains.ReSharper.Psi.VB.Tree;
+using JetBrains.ReSharper.Psi.Tree;
+
+namespace CleanCode.Features.ClassTooBig
+{
+    [ElementProblemAnalyzer(typeof(IClassDeclaration), HighlightingTypes = new []
+    {
+        typeof(ClassTooBigHighlighting)
+    })]
+    public class ClassTooBigCheckVb : ElementProblemAnalyzer<IClassDeclaration>
+    {
+        protected override void Run(IClassDeclaration element, ElementProblemAnalyzerData data, IHighlightingConsumer consumer)
+        {
+            var maxLength = data.SettingsStore.GetValue((CleanCodeSettings s) => s.MaximumMethodsInClass);
+
+            var statementCount = element.CountChildren<IMethodDeclaration>();
+            if (statementCount > maxLength)
+            {
+                var declarationIdentifier = element.Name;
+                var documentRange = declarationIdentifier.GetDocumentRange();
+                var highlighting = new ClassTooBigHighlighting(Warnings.ClassTooBig, documentRange);
+                consumer.AddHighlighting(highlighting);
+            }
+        }
+    }
+}
diff --git a/CleanCode/src/CleanCode/Features/ClassTooBig/ClassTooBigHighlighting.cs b/CleanCode/src/CleanCode/Features/ClassTooBig/ClassTooBigHighlighting.cs
@@ -3,14 +3,15 @@
 using JetBrains.DocumentModel;
 using JetBrains.ReSharper.Feature.Services.Daemon;
 using JetBrains.ReSharper.Psi.CSharp;
+using JetBrains.ReSharper.Psi.VB;
 
 [assembly: RegisterConfigurableSeverity(ClassTooBigHighlighting.SeverityID, null,
     CleanCodeHighlightingGroupIds.CleanCode, "Class too big", "This class contains too many methods",
     Severity.SUGGESTION, false)]
 
 namespace CleanCode.Features.ClassTooBig
-{ 
-    [ConfigurableSeverityHighlighting(SeverityID, CSharpLanguage.Name)]
+{
+    [ConfigurableSeverityHighlighting(SeverityID, CSharpLanguage.Name + "," + VBLanguage.Name)]
     public class ClassTooBigHighlighting : IHighlighting
     {
         internal const string SeverityID = "ClassTooBig";