git.basschouten.com Git - openhab-addons.git/blob

   1 /**
   2  * Copyright (c) 2010-2023 Contributors to the openHAB project
   3  *
   4  * See the NOTICE file(s) distributed with this work for additional
   5  * information.
   6  *
   7  * This program and the accompanying materials are made available under the
   8  * terms of the Eclipse Public License 2.0 which is available at
   9  * http://www.eclipse.org/legal/epl-2.0
  10  *
  11  * SPDX-License-Identifier: EPL-2.0
  12  */
  13 package org.openhab.binding.homeconnect.internal.client;
  14
  15 import java.io.IOException;
  16 import java.util.concurrent.ConcurrentHashMap;
  17
  18 import javax.ws.rs.client.ClientRequestContext;
  19 import javax.ws.rs.client.ClientRequestFilter;
  20 import javax.ws.rs.client.ClientResponseContext;
  21 import javax.ws.rs.client.ClientResponseFilter;
  22 import javax.ws.rs.core.HttpHeaders;
  23 import javax.ws.rs.core.MultivaluedMap;
  24
  25 import org.eclipse.jdt.annotation.NonNullByDefault;
  26 import org.eclipse.jdt.annotation.Nullable;
  27 import org.slf4j.Logger;
  28 import org.slf4j.LoggerFactory;
  29
  30 /**
  31  * Inserts Authorization header for requests on the streaming REST API.
  32  *
  33  * @author Laurent Garnier - Initial contribution
  34  */
  35 @NonNullByDefault
  36 public class HomeConnectStreamingRequestFilter implements ClientRequestFilter, ClientResponseFilter {
  37
  38     private static final String TEXT_EVENT_STREAM = "text/event-stream";
  39
  40     private final Logger logger = LoggerFactory.getLogger(HomeConnectStreamingRequestFilter.class);
  41     private final ConcurrentHashMap<String, String> authorizationHeaders = new ConcurrentHashMap<>();
  42
  43     @Override
  44     public void filter(@Nullable ClientRequestContext requestContext) throws IOException {
  45         if (requestContext != null) {
  46             MultivaluedMap<String, Object> headers = requestContext.getHeaders();
  47             String authorizationHeader = authorizationHeaders.get(requestContext.getUri().toString());
  48             if (authorizationHeader != null) {
  49                 headers.putSingle(HttpHeaders.AUTHORIZATION, authorizationHeader);
  50             } else {
  51                 logger.warn("No authorization header set! uri={}", requestContext.getUri());
  52             }
  53             headers.putSingle(HttpHeaders.CACHE_CONTROL, "no-cache");
  54             headers.putSingle(HttpHeaders.ACCEPT, TEXT_EVENT_STREAM);
  55         }
  56     }
  57
  58     @Override
  59     public void filter(@Nullable ClientRequestContext requestContext, @Nullable ClientResponseContext responseContext)
  60             throws IOException {
  61         if (logger.isDebugEnabled() && requestContext != null) {
  62             StringBuilder sb = new StringBuilder();
  63             sb.append("SSE connection: ");
  64             sb.append(requestContext.getUri()).append("\n");
  65             requestContext.getHeaders()
  66                     .forEach((name, value) -> sb.append("> ").append(name).append(": ").append(value).append("\n"));
  67
  68             if (responseContext != null) {
  69                 responseContext.getHeaders()
  70                         .forEach((name, value) -> sb.append("< ").append(name).append(": ").append(value).append("\n"));
  71             }
  72
  73             logger.debug("{}", sb);
  74         }
  75     }
  76
  77     public void setAuthorizationHeader(String target, String header) {
  78         logger.debug("Set authorization header. target={}, header={}", target, header);
  79         authorizationHeaders.put(target, header);
  80     }
  81 }