1 files changed, 132 insertions, 0 deletions
diff --git a/mgrapp/src/com/google/codereview/manager/ProjectSync.java b/mgrapp/src/com/google/codereview/manager/ProjectSync.java
new file mode 100644
index 0000000000..b3d2d9eb2a
--- /dev/null
+++ b/mgrapp/src/com/google/codereview/manager/ProjectSync.java
@@ -0,0 +1,132 @@
+// Copyright 2008 Google Inc.
+//
+// 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.
+
+package com.google.codereview.manager;
+
+import com.google.codereview.internal.SyncProject.BranchSync;
+import com.google.codereview.internal.SyncProject.SyncProjectRequest;
+import com.google.codereview.internal.SyncProject.SyncProjectResponse;
+import com.google.codereview.rpc.SimpleController;
+import com.google.codereview.util.GitMetaUtil;
+import com.google.protobuf.RpcCallback;
+
+import org.apache.commons.logging.Log;
+import org.apache.commons.logging.LogFactory;
+import org.spearce.jgit.errors.IncorrectObjectTypeException;
+import org.spearce.jgit.lib.Constants;
+import org.spearce.jgit.lib.Ref;
+import org.spearce.jgit.lib.Repository;
+import org.spearce.jgit.revwalk.RevCommit;
+import org.spearce.jgit.revwalk.RevWalk;
+
+import java.io.File;
+import java.io.IOException;
+
+public class ProjectSync {
+  private static final Log LOG = LogFactory.getLog(ProjectSync.class);
+
+  private final Backend server;
+
+  public ProjectSync(final Backend be) {
+    server = be;
+  }
+
+  public void sync() {
+    syncDirectoryImpl(server.getRepositoryCache().getBaseDirectory(), "");
+  }
+
+  private void syncDirectoryImpl(final File root, final String prefix) {
+    final File[] entries = root.listFiles();
+    if (entries == null) {
+      return;
+    }
+
+    for (final File f : entries) {
+      final String fName = f.getName();
+      if (fName.equals(".") || fName.equals("..")) {
+        continue;
+      }
+
+      if (!f.isDirectory()) {
+        continue;
+      }
+
+      String fullName = prefix + fName;
+      final Repository db;
+      try {
+        db = GitMetaUtil.open(f);
+      } catch (IOException e) {
+        LOG.error("Cannot open " + f + ": " + e.toString());
+        continue;
+      }
+
+      if (db != null) {
+        try {
+          if (fullName.endsWith(".git")) {
+            fullName = fullName.substring(0, fullName.length() - 4);
+          }
+          sync(fullName, db);
+        } finally {
+          db.close();
+        }
+      } else {
+        syncDirectoryImpl(f, fullName + "/");
+      }
+    }
+  }
+
+  public void sync(final String name, final Repository db) {
+    final SyncProjectRequest.Builder req = SyncProjectRequest.newBuilder();
+    req.setProjectName(name);
+
+    try {
+      final RevWalk rw = new RevWalk(db);
+      for (final Ref ref : db.getAllRefs().values()) {
+        if (ref.getName().startsWith(Constants.R_HEADS)) {
+          final RevCommit c;
+          try {
+            c = rw.parseCommit(ref.getObjectId());
+          } catch (IncorrectObjectTypeException e) {
+            continue;
+          }
+          req.addBranch(toBranch(ref, c));
+        }
+      }
+    } catch (IOException err) {
+      LOG.error("Cannot synchronize " + name, err);
+    }
+
+    send(req.build());
+  }
+
+  private BranchSync toBranch(final Ref r, final RevCommit c) {
+    final BranchSync.Builder bs = BranchSync.newBuilder();
+    bs.setBranchName(r.getName());
+    bs.setCommit(GitMetaUtil.toGitCommit(c));
+    return bs.build();
+  }
+
+  private void send(final SyncProjectRequest req) {
+    final SimpleController ctrl = new SimpleController();
+    server.getAdminService().syncProject(ctrl, req,
+        new RpcCallback<SyncProjectResponse>() {
+          public void run(final SyncProjectResponse rsp) {
+          }
+        });
+    if (ctrl.failed()) {
+      final String name = req.getProjectName();
+      LOG.error("syncProject " + name + ": " + ctrl.errorText());
+    }
+  }
+}