1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 package info.magnolia.module.blossom.web;
35
36 import info.magnolia.cms.core.AggregationState;
37 import info.magnolia.cms.core.Channel;
38 import info.magnolia.cms.security.MgnlUser;
39 import info.magnolia.cms.security.User;
40 import info.magnolia.context.Context;
41 import info.magnolia.context.MgnlContext;
42 import info.magnolia.context.WebContext;
43 import info.magnolia.module.blossom.template.BlossomAreaDefinition;
44 import info.magnolia.module.blossom.template.BlossomTemplateDefinition;
45 import info.magnolia.module.site.Site;
46 import info.magnolia.module.site.SiteManager;
47 import info.magnolia.module.site.functions.SiteFunctions;
48 import info.magnolia.module.site.theme.Theme;
49 import info.magnolia.objectfactory.Components;
50 import info.magnolia.rendering.context.RenderingContext;
51 import info.magnolia.rendering.template.AreaDefinition;
52 import info.magnolia.rendering.template.TemplateDefinition;
53
54 import javax.jcr.Node;
55
56 import org.springframework.beans.factory.InitializingBean;
57 import org.springframework.core.MethodParameter;
58 import org.springframework.web.bind.support.WebArgumentResolver;
59 import org.springframework.web.context.request.NativeWebRequest;
60
61
62
63
64
65
66
67
68
69
70 public class BlossomWebArgumentResolver implements WebArgumentResolver, InitializingBean {
71
72 private SiteManager siteManager;
73 private SiteFunctions siteFunctions;
74
75 public void setSiteManager(SiteManager siteManager) {
76 this.siteManager = siteManager;
77 }
78
79 public void setSiteFunctions(SiteFunctions siteFunctions) {
80 this.siteFunctions = siteFunctions;
81 }
82
83 @Override
84 public Object resolveArgument(MethodParameter methodParameter, NativeWebRequest webRequest) throws Exception {
85
86 if (methodParameter.getParameterType().isAssignableFrom(Node.class)) {
87 RenderingContext renderingContext = Components.getComponent(RenderingContext.class);
88 if (hasSubsequentParametersOfType(methodParameter, Node.class)) {
89 return renderingContext.getMainContent();
90 }
91 Node currentContent = renderingContext.getCurrentContent();
92 return currentContent != null ? currentContent : renderingContext.getMainContent();
93 }
94
95 if (methodParameter.getParameterType().isAssignableFrom(TemplateDefinition.class)) {
96 return Components.getComponent(RenderingContext.class).getRenderableDefinition();
97 }
98 if (methodParameter.getParameterType().isAssignableFrom(BlossomTemplateDefinition.class)) {
99 return Components.getComponent(RenderingContext.class).getRenderableDefinition();
100 }
101 if (methodParameter.getParameterType().isAssignableFrom(AreaDefinition.class)) {
102 return Components.getComponent(RenderingContext.class).getRenderableDefinition();
103 }
104 if (methodParameter.getParameterType().isAssignableFrom(BlossomAreaDefinition.class)) {
105 return Components.getComponent(RenderingContext.class).getRenderableDefinition();
106 }
107 if (methodParameter.getParameterType().isAssignableFrom(AggregationState.class)) {
108 return MgnlContext.getAggregationState();
109 }
110 if (methodParameter.getParameterType().isAssignableFrom(WebContext.class)) {
111 return MgnlContext.getWebContext();
112 }
113 if (methodParameter.getParameterType().isAssignableFrom(Context.class)) {
114 return MgnlContext.getInstance();
115 }
116 if (methodParameter.getParameterType().isAssignableFrom(User.class)) {
117 return MgnlContext.getUser();
118 }
119 if (methodParameter.getParameterType().isAssignableFrom(MgnlUser.class)) {
120 return MgnlContext.getUser();
121 }
122 if (methodParameter.getParameterType().isAssignableFrom(Channel.class)) {
123 return MgnlContext.getAggregationState().getChannel();
124 }
125 if (methodParameter.getParameterType().isAssignableFrom(Theme.class)) {
126 Site site = siteManager.getCurrentSite();
127 if (site == null) {
128 return null;
129 }
130 return siteFunctions.theme(site);
131 }
132 if (methodParameter.getParameterType().isAssignableFrom(Site.class)) {
133 return siteManager.getCurrentSite();
134 }
135
136 return UNRESOLVED;
137 }
138
139 private boolean hasSubsequentParametersOfType(MethodParameter methodParameter, Class<?> clazz) {
140 Class<?>[] parameterTypes = methodParameter.getMethod().getParameterTypes();
141 for (int i = methodParameter.getParameterIndex() + 1; i < parameterTypes.length; i++) {
142 Class<?> parameterType = parameterTypes[i];
143 if (parameterType.equals(clazz)) {
144 return true;
145 }
146 }
147 return false;
148 }
149
150 @Override
151 public void afterPropertiesSet() throws Exception {
152 if (siteFunctions == null) {
153 this.siteFunctions = Components.getComponent(SiteFunctions.class);
154 }
155 if (siteManager == null) {
156 this.siteManager = Components.getComponent(SiteManager.class);
157 }
158 }
159 }