package com.cflint.plugins.core;

import com.cflint.CF;
import com.cflint.BugList;
import com.cflint.plugins.Context;

import cfml.parsing.cfscript.script.CFCompoundStatement;
import cfml.parsing.cfscript.script.CFScriptStatement;
import net.htmlparser.jericho.Element;
import ro.fortsoft.pf4j.Extension;

@Extension
public class ComponentLengthChecker extends LengthChecker {
    private static final int LENGTH_THRESHOLD = 500;

    @Override
    public void expression(final CFScriptStatement expression, final Context context, final BugList bugs) {
        if (expression instanceof CFCompoundStatement) {
            final CFCompoundStatement component = (CFCompoundStatement) expression;
            final String decompile = component.Decompile(1);
            final String[] lines = decompile.split("\\n");

            checkSize(LENGTH_THRESHOLD, "EXCESSIVE_COMPONENT_LENGTH", context, 1, lines.length, bugs);
        }
    }

    @Override
    public void element(final Element element, final Context context, final BugList bugs) {
        final String elementName = element.getName();

        if (elementName.equals(CF.CFCOMPONENT)) {
            // this includes whitespace-change it
            final int total = element.getAllStartTags().size();

            checkSize(LENGTH_THRESHOLD, "EXCESSIVE_COMPONENT_LENGTH", context, 1, total, bugs);
        }
    }
}

@@ -112,16 +112,73 @@
 ntext;%0A%0A
+import cfml.parsing.cfscript.script.CFCompDeclStatement;%0A
 import c
@@ -218,32 +218,32 @@
 poundStatement;%0A
-
 import cfml.pars
@@ -639,28 +639,28 @@
 nceof CFComp
-ound
+Decl
 Statement) %7B
@@ -680,28 +680,28 @@
 final CFComp
-ound
+Decl
 Statement co
@@ -713,28 +713,28 @@
 nt = (CFComp
-ound
+Decl
 Statement) e
@@ -1291,29 +1291,48 @@
 .get
-AllStartTags().size()
+Content().toString().split(%22%5C%5Cn%22).length
 ;%0A%0A 

Fix ComponentLengthChecker