From 1b5889887bd78b7404a3c2335a216891eebecb81 Mon Sep 17 00:00:00 2001
From: Nikita <abracham.mitchell@gmail.com>
Date: Wed, 19 Feb 2014 12:57:39 +0400
Subject: [PATCH] SpringAnnotationScanner added

---
 pom.xml                                       | 25 +++++-
 .../annotation/SpringAnnotationScanner.java   | 89 +++++++++++++++++++
 2 files changed, 113 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/com/corundumstudio/socketio/annotation/SpringAnnotationScanner.java

diff --git a/pom.xml b/pom.xml
index 0b17ccd..26437bb 100644
--- a/pom.xml
+++ b/pom.xml
@@ -143,6 +143,7 @@
       <artifactId>slf4j-api</artifactId>
       <version>1.7.5</version>
   </dependency>
+
   <dependency>
   	<groupId>com.fasterxml.jackson.core</groupId>
   	<artifactId>jackson-core</artifactId>
@@ -153,6 +154,28 @@
   	<artifactId>jackson-databind</artifactId>
   	<version>2.3.1</version>
   </dependency>
+
+
+  <dependency>
+      <groupId>org.springframework</groupId>
+      <artifactId>spring-beans</artifactId>
+      <version>[2.5,)</version>
+      <scope>provided</scope>
+  </dependency>
+  <dependency>
+      <groupId>org.springframework</groupId>
+      <artifactId>spring-core</artifactId>
+      <version>[2.5,)</version>
+      <exclusions>
+         <exclusion>
+             <groupId>commons-logging</groupId>
+             <artifactId>commons-logging</artifactId>
+         </exclusion>
+      </exclusions>
+      <scope>provided</scope>
+  </dependency>
+
+
     <dependency>
         <groupId>org.redisson</groupId>
         <artifactId>redisson</artifactId>
@@ -270,7 +293,7 @@
                <instructions>
                    <Bundle-SymbolicName>${project.artifactId}</Bundle-SymbolicName>
                    <Import-Package>
-                    com.hazelcast.*;resolution:=optional,org.redisson.*;resolution:=optional,*
+                    org.springframework.*;resolution:=optional,com.hazelcast.*;resolution:=optional,org.redisson.*;resolution:=optional,*
                    </Import-Package>
                    <Export-Package>
                     com.corundumstudio.socketio;version="${project.version}",
diff --git a/src/main/java/com/corundumstudio/socketio/annotation/SpringAnnotationScanner.java b/src/main/java/com/corundumstudio/socketio/annotation/SpringAnnotationScanner.java
new file mode 100644
index 0000000..8aad310
--- /dev/null
+++ b/src/main/java/com/corundumstudio/socketio/annotation/SpringAnnotationScanner.java
@@ -0,0 +1,89 @@
+/**
+ * Copyright 2012 Nikita Koksharov
+ *
+ * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.corundumstudio.socketio.annotation;
+
+import java.lang.annotation.Annotation;
+import java.lang.reflect.Method;
+import java.util.Arrays;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicBoolean;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+import org.springframework.util.ReflectionUtils;
+import org.springframework.util.ReflectionUtils.MethodCallback;
+import org.springframework.util.ReflectionUtils.MethodFilter;
+
+import com.corundumstudio.socketio.SocketIOServer;
+
+public class SpringAnnotationScanner implements BeanPostProcessor {
+
+    private final Logger log = LoggerFactory.getLogger(getClass());
+
+    private final List<Class<? extends Annotation>> annotations =
+                    Arrays.asList(OnConnect.class, OnDisconnect.class, OnEvent.class, OnMessage.class, OnJsonObject.class);
+
+    private final SocketIOServer socketIOServer;
+
+    private Class originalBeanClass;
+
+    public SpringAnnotationScanner(SocketIOServer socketIOServer) {
+        super();
+        this.socketIOServer = socketIOServer;
+    }
+
+    @Override
+    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
+        if (originalBeanClass != null) {
+            socketIOServer.addListeners(bean, originalBeanClass);
+            log.info("{} bean listeners added", beanName);
+            originalBeanClass = null;
+        }
+        return bean;
+    }
+
+    @Override
+    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
+        final AtomicBoolean add = new AtomicBoolean();
+        ReflectionUtils.doWithMethods(bean.getClass(),
+                new MethodCallback() {
+            @Override
+            public void doWith(Method method) throws IllegalArgumentException,
+            IllegalAccessException {
+                add.set(true);
+            }
+        },
+        new MethodFilter() {
+            @Override
+            public boolean matches(Method method) {
+                for (Class<? extends Annotation> annotationClass : annotations) {
+                    if (method.isAnnotationPresent(annotationClass)) {
+                        return true;
+                    }
+                }
+                return false;
+            }
+        });
+
+        if (add.get()) {
+            originalBeanClass = bean.getClass();
+        }
+        return bean;
+    }
+
+}