package com.xuqm.im.controller;

import com.xuqm.common.model.ApiResponse;
import com.xuqm.im.entity.ImFriendEntity;
import com.xuqm.im.repository.ImFriendRepository;
import org.springframework.http.ResponseEntity;
import org.springframework.security.core.annotation.AuthenticationPrincipal;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

@RestController
@RequestMapping("/api/im/friends")
public class FriendController {

    private final ImFriendRepository friendRepository;

    public FriendController(ImFriendRepository friendRepository) {
        this.friendRepository = friendRepository;
    }

    @GetMapping
    public ResponseEntity<ApiResponse<List<String>>> listFriends(
            @AuthenticationPrincipal String userId,
            @RequestParam String appId) {
        List<String> friendIds = friendRepository.findByAppIdAndUserId(appId, userId)
                .stream()
                .map(ImFriendEntity::getFriendId)
                .toList();
        return ResponseEntity.ok(ApiResponse.success(friendIds));
    }

    @PostMapping
    public ResponseEntity<ApiResponse<ImFriendEntity>> addFriend(
            @AuthenticationPrincipal String userId,
            @RequestParam String appId,
            @RequestParam String friendId) {
        // Insert userId -> friendId if not already present
        ImFriendEntity forward = friendRepository
                .findByAppIdAndUserIdAndFriendId(appId, userId, friendId)
                .orElseGet(() -> {
                    ImFriendEntity e = new ImFriendEntity();
                    e.setAppId(appId);
                    e.setUserId(userId);
                    e.setFriendId(friendId);
                    return friendRepository.save(e);
                });

        // Insert friendId -> userId bi-directionally if not already present
        friendRepository.findByAppIdAndUserIdAndFriendId(appId, friendId, userId)
                .orElseGet(() -> {
                    ImFriendEntity e = new ImFriendEntity();
                    e.setAppId(appId);
                    e.setUserId(friendId);
                    e.setFriendId(userId);
                    return friendRepository.save(e);
                });

        return ResponseEntity.ok(ApiResponse.success(forward));
    }

    @DeleteMapping("/{friendId}")
    public ResponseEntity<ApiResponse<Void>> removeFriend(
            @AuthenticationPrincipal String userId,
            @PathVariable String friendId,
            @RequestParam String appId) {
        // Remove both directions
        friendRepository.deleteByAppIdAndUserIdAndFriendId(appId, userId, friendId);
        friendRepository.deleteByAppIdAndUserIdAndFriendId(appId, friendId, userId);
        return ResponseEntity.ok(ApiResponse.success(null));
    }
}