/*
 * Copyright (C) 2008-2023, 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.api.webfinger;

import com.juick.model.User;
import com.juick.www.api.webfinger.model.Account;
import com.juick.www.api.webfinger.model.Link;
import com.juick.util.HttpNotFoundException;
import com.juick.service.UserService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.http.MediaType;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.util.UriComponentsBuilder;
import rocks.xmpp.addr.Jid;

import jakarta.inject.Inject;
import java.util.List;

import static com.juick.www.api.activity.model.Context.ACTIVITY_MEDIA_TYPE;

@RestController
public class Resource {
    public static final String MEDIA_TYPE = "application/jrd+json";
    @Inject
    private UserService userService;
    @Value("${web_domain:localhost}")
    private String domain;
    @Value("${ap_base_uri:http://localhost:8080/}")
    private String baseUri;

    @GetMapping(value = "/.well-known/webfinger", produces = {
            Resource.MEDIA_TYPE, MediaType.APPLICATION_JSON_VALUE })
    public Account getWebResource(@RequestParam String resource) {
        if (resource.startsWith("acct:")) {
            try {
                Jid account = Jid.of(resource.substring(5));
                if (account.getDomain().equals(domain)) {
                    User user = userService.getUserByName(account.getLocal());
                    if (!user.isAnonymous()) {
                        UriComponentsBuilder builder = UriComponentsBuilder.fromHttpUrl(baseUri);
                        builder.path(String.format("/u/%s", user.getName()));
                        Link blog = new Link("self", ACTIVITY_MEDIA_TYPE, builder.toUriString());
                        return new Account(resource, List.of(blog));
                    }
                }
            } catch (NullPointerException | IllegalArgumentException e) {
                throw new HttpNotFoundException();
            }
        }
        throw new HttpNotFoundException();
    }
}