core.go 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163
  1. package hercules
  2. import (
  3. git "gopkg.in/src-d/go-git.v4"
  4. "gopkg.in/src-d/go-git.v4/plumbing/object"
  5. "gopkg.in/src-d/hercules.v4/internal/core"
  6. "gopkg.in/src-d/hercules.v4/internal/plumbing"
  7. "gopkg.in/src-d/hercules.v4/internal/plumbing/identity"
  8. "gopkg.in/src-d/hercules.v4/internal/plumbing/uast"
  9. "gopkg.in/src-d/hercules.v4/leaves"
  10. "gopkg.in/src-d/hercules.v4/internal/yaml"
  11. )
  12. // ConfigurationOptionType represents the possible types of a ConfigurationOption's value.
  13. type ConfigurationOptionType = core.ConfigurationOptionType
  14. const (
  15. // BoolConfigurationOption reflects the boolean value type.
  16. BoolConfigurationOption = core.BoolConfigurationOption
  17. // IntConfigurationOption reflects the integer value type.
  18. IntConfigurationOption = core.IntConfigurationOption
  19. // StringConfigurationOption reflects the string value type.
  20. StringConfigurationOption = core.StringConfigurationOption
  21. // FloatConfigurationOption reflects a floating point value type.
  22. FloatConfigurationOption = core.FloatConfigurationOption
  23. // StringsConfigurationOption reflects the array of strings value type.
  24. StringsConfigurationOption = core.StringsConfigurationOption
  25. )
  26. // ConfigurationOption allows for the unified, retrospective way to setup PipelineItem-s.
  27. type ConfigurationOption = core.ConfigurationOption
  28. // PipelineItem is the interface for all the units in the Git commits analysis pipeline.
  29. type PipelineItem = core.PipelineItem
  30. // FeaturedPipelineItem enables switching the automatic insertion of pipeline items on or off.
  31. type FeaturedPipelineItem = core.FeaturedPipelineItem
  32. // LeafPipelineItem corresponds to the top level pipeline items which produce the end results.
  33. type LeafPipelineItem = core.LeafPipelineItem
  34. // ResultMergeablePipelineItem specifies the methods to combine several analysis results together.
  35. type ResultMergeablePipelineItem = core.ResultMergeablePipelineItem
  36. // CommonAnalysisResult holds the information which is always extracted at Pipeline.Run().
  37. type CommonAnalysisResult = core.CommonAnalysisResult
  38. // NoopMerger provides an empty Merge() method suitable for PipelineItem.
  39. type NoopMerger = core.NoopMerger
  40. // OneShotMergeProcessor provides the convenience method to consume merges only once.
  41. type OneShotMergeProcessor = core.OneShotMergeProcessor
  42. // MetadataToCommonAnalysisResult copies the data from a Protobuf message.
  43. func MetadataToCommonAnalysisResult(meta *core.Metadata) *CommonAnalysisResult {
  44. return core.MetadataToCommonAnalysisResult(meta)
  45. }
  46. // Pipeline is the core Hercules entity which carries several PipelineItems and executes them.
  47. // See the extended example of how a Pipeline works in doc.go
  48. type Pipeline = core.Pipeline
  49. const (
  50. // ConfigPipelineDumpPath is the name of the Pipeline configuration option (Pipeline.Initialize())
  51. // which enables saving the items DAG to the specified file.
  52. ConfigPipelineDumpPath = core.ConfigPipelineDumpPath
  53. // ConfigPipelineDryRun is the name of the Pipeline configuration option (Pipeline.Initialize())
  54. // which disables Configure() and Initialize() invocation on each PipelineItem during the
  55. // Pipeline initialization.
  56. // Subsequent Run() calls are going to fail. Useful with ConfigPipelineDumpPath=true.
  57. ConfigPipelineDryRun = core.ConfigPipelineDryRun
  58. // ConfigPipelineCommits is the name of the Pipeline configuration option (Pipeline.Initialize())
  59. // which allows to specify the custom commit sequence. By default, Pipeline.Commits() is used.
  60. ConfigPipelineCommits = core.ConfigPipelineCommits
  61. )
  62. // NewPipeline initializes a new instance of Pipeline struct.
  63. func NewPipeline(repository *git.Repository) *Pipeline {
  64. return core.NewPipeline(repository)
  65. }
  66. // LoadCommitsFromFile reads the file by the specified FS path and generates the sequence of commits
  67. // by interpreting each line as a Git commit hash.
  68. func LoadCommitsFromFile(path string, repository *git.Repository) ([]*object.Commit, error) {
  69. return core.LoadCommitsFromFile(path, repository)
  70. }
  71. // ForkSamePipelineItem clones items by referencing the same origin.
  72. func ForkSamePipelineItem(origin PipelineItem, n int) []PipelineItem {
  73. return core.ForkSamePipelineItem(origin ,n)
  74. }
  75. // ForkCopyPipelineItem clones items by copying them by value from the origin.
  76. func ForkCopyPipelineItem(origin PipelineItem, n int) []PipelineItem {
  77. return core.ForkCopyPipelineItem(origin ,n)
  78. }
  79. // PipelineItemRegistry contains all the known PipelineItem-s.
  80. type PipelineItemRegistry = core.PipelineItemRegistry
  81. // Registry contains all known pipeline item types.
  82. var Registry = core.Registry
  83. const (
  84. // DependencyCommit is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  85. // which always exists. It corresponds to the currently analyzed commit.
  86. DependencyCommit = core.DependencyCommit
  87. // DependencyIndex is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  88. // which always exists. It corresponds to the currently analyzed commit's index.
  89. DependencyIndex = core.DependencyIndex
  90. // DependencyIsMerge is the name of one of the three items in `deps` supplied to PipelineItem.Consume()
  91. // which always exists. It indicates whether the analyzed commit is a merge commit.
  92. // Checking the number of parents is not correct - we remove the back edges during the DAG simplification.
  93. DependencyIsMerge = core.DependencyIsMerge
  94. // DependencyAuthor is the name of the dependency provided by identity.Detector.
  95. DependencyAuthor = identity.DependencyAuthor
  96. // DependencyBlobCache identifies the dependency provided by BlobCache.
  97. DependencyBlobCache = plumbing.DependencyBlobCache
  98. // DependencyDay is the name of the dependency which DaysSinceStart provides - the number
  99. // of days since the first commit in the analysed sequence.
  100. DependencyDay = plumbing.DependencyDay
  101. // DependencyFileDiff is the name of the dependency provided by FileDiff.
  102. DependencyFileDiff = plumbing.DependencyFileDiff
  103. // DependencyTreeChanges is the name of the dependency provided by TreeDiff.
  104. DependencyTreeChanges = plumbing.DependencyTreeChanges
  105. // DependencyUastChanges is the name of the dependency provided by Changes.
  106. DependencyUastChanges = uast.DependencyUastChanges
  107. // DependencyUasts is the name of the dependency provided by Extractor.
  108. DependencyUasts = uast.DependencyUasts
  109. // FactCommitsByDay contains the mapping between day indices and the corresponding commits.
  110. FactCommitsByDay = plumbing.FactCommitsByDay
  111. // FactIdentityDetectorPeopleCount is the name of the fact which is inserted in
  112. // identity.Detector.Configure(). It is equal to the overall number of unique authors
  113. // (the length of ReversedPeopleDict).
  114. FactIdentityDetectorPeopleCount = identity.FactIdentityDetectorPeopleCount
  115. // FactIdentityDetectorPeopleDict is the name of the fact which is inserted in
  116. // identity.Detector.Configure(). It corresponds to identity.Detector.PeopleDict - the mapping
  117. // from the signatures to the author indices.
  118. FactIdentityDetectorPeopleDict = identity.FactIdentityDetectorPeopleDict
  119. // FactIdentityDetectorReversedPeopleDict is the name of the fact which is inserted in
  120. // identity.Detector.Configure(). It corresponds to identity.Detector.ReversedPeopleDict -
  121. // the mapping from the author indices to the main signature.
  122. FactIdentityDetectorReversedPeopleDict = identity.FactIdentityDetectorReversedPeopleDict
  123. )
  124. // FileDiffData is the type of the dependency provided by plumbing.FileDiff.
  125. type FileDiffData = plumbing.FileDiffData
  126. // CountLines returns the number of lines in a *object.Blob.
  127. func CountLines(file *object.Blob) (int, error) {
  128. return plumbing.CountLines(file)
  129. }
  130. // SafeYamlString escapes the string so that it can be reliably used in YAML.
  131. func SafeYamlString(str string) string {
  132. return yaml.SafeString(str)
  133. }
  134. func init() {
  135. // hack to link with .leaves
  136. _ = leaves.BurndownAnalysis{}
  137. }