feat: manually mirror opencoze's code from bytedance
Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
@@ -0,0 +1,95 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { useLocation, useNavigate } from 'react-router-dom';
|
||||
|
||||
import { renderHook } from '@testing-library/react';
|
||||
import { useCheckLoginBase } from '@coze-foundation/account-base';
|
||||
|
||||
import { signPath, signRedirectKey } from '../../utils/constants';
|
||||
import { checkLoginImpl } from '../../utils';
|
||||
import { useCheckLogin } from '..';
|
||||
|
||||
vi.mock('react-router-dom', () => ({
|
||||
useLocation: vi.fn(),
|
||||
useNavigate: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock('../../src/utils', () => ({
|
||||
checkLoginImpl: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock('@coze-foundation/account-base', () => ({
|
||||
useCheckLoginBase: vi.fn(),
|
||||
}));
|
||||
|
||||
const mockNavigate = vi.fn();
|
||||
const mockUseLocation = vi.mocked(useLocation);
|
||||
const mockUseNavigate = vi.mocked(useNavigate);
|
||||
|
||||
describe('useCheckLogin', () => {
|
||||
const mockLocation = {
|
||||
pathname: '/test',
|
||||
search: '?query=123',
|
||||
} as any;
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
mockUseNavigate.mockReturnValue(mockNavigate);
|
||||
mockUseLocation.mockReturnValue(mockLocation);
|
||||
});
|
||||
|
||||
it('should call useCheckLoginBase with correct parameters', () => {
|
||||
renderHook(() => useCheckLogin({ needLogin: true }));
|
||||
expect(useCheckLoginBase).toBeCalledWith(
|
||||
true,
|
||||
checkLoginImpl,
|
||||
expect.any(Function),
|
||||
);
|
||||
});
|
||||
|
||||
it('should navigate to default loginFallbackPath when not provided and user is not logged in', () => {
|
||||
const mockUseCheckLoginBase = vi.mocked(useCheckLoginBase);
|
||||
mockUseCheckLoginBase.mockImplementation(
|
||||
(needLogin, checkLoginImpl, goLogin) => {
|
||||
goLogin();
|
||||
},
|
||||
);
|
||||
|
||||
renderHook(() => useCheckLogin({ needLogin: true }));
|
||||
|
||||
expect(mockNavigate).toBeCalledWith(
|
||||
`${signPath}?${signRedirectKey}=${encodeURIComponent(`${mockLocation.pathname}${mockLocation.search}`)}`,
|
||||
);
|
||||
});
|
||||
|
||||
it('should navigate to custom loginFallbackPath when provided and user is not logged in', () => {
|
||||
const loginFallbackPath = '/custom-login';
|
||||
const mockUseCheckLoginBase = vi.mocked(useCheckLoginBase);
|
||||
mockUseCheckLoginBase.mockImplementation(
|
||||
(needLogin, checkLoginImpl, goLogin) => {
|
||||
goLogin();
|
||||
},
|
||||
);
|
||||
|
||||
renderHook(() => useCheckLogin({ needLogin: true, loginFallbackPath }));
|
||||
|
||||
expect(mockNavigate).toBeCalledWith(
|
||||
`${loginFallbackPath}${mockLocation.search}`,
|
||||
{ replace: true },
|
||||
);
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,48 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { useLocation, useNavigate } from 'react-router-dom';
|
||||
|
||||
import { useCheckLoginBase } from '@coze-foundation/account-base';
|
||||
|
||||
import { signPath, signRedirectKey } from '../utils/constants';
|
||||
import { checkLoginImpl } from '../utils';
|
||||
|
||||
const useGoLogin = (loginFallbackPath?: string) => {
|
||||
const navigate = useNavigate();
|
||||
const { pathname, search } = useLocation();
|
||||
return () => {
|
||||
const redirectPath = `${pathname}${search}`;
|
||||
if (loginFallbackPath) {
|
||||
navigate(`${loginFallbackPath}${search}`, { replace: true });
|
||||
} else {
|
||||
navigate(
|
||||
`${signPath}?${signRedirectKey}=${encodeURIComponent(redirectPath)}`,
|
||||
);
|
||||
}
|
||||
};
|
||||
};
|
||||
|
||||
export const useCheckLogin = ({
|
||||
needLogin,
|
||||
loginFallbackPath,
|
||||
}: {
|
||||
needLogin?: boolean;
|
||||
loginFallbackPath?: string;
|
||||
}) => {
|
||||
const goLogin = useGoLogin(loginFallbackPath);
|
||||
useCheckLoginBase(!!needLogin, checkLoginImpl, goLogin);
|
||||
};
|
||||
44
frontend/packages/foundation/account-adapter/src/index.ts
Normal file
44
frontend/packages/foundation/account-adapter/src/index.ts
Normal file
@@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
export {
|
||||
getUserInfo,
|
||||
getLoginStatus,
|
||||
resetUserStore,
|
||||
setUserInfo,
|
||||
getUserLabel,
|
||||
useUserInfo,
|
||||
useLoginStatus,
|
||||
useAlterOnLogout,
|
||||
useHasError,
|
||||
useUserLabel,
|
||||
useUserAuthInfo,
|
||||
getUserAuthInfos,
|
||||
subscribeUserAuthInfos,
|
||||
useSyncLocalStorageUid,
|
||||
usernameRegExpValidate,
|
||||
type UserInfo,
|
||||
type LoginStatus,
|
||||
} from '@coze-foundation/account-base';
|
||||
export {
|
||||
refreshUserInfo,
|
||||
logout,
|
||||
checkLogin,
|
||||
connector2Redirect,
|
||||
} from './utils';
|
||||
export { useCheckLogin } from './hooks';
|
||||
|
||||
export { passportApi } from './passport-api';
|
||||
@@ -0,0 +1,95 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { vi, describe, it, expect, beforeEach } from 'vitest';
|
||||
import { passport } from '@coze-studio/api-schema';
|
||||
import { passportApi } from '../index';
|
||||
|
||||
// 模拟 passport API
|
||||
vi.mock('@coze-studio/api-schema/passport', () => ({}));
|
||||
vi.mock('@coze-studio/api-schema', () => ({
|
||||
passport: {
|
||||
PassportAccountInfoV2: vi.fn(),
|
||||
PassportWebLogoutGet: vi.fn(),
|
||||
UserUpdateAvatar: vi.fn(),
|
||||
PassportWebEmailPasswordResetGet: vi.fn(),
|
||||
UserUpdateProfile: vi.fn(),
|
||||
},
|
||||
}));
|
||||
vi.mock('@coze-foundation/account-base', () => ({
|
||||
resetUserStore: vi.fn(),
|
||||
}));
|
||||
|
||||
describe('passportApi', () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('checkLogin', () => {
|
||||
it('should correctly return user information', async () => {
|
||||
const mockUserInfo = { name: 'test' };
|
||||
vi.mocked(passport.PassportAccountInfoV2).mockResolvedValueOnce({
|
||||
data: mockUserInfo,
|
||||
});
|
||||
|
||||
const result = await passportApi.checkLogin();
|
||||
expect(result).toEqual(mockUserInfo);
|
||||
expect(passport.PassportAccountInfoV2).toHaveBeenCalledWith({});
|
||||
});
|
||||
});
|
||||
|
||||
describe('logout', () => {
|
||||
it('should correctly call the logout API', async () => {
|
||||
await passportApi.logout();
|
||||
expect(passport.PassportWebLogoutGet).toHaveBeenCalledWith({
|
||||
next: '/',
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('uploadAvatar', () => {
|
||||
it('should correctly upload avatar', async () => {
|
||||
const mockFile = new File([''], 'test.png');
|
||||
const mockResponse = { data: { url: 'test-url' } };
|
||||
vi.mocked(passport.UserUpdateAvatar).mockResolvedValueOnce(mockResponse);
|
||||
|
||||
const result = await passportApi.uploadAvatar({ avatar: mockFile });
|
||||
expect(result).toEqual(mockResponse.data);
|
||||
expect(passport.UserUpdateAvatar).toHaveBeenCalledWith({
|
||||
avatar: mockFile,
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('updatePassword', () => {
|
||||
it('should correctly call the password reset API', async () => {
|
||||
const params = { password: 'newpass', email: 'test@example.com' };
|
||||
await passportApi.updatePassword(params);
|
||||
expect(passport.PassportWebEmailPasswordResetGet).toHaveBeenCalledWith({
|
||||
...params,
|
||||
code: '',
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('updateUserProfile', () => {
|
||||
it('should correctly update user profile', async () => {
|
||||
const mockProfile = { nickname: 'newname' };
|
||||
await passportApi.updateUserProfile(mockProfile);
|
||||
expect(passport.UserUpdateProfile).toHaveBeenCalledWith(mockProfile);
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,51 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { type UserUpdateProfileRequest } from '@coze-studio/api-schema/passport';
|
||||
import { passport } from '@coze-studio/api-schema';
|
||||
import { resetUserStore, type UserInfo } from '@coze-foundation/account-base';
|
||||
|
||||
export const passportApi = {
|
||||
checkLogin: async () => {
|
||||
const res = (await passport.PassportAccountInfoV2({})) as unknown as {
|
||||
data: UserInfo;
|
||||
};
|
||||
return res.data;
|
||||
},
|
||||
|
||||
logout: async () => {
|
||||
await passport.PassportWebLogoutGet({
|
||||
next: '/',
|
||||
});
|
||||
},
|
||||
|
||||
uploadAvatar: async ({ avatar }: { avatar: File }) => {
|
||||
const res = await passport.UserUpdateAvatar({
|
||||
avatar,
|
||||
});
|
||||
|
||||
return res.data;
|
||||
},
|
||||
|
||||
updatePassword: async (params: { password: string; email: string }) => {
|
||||
await passport.PassportWebEmailPasswordResetGet({ ...params, code: '' });
|
||||
// 更新密码后,当前登录态失效,重置 store
|
||||
resetUserStore();
|
||||
},
|
||||
|
||||
updateUserProfile: (params: UserUpdateProfileRequest) =>
|
||||
passport.UserUpdateProfile(params),
|
||||
};
|
||||
17
frontend/packages/foundation/account-adapter/src/typings.d.ts
vendored
Normal file
17
frontend/packages/foundation/account-adapter/src/typings.d.ts
vendored
Normal file
@@ -0,0 +1,17 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
/// <reference types='@coze-arch/bot-typings' />
|
||||
@@ -0,0 +1,97 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
import { vi, describe, it, expect, beforeEach } from 'vitest';
|
||||
import {
|
||||
refreshUserInfo,
|
||||
logout,
|
||||
checkLoginImpl,
|
||||
checkLogin,
|
||||
connector2Redirect,
|
||||
} from '../index';
|
||||
import {
|
||||
refreshUserInfoBase,
|
||||
logoutBase,
|
||||
checkLoginBase,
|
||||
} from '@coze-foundation/account-base';
|
||||
import { passportApi } from '../../passport-api';
|
||||
|
||||
// Mock dependencies
|
||||
vi.mock('@coze-foundation/account-base', () => ({
|
||||
refreshUserInfoBase: vi.fn(),
|
||||
logoutBase: vi.fn(),
|
||||
checkLoginBase: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock('../../passport-api', () => ({
|
||||
passportApi: {
|
||||
checkLogin: vi.fn(),
|
||||
logout: vi.fn(),
|
||||
},
|
||||
}));
|
||||
|
||||
describe('utils/index.ts', () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
describe('refreshUserInfo', () => {
|
||||
it('should call refreshUserInfoBase with passportApi.checkLogin', () => {
|
||||
refreshUserInfo();
|
||||
expect(refreshUserInfoBase).toHaveBeenCalledWith(passportApi.checkLogin);
|
||||
});
|
||||
});
|
||||
|
||||
describe('logout', () => {
|
||||
it('should call logoutBase with passportApi.logout', () => {
|
||||
logout();
|
||||
expect(logoutBase).toHaveBeenCalledWith(passportApi.logout);
|
||||
});
|
||||
});
|
||||
|
||||
describe('checkLoginImpl', () => {
|
||||
it('should return userInfo when passportApi.checkLogin succeeds', async () => {
|
||||
const mockUserInfo = { id: '123', name: 'test' };
|
||||
vi.mocked(passportApi.checkLogin).mockResolvedValue(mockUserInfo);
|
||||
|
||||
const result = await checkLoginImpl();
|
||||
expect(result).toEqual({ userInfo: mockUserInfo });
|
||||
});
|
||||
|
||||
it('should return undefined userInfo when passportApi.checkLogin fails', async () => {
|
||||
vi.mocked(passportApi.checkLogin).mockRejectedValue(
|
||||
new Error('API error'),
|
||||
);
|
||||
|
||||
const result = await checkLoginImpl();
|
||||
expect(result).toEqual({ userInfo: undefined });
|
||||
});
|
||||
});
|
||||
|
||||
describe('checkLogin', () => {
|
||||
it('should call checkLoginBase with checkLoginImpl', () => {
|
||||
checkLogin();
|
||||
expect(checkLoginBase).toHaveBeenCalledWith(checkLoginImpl);
|
||||
});
|
||||
});
|
||||
|
||||
describe('connector2Redirect', () => {
|
||||
it('should return undefined (open source version)', () => {
|
||||
const result = connector2Redirect();
|
||||
expect(result).toBeUndefined();
|
||||
});
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,18 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
export const signPath = '/sign';
|
||||
export const signRedirectKey = 'redirect';
|
||||
@@ -0,0 +1,44 @@
|
||||
/*
|
||||
* Copyright 2025 coze-dev Authors
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
/* eslint-disable @coze-arch/use-error-in-catch */
|
||||
import {
|
||||
refreshUserInfoBase,
|
||||
logoutBase,
|
||||
checkLoginBase,
|
||||
type Connector2Redirect,
|
||||
} from '@coze-foundation/account-base';
|
||||
|
||||
import { passportApi } from '../passport-api';
|
||||
|
||||
export const refreshUserInfo = () =>
|
||||
refreshUserInfoBase(passportApi.checkLogin);
|
||||
|
||||
export const logout = () => logoutBase(passportApi.logout);
|
||||
|
||||
export const checkLoginImpl = async () => {
|
||||
try {
|
||||
const res = await passportApi.checkLogin();
|
||||
return { userInfo: res };
|
||||
} catch (e) {
|
||||
return { userInfo: undefined };
|
||||
}
|
||||
};
|
||||
|
||||
export const checkLogin = () => checkLoginBase(checkLoginImpl);
|
||||
|
||||
// 开源版本不支持渠道授权,暂无实现
|
||||
export const connector2Redirect: Connector2Redirect = () => undefined;
|
||||
Reference in New Issue
Block a user