feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf
This commit is contained in:
fanlv
2025-07-20 17:36:12 +08:00
commit 890153324f
14811 changed files with 1923430 additions and 0 deletions

View File

@@ -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 },
);
});
});

View File

@@ -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);
};

View 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';

View File

@@ -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);
});
});
});

View File

@@ -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),
};

View 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' />

View File

@@ -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();
});
});
});

View File

@@ -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';

View 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.
*/
/* 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;