/*
 * Copyright (C) 2008-2020, Juick
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.juick.www.filters;

import com.juick.model.User;
import com.juick.util.WebUtils;

import jakarta.servlet.FilterChain;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;

import com.juick.service.MessagesService;
import com.juick.service.UserService;
import org.apache.commons.lang3.math.NumberUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
import org.springframework.web.util.UriComponents;

import javax.inject.Inject;
import java.io.IOException;

@Component
public class AnythingFilter extends OncePerRequestFilter {
    @Inject
    private MessagesService messagesService;
    @Inject
    private UserService userService;

    @Override
    public void doFilterInternal(HttpServletRequest servletRequest,
                                 HttpServletResponse servletResponse,
                                 FilterChain filterChain) throws IOException, ServletException {
        String upgrade = servletRequest.getHeader("Connection");
        if (upgrade != null && upgrade.equals("Upgrade")) {
            filterChain.doFilter(servletRequest, servletResponse);
            return;
        }
        UriComponents components = ServletUriComponentsBuilder.fromCurrentRequestUri().build();
        String anything = components.getPath().substring(1);
        int before = NumberUtils.toInt(components.getQueryParams().getFirst("before"), 0);
        if (before == 0) {
            boolean isPostNumber = WebUtils.isPostNumber(anything);
            int messageId = isPostNumber ?
                    NumberUtils.toInt(anything) : 0;

            if (isPostNumber && anything.equals(Integer.toString(messageId))) {
                if (messageId > 0) {
                    User author = messagesService.getMessageAuthor(messageId);

                    if (author != null) {
                        servletResponse.setStatus(HttpServletResponse.SC_MOVED_PERMANENTLY);
                        servletResponse.setHeader("Location", "/" + author.getName() + "/" + anything);
                        return;
                    }
                }
            }
             ;
            if (userService.getUsernames().stream().anyMatch(name -> name.equals(anything))) {
                servletResponse.sendRedirect("/" + anything + "/");
            } else {
                filterChain.doFilter(servletRequest, servletResponse);
            }
        } else {
            if (userService.getUsernames().stream().anyMatch(name -> name.equals(anything))) {
                servletResponse.sendRedirect("/" + anything + "/?before=" + before);
            } else {
                filterChain.doFilter(servletRequest, servletResponse);
            }
        }
    }
}