Facti13.Web.Backend.Java.V2/src/main/java/app/annotations/impl/WaitAfterNextAspect.java

package app.annotations.impl;

import app.annotations.exceptions.NeedCookie;
import app.annotations.exceptions.WaitRateLimit;
import app.annotations.interfaces.WaitAfterNext;
import jakarta.servlet.http.HttpServletRequest;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.After;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.context.annotation.Configuration;

import java.util.HashSet;

/**
 * АОП для обработки рейт лимитов
 */
@Aspect
@Configuration
public class WaitAfterNextAspect {
    HashSet<String> wait_order = new HashSet<>();

    @Before("@annotation(app.annotations.interfaces.WaitAfterNext) && args(request,..)")
    public void before(JoinPoint joinPoint, HttpServletRequest request) {
        final String order = ((MethodSignature) joinPoint.getSignature()).getMethod().getAnnotation(WaitAfterNext.class).order();
        final String hash = new StringBuilder().append(getSteam64fromCookie(request)).append(getIp(request)).append(order).toString();

        if (hash.isEmpty()) return;
        if (wait_order.contains(hash)) throw new WaitRateLimit();
        wait_order.add(hash);
    }

    @After("@annotation(app.annotations.interfaces.WaitAfterNext) && args(request,..)")
    public void after(JoinPoint joinPoint, HttpServletRequest request) {
        final String order = ((MethodSignature) joinPoint.getSignature()).getMethod().getAnnotation(WaitAfterNext.class).order();
        final String hash = new StringBuilder().append(getSteam64fromCookie(request)).append(getIp(request)).append(order).toString();
        if (hash.isEmpty()) return;
        if (wait_order.contains(hash)) wait_order.remove(hash);
    }

    public String getSteam64fromCookie(HttpServletRequest request) {
        if(request.getHeader("Cookie") == null) {
            return "";
        }

        String[] rawCookieParams = request.getHeader("Cookie").split(";");
        String steam64 = "";
        for(String rawCookie: rawCookieParams) {
            if(rawCookie.contains("steam64=")) {
                steam64 = rawCookie.split("=")[1];
                continue;
            }
        }
        return steam64;
    }

    public String getIp(HttpServletRequest request) {
        if (request.getHeader("X-Forwarded-For") == null) {
            return "";
        }

        return request.getHeader("X-Forwarded-For").split(",")[0];
    }
}