src/main/scala/com/google/gimd/jgit/JGitSnapshot.scala


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

// Copyright (C) 2009 The Android Open Source Project
//
// 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.gimd.jgit

import com.google.gimd.Snapshot
import com.google.gimd.file.{File, FileType}
import org.spearce.jgit.lib.{Repository, FileMode}
import org.spearce.jgit.revwalk.RevCommit
import org.spearce.jgit.treewalk.filter.{PathFilter, PathSuffixFilter, AndTreeFilter, TreeFilter}
import org.spearce.jgit.treewalk.TreeWalk

final class JGitSnapshot(val branch: JGitBranch, val commit: RevCommit) extends Snapshot {

  def all[T](fileType: FileType[T]): Iterator[File[T]] = {
    val treeWalk = new TreeWalk(branch.repository)
    treeWalk.reset(commit.getTree)
    treeWalk.setRecursive(true)
    treeWalk.setFilter(FileTypeTreeFilter(fileType))

    val fileIterator = new Iterator[File[T]] {
      private var doesHasNext = treeWalk.next
      def hasNext = doesHasNext
      def next = {
        if (!hasNext)
          throw new NoSuchElementException
        val result =
          new JGitFile(treeWalk.getPathString, treeWalk.getObjectId(0), fileType, branch)
        doesHasNext = treeWalk.next
        result
      }
    }
    fileIterator
  }

}