优化

src/utils/auth/index.ts

@@ -1,5 +1,8 @@
 const TokenKey = 'accessToken';
 const CompanyIdKey = 'companyId';
+const ReferrerUserIdKey = 'referrerUserId';
+const RegisterStoreIdKey = 'storeId';
+
 const TokenPrefix = 'Bearer ';
 
 function isLogin() {
@@ -22,9 +25,37 @@ function setCompanyId(companyId: string) {
   uni.setStorageSync(CompanyIdKey, companyId);
 }
 
+function getReferrerUserId() {
+  return uni.getStorageSync(ReferrerUserIdKey);
+}
+
+function setReferrerUserId(referrerUserId: string) {
+  uni.setStorageSync(ReferrerUserIdKey, referrerUserId);
+}
+
+function setRegisterStoreId(storeId: string) {
+  uni.setStorageSync(RegisterStoreIdKey, storeId);
+}
+
+function getRegisterStoreId() {
+  return uni.getStorageSync(RegisterStoreIdKey);
+}
+
 function clearToken() {
   uni.removeStorageSync(TokenKey);
   uni.removeStorageSync(CompanyIdKey);
 }
 
-export { TokenPrefix, isLogin, getToken, setToken, getCompanyId, setCompanyId, clearToken };
+export {
+  TokenPrefix,
+  isLogin,
+  getToken,
+  setToken,
+  getCompanyId,
+  setCompanyId,
+  getReferrerUserId,
+  setReferrerUserId,
+  getRegisterStoreId,
+  setRegisterStoreId,
+  clearToken
+};