src/main/java/HttpRequest.java - master - dixsu/websocket-client

src/main/java/HttpRequest.java
216 строк10 KB
Сахно Роман Александрович
разбираюсь отправкой зазипованных данных, планируется переделать отправку заголовков отдельно от тела. для разных типов данных как для текстовых так и для бинарных. т.к. зазипованные данные всегда будут бинарными в том числе и для текстового типа контента
09 фев 2025, 08:06
09 фев 2025, 08:0642d6d18
import org.jetbrains.annotations.NotNull;

import java.util.HashMap;
import java.util.Map;

public class HttpRequest {
//    private final Map<String, String> cookiesMap = new HashMap<>(); // Хранилище для кук
    private Map<String, Map<String, String>> requestForProxyCookiesMap = new HashMap<>();
    private Map<String, Map<String, String>> requestForProxyHeadersMap = new HashMap<>();

    // Метод для извлечения requestId из заголовков
    protected static String extractRequestId(String message) {
        for (String line : message.split("\n")) {
            if (line.startsWith("X-Request-Id:")) {
                return line.substring("X-Request-Id:".length()).trim();
            }
        }
        return null;
    }

    /**
     * Проверяет, является ли запрос запросом к HTML-странице.
     */
//    protected boolean isHtmlPageRequest(@NotNull String resourcePath) {
//        // Если путь заканчивается на "/" или на ".html", это запрос к HTML-странице
//        if (resourcePath.isEmpty() || resourcePath.equals("/") || resourcePath.endsWith(".html") || resourcePath.endsWith(".htm") || resourcePath.endsWith(".js") || resourcePath.endsWith(".css") || !resourcePath.contains("."))
//        {
//            UtilsLayer.logClient("ресурс определён как html", true);
//            return true;
//        }
//        UtilsLayer.logClient("ресурс НЕ определён как html", true);
//        return false;
//    }


//    protected boolean isTextResponse(@NotNull String contentType) {
//        // Проверяем, текстовые ли это данные
//        boolean isTextResponse = contentType != null && contentType.startsWith("text/") ||
//                contentType != null && contentType.contains("javascript") ||
//                contentType != null && contentType.contains("json") ||
//                contentType != null && contentType.contains("xml");
//        UtilsLayer.logClient("Контент тип ресурса определены как Текстовые данные? = ["+isTextResponse+"]", true);
//        return isTextResponse;
//    }
//
//    protected boolean isGzipped(@NotNull HttpResponse httpResponse) {
//        // Проверяем, сжат ли контент
//        boolean isGzipped = httpResponse.getHeaders().containsKey("Content-Encoding") &&
//                "gzip".equalsIgnoreCase(httpResponse.getHeaders().get("Content-Encoding"));
//        UtilsLayer.logClient("Контент ресурса сжат? = ["+isGzipped+"]", true);
//        return isGzipped;
//    }
//





    /**
     * Проверяет, является ли запрос запросом к HTML-странице.
     */
    protected static boolean isHtmlPure(@NotNull String resourcePath) {
        // Если путь заканчивается на "/" или на ".html", это запрос к HTML-странице
        if (resourcePath.isEmpty() || resourcePath.equals("/") || resourcePath.endsWith(".html") || resourcePath.endsWith(".htm") || !resourcePath.contains("."))
        {
            UtilsLayer.logClient("ресурс определён как html", true);
            return true;
        }
        UtilsLayer.logClient("ресурс НЕ определён как html", true);
        return false;
    }

    public static String getContentType(String resourcePath) {
        if (resourcePath.equals("")) {
            UtilsLayer.logClient(resourcePath+" text");
            return "text/html";
    } else if (!resourcePath.contains(".")) { // Проверка на отсутствие расширения
        UtilsLayer.logClient(resourcePath + " directory");
        return "text/html"; // Возвращаем text/html для путей без расширения
    } else
        if (resourcePath.endsWith(".tmpl")) {
            UtilsLayer.logClient(resourcePath+" text");
            return "text/css";
        } else if (resourcePath.endsWith(".html") || resourcePath.endsWith(".htm")) {
            UtilsLayer.logClient(resourcePath+" text");
            return "text/html";
        } else if (resourcePath.endsWith(".css")) {
            UtilsLayer.logClient(resourcePath+" text");
            return "text/css";
        } else if (resourcePath.endsWith(".js")) {
            UtilsLayer.logClient(resourcePath+" js");
            return "application/javascript";
        } else if (resourcePath.endsWith(".png")) {

            UtilsLayer.logClient(resourcePath+" png");
            return "image/png";
        } else if (resourcePath.endsWith(".jpg") || resourcePath.endsWith(".jpeg")) {

            UtilsLayer.logClient(resourcePath+" jpeg");
            return "image/jpeg";
        } else if (resourcePath.endsWith(".gif")) {

            UtilsLayer.logClient(resourcePath+" gif");
            return "image/gif";
        } else if (resourcePath.endsWith(".svg")) {

            UtilsLayer.logClient(resourcePath+" svg");
            return "image/svg+xml";
        }
         else if (resourcePath.endsWith(".webp")) {
            UtilsLayer.logClient(resourcePath + " image/webp");
            return "image/webp";
        } else if (resourcePath.endsWith(".ico")) {
            UtilsLayer.logClient(resourcePath+" ico");
            return "image/x-icon";
        } else if (resourcePath.endsWith(".woff")) {
            UtilsLayer.logClient(resourcePath+" woff");
            return "font/woff";
        } else if (resourcePath.endsWith(".woff2")) {
            UtilsLayer.logClient(resourcePath+" woff2");
            return "font/woff2";
        } else if (resourcePath.endsWith(".ttf")) {
            UtilsLayer.logClient(resourcePath+" tft");
            return "font/ttf";
        } else if (resourcePath.endsWith(".eot")) {
            UtilsLayer.logClient(resourcePath+" fontonject");
            return "application/vnd.ms-fontobject";
        } else if (resourcePath.endsWith(".otf")) {
            UtilsLayer.logClient(resourcePath+" otf");
            return "font/otf";
        } else if (resourcePath.endsWith(".html") || resourcePath.endsWith(".htm")) {
            return "text/html";
        } else if ((resourcePath.endsWith(".json")||(resourcePath.endsWith(".map")))) {
            return "application/json";
        } else if (resourcePath.endsWith(".xml")) {
            return "application/xml";
        } else if (resourcePath.endsWith(".txt")) {
            return "text/plain";
        } else if (resourcePath.endsWith(".pdf")) {
            return "application/pdf";
        } else if (resourcePath.endsWith(".zip")) {
            return "application/zip";
        } else if (resourcePath.endsWith(".tar")) {
            return "application/x-tar";
        } else if (resourcePath.endsWith(".gz")) {
            return "application/gzip";
        } else if (resourcePath.endsWith(".mp3")) {
            return "audio/mpeg";
        } else if (resourcePath.endsWith(".wav")) {
            return "audio/wav";
        } else if (resourcePath.endsWith(".mp4")) {
            return "video/mp4";
        } else if (resourcePath.endsWith(".webm")) {
            return "video/webm";
        } else if (resourcePath.endsWith(".ogg")) {
            return "audio/ogg";
        } else if (resourcePath.endsWith(".webp")) {
            return "image/webp";
        } else if (resourcePath.endsWith(".csv")) {
            return "text/csv";
        } else if (resourcePath.endsWith(".doc")) {
            return "application/msword";
        } else if (resourcePath.endsWith(".docx")) {
            return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
        } else if (resourcePath.endsWith(".xls")) {
            return "application/vnd.ms-excel";
        } else if (resourcePath.endsWith(".xlsx")) {
            return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
        } else if (resourcePath.endsWith(".ppt")) {
            return "application/vnd.ms-powerpoint";
        } else if (resourcePath.endsWith(".pptx")) {
            return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
        } else {
            return "application/octet-stream"; // По умолчанию для неизвестных типов
        }
    }


    protected void saveCookies(String requestId, String cookie) {
        // Получаем хранилище кук для запроса
        Map<String, String> cookiesMap = requestForProxyCookiesMap.computeIfAbsent(requestId, k -> new HashMap<>());

        // Сохраняем куки в хранилище
        String[] cookieParts = cookie.split(";");
        for (String part : cookieParts) {
            String[] keyValue = part.trim().split("=", 2);
            if (keyValue.length == 2) {
                cookiesMap.put(keyValue[0], keyValue[1]);
            }
        }
    }

    protected void saveHeaders(String requestId, Map<String, String> headers) {
        requestForProxyHeadersMap.put(requestId, headers);
        //не забыть потом удалять их из мапы ну нужном этапе
    }

    protected Map<String, String> getHeadersForRequest(String requestId) {
        return requestForProxyHeadersMap.get(requestId);
    }

    protected String getCookiesForRequest(String requestId) {
        // Получаем хранилище кук для запроса
        Map<String, String> cookiesMap = requestForProxyCookiesMap.get(requestId);
        if (cookiesMap == null || cookiesMap.isEmpty()) {
            return null;
        }

        // Формируем строку кук для запроса
        StringBuilder cookiesBuilder = new StringBuilder();
        for (Map.Entry<String, String> entry : cookiesMap.entrySet()) {
            cookiesBuilder.append(entry.getKey()).append("=").append(entry.getValue()).append("; ");
        }
        return cookiesBuilder.toString();
    }
}