Spaces:

Jack698
/

velo

Sleeping

velo / web /src /hooks /useAuthForm.js

Upload folder using huggingface_hub

adc1e1c verified about 1 year ago

4.89 kB

/*
Copyright (c) 2025 Tethys Plex
This file is part of Veloera.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
import { useState, useEffect, useContext } from 'react';
import { useNavigate, useSearchParams } from 'react-router-dom';
import { UserContext } from '../context/User';
import { API, showError, showInfo, showSuccess, updateAPI } from '../helpers';
import { setUserData } from '../helpers/data.js';
import { TELEGRAM_LOGIN_FIELDS, VALIDATION_MESSAGES } from '../utils/authConstants';

export const useAuthForm = (initialInputs = {}) => {
  const [inputs, setInputs] = useState(initialInputs);
  const [searchParams] = useSearchParams();
  const [userState, userDispatch] = useContext(UserContext);
  const [status, setStatus] = useState({});
  const [showWeChatLoginModal, setShowWeChatLoginModal] = useState(false);
  const navigate = useNavigate();

  // Handle input changes
  const handleChange = (name, value) => {
    setInputs((inputs) => ({ ...inputs, [name]: value }));
  };

  // Load status from localStorage and listen for updates
  useEffect(() => {
    const updateStatus = () => {
      let status = localStorage.getItem('status');
      if (status) {
        status = JSON.parse(status);
        setStatus(status);
      }
    };

    // Initial status load
    updateStatus();

    // Listen for storage changes to handle async status loading
    const handleStorageChange = (e) => {
      if (e.key === 'status') {
        updateStatus();
      }
    };

    // Listen for custom status update events
    const handleStatusUpdate = () => {
      updateStatus();
    };

    window.addEventListener('storage', handleStorageChange);
    window.addEventListener('statusUpdated', handleStatusUpdate);

    return () => {
      window.removeEventListener('storage', handleStorageChange);
      window.removeEventListener('statusUpdated', handleStatusUpdate);
    };
  }, []);

  // Handle AFF code processing
  const processAffCode = () => {
    let affCode = null;
    let statusFromStorage = localStorage.getItem('status');
    if (statusFromStorage) {
      statusFromStorage = JSON.parse(statusFromStorage);
      if (statusFromStorage.aff_enabled === true) {
        affCode = new URLSearchParams(window.location.search).get('aff');
        if (affCode) {
          localStorage.setItem('aff', affCode);
        }
      }
    }
    return affCode;
  };

  // Re-process AFF code when status changes
  useEffect(() => {
    processAffCode();
  }, [status]);

  // WeChat login handlers
  const onWeChatLoginClicked = () => {
    setShowWeChatLoginModal(true);
  };

  const onSubmitWeChatVerificationCode = async (turnstileToken) => {
    if (status.turnstile_check && turnstileToken === '') {
      showInfo(VALIDATION_MESSAGES.TURNSTILE_WAIT);
      return;
    }
    const res = await API.get(
      `/api/oauth/wechat?code=${inputs.wechat_verification_code}`,
    );
    const { success, message, data } = res.data;
    if (success) {
      userDispatch({ type: 'login', payload: data });
      localStorage.setItem('user', JSON.stringify(data));
      setUserData(data);
      updateAPI();
      navigate(searchParams.get('returnTo') || '/');
      showSuccess(VALIDATION_MESSAGES.LOGIN_SUCCESS);
      setShowWeChatLoginModal(false);
    } else {
      showError(message);
    }
  };

  // Telegram login handler
  const onTelegramLoginClicked = async (response) => {
    const params = {};
    TELEGRAM_LOGIN_FIELDS.forEach((field) => {
      if (response[field]) {
        params[field] = response[field];
      }
    });
    const res = await API.get(`/api/oauth/telegram/login`, { params });
    const { success, message, data } = res.data;
    if (success) {
      userDispatch({ type: 'login', payload: data });
      localStorage.setItem('user', JSON.stringify(data));
      showSuccess(VALIDATION_MESSAGES.LOGIN_SUCCESS);
      setUserData(data);
      updateAPI();
      navigate(searchParams.get('returnTo') || '/');
    } else {
      showError(message);
    }
  };

  return {
    inputs,
    setInputs,
    handleChange,
    status,
    setStatus,
    showWeChatLoginModal,
    setShowWeChatLoginModal,
    userDispatch,
    navigate,
    searchParams,
    processAffCode,
    onWeChatLoginClicked,
    onSubmitWeChatVerificationCode,
    onTelegramLoginClicked,
  };
};

	/*
	Copyright (c) 2025 Tethys Plex

	This file is part of Veloera.

	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU 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 General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with this program. If not, see <https://www.gnu.org/licenses/>.
	*/
	import { useState, useEffect, useContext } from 'react';
	import { useNavigate, useSearchParams } from 'react-router-dom';
	import { UserContext } from '../context/User';
	import { API, showError, showInfo, showSuccess, updateAPI } from '../helpers';
	import { setUserData } from '../helpers/data.js';
	import { TELEGRAM_LOGIN_FIELDS, VALIDATION_MESSAGES } from '../utils/authConstants';

	export const useAuthForm = (initialInputs = {}) => {
	const [inputs, setInputs] = useState(initialInputs);
	const [searchParams] = useSearchParams();
	const [userState, userDispatch] = useContext(UserContext);
	const [status, setStatus] = useState({});
	const [showWeChatLoginModal, setShowWeChatLoginModal] = useState(false);
	const navigate = useNavigate();

	// Handle input changes
	const handleChange = (name, value) => {
	setInputs((inputs) => ({ ...inputs, [name]: value }));
	};

	// Load status from localStorage and listen for updates
	useEffect(() => {
	const updateStatus = () => {
	let status = localStorage.getItem('status');
	if (status) {
	status = JSON.parse(status);
	setStatus(status);
	}
	};

	// Initial status load
	updateStatus();

	// Listen for storage changes to handle async status loading
	const handleStorageChange = (e) => {
	if (e.key === 'status') {
	updateStatus();
	}
	};

	// Listen for custom status update events
	const handleStatusUpdate = () => {
	updateStatus();
	};

	window.addEventListener('storage', handleStorageChange);
	window.addEventListener('statusUpdated', handleStatusUpdate);

	return () => {
	window.removeEventListener('storage', handleStorageChange);
	window.removeEventListener('statusUpdated', handleStatusUpdate);
	};
	}, []);

	// Handle AFF code processing
	const processAffCode = () => {
	let affCode = null;
	let statusFromStorage = localStorage.getItem('status');
	if (statusFromStorage) {
	statusFromStorage = JSON.parse(statusFromStorage);
	if (statusFromStorage.aff_enabled === true) {
	affCode = new URLSearchParams(window.location.search).get('aff');
	if (affCode) {
	localStorage.setItem('aff', affCode);
	}
	}
	}
	return affCode;
	};

	// Re-process AFF code when status changes
	useEffect(() => {
	processAffCode();
	}, [status]);

	// WeChat login handlers
	const onWeChatLoginClicked = () => {
	setShowWeChatLoginModal(true);
	};

	const onSubmitWeChatVerificationCode = async (turnstileToken) => {
	if (status.turnstile_check && turnstileToken === '') {
	showInfo(VALIDATION_MESSAGES.TURNSTILE_WAIT);
	return;
	}
	const res = await API.get(
	`/api/oauth/wechat?code=${inputs.wechat_verification_code}`,
	);
	const { success, message, data } = res.data;
	if (success) {
	userDispatch({ type: 'login', payload: data });
	localStorage.setItem('user', JSON.stringify(data));
	setUserData(data);
	updateAPI();
	navigate(searchParams.get('returnTo') \|\| '/');
	showSuccess(VALIDATION_MESSAGES.LOGIN_SUCCESS);
	setShowWeChatLoginModal(false);
	} else {
	showError(message);
	}
	};

	// Telegram login handler
	const onTelegramLoginClicked = async (response) => {
	const params = {};
	TELEGRAM_LOGIN_FIELDS.forEach((field) => {
	if (response[field]) {
	params[field] = response[field];
	}
	});
	const res = await API.get(`/api/oauth/telegram/login`, { params });
	const { success, message, data } = res.data;
	if (success) {
	userDispatch({ type: 'login', payload: data });
	localStorage.setItem('user', JSON.stringify(data));
	showSuccess(VALIDATION_MESSAGES.LOGIN_SUCCESS);
	setUserData(data);
	updateAPI();
	navigate(searchParams.get('returnTo') \|\| '/');
	} else {
	showError(message);
	}
	};

	return {
	inputs,
	setInputs,
	handleChange,
	status,
	setStatus,
	showWeChatLoginModal,
	setShowWeChatLoginModal,
	userDispatch,
	navigate,
	searchParams,
	processAffCode,
	onWeChatLoginClicked,
	onSubmitWeChatVerificationCode,
	onTelegramLoginClicked,
	};
	};