2121import org .lowcoder .plugin .api .data .EndpointRequest ;
2222import org .lowcoder .plugin .api .data .EndpointResponse ;
2323import org .lowcoder .sdk .exception .BaseException ;
24+ import org .springframework .beans .factory .support .DefaultListableBeanFactory ;
2425import org .springframework .context .ApplicationContext ;
2526import org .springframework .context .support .GenericApplicationContext ;
2627import org .springframework .core .ResolvableType ;
3031import org .springframework .web .reactive .function .server .RouterFunction ;
3132import org .springframework .web .reactive .function .server .ServerResponse ;
3233import org .springframework .web .reactive .function .server .ServerResponse .BodyBuilder ;
33- import org .springframework .web .reactive .function .server .support .RouterFunctionMapping ;
3434
3535import lombok .RequiredArgsConstructor ;
3636import lombok .extern .slf4j .Slf4j ;
@@ -45,7 +45,7 @@ public class PluginEndpointHandlerImpl implements PluginEndpointHandler
4545private List <RouterFunction <ServerResponse >>routes =new ArrayList <>();
4646
4747private final ApplicationContext applicationContext ;
48- private final RouterFunctionMapping routerFunctionMapping ;
48+ private final DefaultListableBeanFactory beanFactory ;
4949
5050@ Override
5151public void registerEndpoints (String pluginUrlPrefix ,List <PluginEndpoint >endpoints )
@@ -54,20 +54,24 @@ public void registerEndpoints(String pluginUrlPrefix, List<PluginEndpoint> endpo
5454
5555if (CollectionUtils .isNotEmpty (endpoints ))
5656{
57-
57+ List < EndpointExtension > toAuthorize = new ArrayList <>();
5858for (PluginEndpoint endpoint :endpoints )
5959{
6060Method []handlers =endpoint .getClass ().getDeclaredMethods ();
6161if (handlers !=null &&handlers .length >0 )
6262{
6363for (Method handler :handlers )
6464{
65- registerEndpointHandler (urlPrefix ,endpoint ,handler );
65+ toAuthorize . addAll ( registerEndpointHandler (urlPrefix ,endpoint ,handler ) );
6666}
6767}
6868}
6969
70- ((ReloadableRouterFunctionMapping )routerFunctionMapping ).reloadFunctionMappings ();
70+ ((ReloadableRouterFunctionMapping )beanFactory .getBean ("routerFunctionMapping" )).reloadFunctionMappings ();
71+ if (!toAuthorize .isEmpty ())
72+ {
73+ // TODO: ludomikula: finish endpoint authorization
74+ }
7175}
7276}
7377
@@ -77,8 +81,10 @@ public List<RouterFunction<ServerResponse>> registeredEndpoints()
7781return routes ;
7882}
7983
80- private void registerEndpointHandler (String urlPrefix ,PluginEndpoint endpoint ,Method handler )
84+ private List < EndpointExtension > registerEndpointHandler (String urlPrefix ,PluginEndpoint endpoint ,Method handler )
8185{
86+ List <EndpointExtension >toAuthorize =new ArrayList <>();
87+
8288if (handler .isAnnotationPresent (EndpointExtension .class ))
8389{
8490if (checkHandlerMethod (handler ))
@@ -103,13 +109,20 @@ private void registerEndpointHandler(String urlPrefix, PluginEndpoint endpoint,
103109routes .add (routerFunction );
104110registerRouterFunctionMapping (endpointName ,routerFunction );
105111
112+ if (endpointMeta .authenticated ())
113+ {
114+ toAuthorize .add (endpointMeta );
115+ }
116+
106117log .info ("Registered endpoint: {} -> {}: {}" ,endpoint .getClass ().getSimpleName (),endpointMeta .method (),urlPrefix +endpointMeta .uri ());
107118}
108119else
109120{
110121log .error ("Cannot register plugin endpoint: {} -> {}! Handler method must be defined as: public Mono<ServerResponse> {}(ServerRequest request)" ,endpoint .getClass ().getSimpleName (),handler .getName (),handler .getName ());
111122}
112123}
124+
125+ return toAuthorize ;
113126}
114127
115128private void registerRouterFunctionMapping (String endpointName ,RouterFunction <ServerResponse >routerFunction )