save2

src/NewPageHeader.tsx

@@ -164,7 +164,7 @@ const NewPage: React.FC = props => {
                 onClick={() => setIsShowMenu(!isShowMenu)}
               />
             ) : (
-              <Link to={`/login2p?next=${location.pathname}`}>
+              <Link to={`/login2p?next=${location.pathname + location.search}`}>
                 <Button type="primary" className="mobile-login2p-btn">
                   登录
                 </Button>
@@ -227,7 +227,7 @@ const NewPage: React.FC = props => {
               ) : null}
               {/* (
               <div className="mobile-header-item" onClick={login2p.logout}>
-                <Link to={`/login2p?next=${location.pathname}`}>
+                <Link to={`/login2p?next=${location.pathname+ location.search}`}>
                   <Button type="primary" className="mobile-login2p-btn">
                     登录
                   </Button>
@@ -351,7 +351,7 @@ const NewPage: React.FC = props => {
                 </div>
               </div>
             ) : (
-              <Link to={`/login2p?next=${location.pathname}`}>
+              <Link to={`/login2p?next=${location.pathname + location.search}`}>
                 <Button type="primary" className={styles['login2p-btn']}>
                   登录
                 </Button>

src/Vmp.tsx

@@ -29,7 +29,7 @@ const Vmp: React.FC = () => {
       sessionStorage.setItem('invite_code', inviteCode)
     }
     // 如果有邀请码，带上邀请码
-    if (sessionStorage.getItem('invite_code')) {
+    if (sessionStorage.getItem('invite_code') && !window.location.search.includes('invite_code')) {
       history.push(`${window.location.pathname}?invite_code=${sessionStorage.getItem('invite_code')}`)
     }
     if (login2p.jwt) {