001/* 002 * Copyright 2012-2013 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * http://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016 017package org.springframework.boot.cli.compiler.autoconfigure; 018 019import org.codehaus.groovy.ast.ClassNode; 020import org.codehaus.groovy.control.customizers.ImportCustomizer; 021 022import org.springframework.boot.cli.compiler.AstUtils; 023import org.springframework.boot.cli.compiler.CompilerAutoConfiguration; 024import org.springframework.boot.cli.compiler.DependencyCustomizer; 025 026/** 027 * {@link CompilerAutoConfiguration} for Spring Websocket. 028 * 029 * @author Dave Syer 030 */ 031public class SpringWebsocketCompilerAutoConfiguration extends CompilerAutoConfiguration { 032 033 @Override 034 public boolean matches(ClassNode classNode) { 035 return AstUtils.hasAtLeastOneAnnotation(classNode, "EnableWebSocket", 036 "EnableWebSocketMessageBroker"); 037 } 038 039 @Override 040 public void applyDependencies(DependencyCustomizer dependencies) { 041 dependencies 042 .ifAnyMissingClasses( 043 "org.springframework.web.socket.config.annotation.EnableWebSocket") 044 .add("spring-boot-starter-websocket").add("spring-messaging"); 045 } 046 047 @Override 048 public void applyImports(ImportCustomizer imports) { 049 imports.addStarImports("org.springframework.messaging.handler.annotation", 050 "org.springframework.messaging.simp.config", 051 "org.springframework.web.socket.handler", 052 "org.springframework.web.socket.sockjs.transport.handler", 053 "org.springframework.web.socket.config.annotation") 054 .addImports("org.springframework.web.socket.WebSocketHandler"); 055 } 056 057}