需求:ant design pro的后台管理系统默认是必须要登陆的,如果不登陆则右边个人窗口是空白,现在要实现一个功能,判断当前是否登录,如果登录的话,显示原来的个人账号信息。如果没登陆,则显示请登录按钮,当点击按钮后,会跳转到登录页面(/login)。
这是官方登陆后的右上角用户信息。
这是魔改免登录后的右上角。是个空白
步骤一:一定要清楚了解需求的意思。如果不清楚就问产品经理,复述一遍。直到双方都确认无误。再开始做。否则会南辕北辙。
步骤二:定位代码。浏览器F12,鼠标放到要修改的地方,尽可能多的搜集class,id或者页面文本。
这个地方举例子:
搜集到一下信息:serati ma、退出登录、ant-dropdown等class样式。
打开Web strom编译器,CTRL+SHIFT+R会打开全局搜索,一个一个去搜索页面上看到的蛛丝马迹,知道定位到具体位置位置。如果找不到是因为没顺藤摸瓜,可能文件A引用了文件B。要分析他们的关联,直到找出最可疑的文件,最终在这个最可疑的文件上修改,刷新页面可能会报错,或者显示出你改的内容。这就算定位到要修改的文件了。
打开大模型
输入你要改的需求,并把刚才定位到的整个文件代码附上。
如果登陆后会显示用户信息,但是现在没登录好像因为children为空,所以页面上什么不显示我想加一个判断,如果没登陆,或者children为空,则显示请登录按钮,点击跳转/login页面
import { outLogin } from '@/services/ant-design-pro/api';
import { LogoutOutlined, SettingOutlined, UserOutlined } from '@ant-design/icons';
import { history, useModel } from '@umijs/max';
import { Spin } from 'antd';
import { createStyles } from 'antd-style';
import { stringify } from 'querystring';
import React, { useCallback } from 'react';
import { flushSync } from 'react-dom';
import HeaderDropdown from '../HeaderDropdown';
export type GlobalHeaderRightProps = {
menu?: boolean;
children?: React.ReactNode;
};
export const AvatarName = () => {
const { initialState } = useModel('@@initialState');
const { currentUser } = initialState || {};
return <span className="anticon">{currentUser?.name}</span>;
};
const useStyles = createStyles(({ token }) => {
return {
action: {
display: 'flex',
height: '48px',
marginLeft: 'auto',
overflow: 'hidden',
alignItems: 'center',
padding: '0 8px',
cursor: 'pointer',
borderRadius: token.borderRadius,
'&:hover': {
backgroundColor: token.colorBgTextHover,
},
},
};
});
export const AvatarDropdown: React.FC<GlobalHeaderRightProps> = ({ menu, children }) => {
/**
* 退出登录,并且将当前的 url 保存
*/
const loginOut = async () => {
await outLogin();
const { search, pathname } = window.location;
const urlParams = new URL(window.location.href).searchParams;
/** 此方法会跳转到 redirect 参数所在的位置 */
const redirect = urlParams.get('redirect');
// Note: There may be security issues, please note
if (window.location.pathname !== '/user/login' && !redirect) {
history.replace({
pathname: '/user/login',
search: stringify({
redirect: pathname + search,
}),
});
}
};
const { styles } = useStyles();
const { initialState, setInitialState } = useModel('@@initialState');
const onMenuClick = useCallback(
(event: any) => {
const { key } = event;
if (key === 'logout') {
flushSync(() => {
setInitialState((s) => ({ ...s, currentUser: undefined }));
});
loginOut();
return;
}
history.push(`/account/${key}`);
},
[setInitialState],
);
const loading = (
<span className={styles.action}>
<Spin
size="small"
style={{
marginLeft: 8,
marginRight: 8,
}}
/>
</span>
);
if (!initialState) {
return loading;
}
const { currentUser } = initialState;
let menuItems = [
...(menu
? [
{
key: 'center',
icon: <UserOutlined />,
label: '个人中心',
},
{
key: 'settings',
icon: <SettingOutlined />,
label: '个人设置',
},
{
type: 'divider' as const,
},
]
: []),
{
key: 'logout',
icon: <LogoutOutlined />,
label: '退出登录',
},
];
return (
<HeaderDropdown
menu={{
selectedKeys: [],
onClick: onMenuClick,
items: menuItems,
}}
>
{children}
</HeaderDropdown>
);
};
会给出修改结果,拿着代码复制到编译器。刷新页面后。即可完成效果。
有时候问大模型的提示词清晰,具体,那大概率一次就可以完成功能。
如果一次完成不了 甚至复制过去报错,那就要继续把报错提示给他,让他继续完善,甚至是修改提示词,让提示词更容易理解更清晰。不断修改提示词,当机器人明白你的意思,并且不断根据报错完善代码。最终99%的功能都可以实现。
刚开始这个过程可能比较漫长需要几个小时。熟悉之后,一个功能可能只需要几分钟。