feat: 支持强制 reinit head

Author: Ricbet

packages/code-api/src/atomgit/atomgit.service.ts

@@ -89,10 +89,9 @@ export class AtomGitAPIService implements ICodeAPIService {
             });
             if (tokenResult && tokenResult.access_token) {
               this.helper.ATOMGIT_TOKEN = tokenResult.access_token;
-              resolve(true);
-              return;
+              this.helper.reinitializeCodeService(true);
             }
-            resolve(false);
+            resolve(true);
           }
         } catch (error) {
           reject(error);

packages/code-api/src/common/service.ts

@@ -104,8 +104,8 @@ export class HelperService {
     this.commandService.executeCommand(`workbench.view.${id}`);
   }
 
-  reinitializeCodeService() {
-    this.commandService.executeCommand('code-service.reinitialize');
+  reinitializeCodeService(isForce: boolean = false) {
+    this.commandService.executeCommand('code-service.reinitialize', isForce);
   }
 
   showMessage(

packages/code-service/src/code-model.service.ts

@@ -89,12 +89,13 @@ export class CodeModelService {
   /**
    * 无权限时需要再设置 token 后重新初始化
    */
-  async reinitialize() {
+  async reinitialize(isForce?: boolean) {
     const { codeServiceConfig: config } = this;
     await this.rootRepository.initHEAD({
       commit: config.commit,
       ref: config.branch || config.tag || config.ref,
       refPath: config.refPath,
+      isForce
     });
   }
 

packages/code-service/src/code-service.contribution.ts

@@ -327,8 +327,8 @@ export class CodeContribution
       }),
 
       commandRegistry.registerCommand(CODE_SERVICE_COMMANDS.REINITIALIZE, {
-        execute: () => {
-          this.codeModel.reinitialize();
+        execute: (isForce?: boolean) => {
+          this.codeModel.reinitialize(isForce);
         },
       }),
       commandRegistry.registerCommand(CODE_SERVICE_COMMANDS.CREATEBRANCH, {

packages/code-service/src/repository.ts

@@ -145,7 +145,7 @@ export class RootRepository extends Repository {
   private _onDidChangeRef = new Emitter<void>();
   readonly onDidChangeRef = this._onDidChangeRef.event;
 
-  private _initialized = false;
+  public _initialized = false;
 
   private _refs: Refs = { branches: [], tags: [] };
   get refs() {
@@ -194,8 +194,8 @@ export class RootRepository extends Repository {
    * @param ref tag, branch
    * @param refPath [tree|blob]/branch/path
    */
-  async initHEAD({ commit, ref, refPath }: { commit?: string; ref?: string; refPath?: string }) {
-    if (this._initialized) {
+  async initHEAD({ commit, ref, refPath, isForce }: { commit?: string; ref?: string; refPath?: string, isForce?: boolean }) {
+    if (this._initialized && !isForce) {
       return;
     }
     this._initialized = true;