import { createSlice, createAsyncThunk, PayloadAction } from '@reduxjs/toolkit';
import axios from '../../axios';

// Типы данных
interface AuthState {
    jwt: string | null;
    refreshToken: string | null;
    username: string | null;
    status: 'idle' | 'loading' | 'successful' | 'failed';
    error: string | null;
}

// Инициализация состояния
const initialState: AuthState = {
    jwt: null,
    refreshToken: null,
    username: null,
    status: 'idle',
    error: null,
};

// AsyncThunk: Регистрация
export const registerUser = createAsyncThunk(
    'auth/register',
    async (
        {
            username,
            email,
            password,
        }: { username: string; email: string; password: string },
        { rejectWithValue },
    ) => {
        try {
            const response = await axios.post('/authentication/register', {
                username,
                email,
                password,
            });
            return response.data; // { jwt, refreshToken }
        } catch (err: any) {
            return rejectWithValue(
                err.response?.data?.message || 'Registration failed',
            );
        }
    },
);

// AsyncThunk: Логин
export const loginUser = createAsyncThunk(
    'auth/login',
    async (
        { username, password }: { username: string; password: string },
        { rejectWithValue },
    ) => {
        try {
            const response = await axios.post('/authentication/login', {
                username,
                password,
            });
            return response.data; // { jwt, refreshToken }
        } catch (err: any) {
            return rejectWithValue(
                err.response?.data?.message || 'Login failed',
            );
        }
    },
);

// AsyncThunk: Обновление токена
export const refreshToken = createAsyncThunk(
    'auth/refresh',
    async ({ refreshToken }: { refreshToken: string }, { rejectWithValue }) => {
        try {
            const response = await axios.post('/authentication/refresh', {
                refreshToken,
            });
            return response.data; // { username }
        } catch (err: any) {
            return rejectWithValue(
                err.response?.data?.message || 'Refresh token failed',
            );
        }
    },
);

// AsyncThunk: Получение информации о пользователе
export const fetchWhoAmI = createAsyncThunk(
    'auth/whoami',
    async (_, { rejectWithValue }) => {
        try {
            const response = await axios.get('/authentication/whoami');
            return response.data; // { username }
        } catch (err: any) {
            return rejectWithValue(
                err.response?.data?.message || 'Failed to fetch user info',
            );
        }
    },
);

// AsyncThunk: Загрузка токенов из localStorage
export const loadTokensFromLocalStorage = createAsyncThunk(
    'auth/loadTokens',
    async (_, {}) => {
        const jwt = localStorage.getItem('jwt');
        const refreshToken = localStorage.getItem('refreshToken');

        if (jwt && refreshToken) {
            axios.defaults.headers.common['Authorization'] = `Bearer ${jwt}`;
            return { jwt, refreshToken };
        } else {
            return { jwt: null, refreshToken: null };
        }
    },
);

// Slice
const authSlice = createSlice({
    name: 'auth',
    initialState,
    reducers: {
        logout: (state) => {
            state.jwt = null;
            state.refreshToken = null;
            state.username = null;
            state.status = 'idle';
            state.error = null;
            localStorage.removeItem('jwt');
            localStorage.removeItem('refreshToken');
            delete axios.defaults.headers.common['Authorization'];
        },
    },
    extraReducers: (builder) => {
        // Регистрация
        builder.addCase(registerUser.pending, (state) => {
            state.status = 'loading';
            state.error = null;
        });
        builder.addCase(
            registerUser.fulfilled,
            (
                state,
                action: PayloadAction<{ jwt: string; refreshToken: string }>,
            ) => {
                state.status = 'successful';
                state.jwt = action.payload.jwt;
                state.refreshToken = action.payload.refreshToken;
                axios.defaults.headers.common[
                    'Authorization'
                ] = `Bearer ${action.payload.jwt}`;
                localStorage.setItem('jwt', action.payload.jwt);
                localStorage.setItem(
                    'refreshToken',
                    action.payload.refreshToken,
                );
            },
        );
        builder.addCase(
            registerUser.rejected,
            (state, action: PayloadAction<any>) => {
                state.status = 'failed';
                state.error = action.payload;
            },
        );

        // Логин
        builder.addCase(loginUser.pending, (state) => {
            state.status = 'loading';
            state.error = null;
        });
        builder.addCase(
            loginUser.fulfilled,
            (
                state,
                action: PayloadAction<{ jwt: string; refreshToken: string }>,
            ) => {
                state.status = 'successful';
                state.jwt = action.payload.jwt;
                state.refreshToken = action.payload.refreshToken;
                axios.defaults.headers.common[
                    'Authorization'
                ] = `Bearer ${action.payload.jwt}`;
                localStorage.setItem('jwt', action.payload.jwt);
                localStorage.setItem(
                    'refreshToken',
                    action.payload.refreshToken,
                );
            },
        );
        builder.addCase(
            loginUser.rejected,
            (state, action: PayloadAction<any>) => {
                state.status = 'failed';
                state.error = action.payload;
            },
        );

        // Обновление токена
        builder.addCase(refreshToken.pending, (state) => {
            state.status = 'loading';
            state.error = null;
        });
        builder.addCase(
            refreshToken.fulfilled,
            (state, action: PayloadAction<{ username: string }>) => {
                state.status = 'successful';
                state.username = action.payload.username;
            },
        );
        builder.addCase(
            refreshToken.rejected,
            (state, action: PayloadAction<any>) => {
                state.status = 'failed';
                state.error = action.payload;
            },
        );

        // Получение информации о пользователе
        builder.addCase(fetchWhoAmI.pending, (state) => {
            state.status = 'loading';
            state.error = null;
        });
        builder.addCase(
            fetchWhoAmI.fulfilled,
            (state, action: PayloadAction<{ username: string }>) => {
                state.status = 'successful';
                state.username = action.payload.username;
            },
        );
        builder.addCase(
            fetchWhoAmI.rejected,
            (state, action: PayloadAction<any>) => {
                state.status = 'failed';
                state.error = action.payload;
            },
        );

        // Загрузка токенов из localStorage
        builder.addCase(
            loadTokensFromLocalStorage.fulfilled,
            (
                state,
                action: PayloadAction<{
                    jwt: string | null;
                    refreshToken: string | null;
                }>,
            ) => {
                state.jwt = action.payload.jwt;
                state.refreshToken = action.payload.refreshToken;
                if (action.payload.jwt) {
                    axios.defaults.headers.common[
                        'Authorization'
                    ] = `Bearer ${action.payload.jwt}`;
                }
            },
        );
    },
});

export const { logout } = authSlice.actions;
export const authReducer = authSlice.reducer;