From 21c72c8b96757847e578c2ff4f68a8715eb5a199 Mon Sep 17 00:00:00 2001
From: Daniel Siepmann <coding@daniel-siepmann.de>
Date: Tue, 21 Mar 2017 17:22:18 +0100
Subject: [PATCH] WIP|FEATURE: Handle trait and interfaces beside classes

* Also fix check whether another line should be added after namespace
  definition.

Relates: #36
---
 .../Sniffs/LegacyClassnames/MissingNamespaceSniff.php    | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/Standards/Typo3Update/Sniffs/LegacyClassnames/MissingNamespaceSniff.php b/src/Standards/Typo3Update/Sniffs/LegacyClassnames/MissingNamespaceSniff.php
index 2d74d70..391c357 100644
--- a/src/Standards/Typo3Update/Sniffs/LegacyClassnames/MissingNamespaceSniff.php
+++ b/src/Standards/Typo3Update/Sniffs/LegacyClassnames/MissingNamespaceSniff.php
@@ -34,7 +34,11 @@ class Typo3Update_Sniffs_LegacyClassnames_MissingNamespaceSniff extends Abstract
      */
     public function register()
     {
-        return [T_CLASS];
+        return [
+            T_CLASS,
+            T_INTERFACE,
+            T_TRAIT,
+        ];
     }
 
     /**
@@ -97,8 +101,7 @@ class Typo3Update_Sniffs_LegacyClassnames_MissingNamespaceSniff extends Abstract
         $lineEndings = PhpCsFile::detectLineEndings($phpcsFile->getFilename());
         $suffix = $lineEndings;
 
-        // TODO, does not work.
-        if ($tokens[1] === "\n") {
+        if ($tokens[1]['code'] !== T_WHITESPACE) {
             $suffix .= $lineEndings;
         }