Skip to content

Commit e9e2c78

Browse files
committed
summarise
1 parent d76db70 commit e9e2c78

File tree

3 files changed

+25
-5
lines changed

3 files changed

+25
-5
lines changed

backend/controllers/github.go

+5-3
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ import (
3434
"net/http"
3535
"net/url"
3636
"os"
37-
"path"
3837
"path/filepath"
3938
"reflect"
4039
"runtime/debug"
@@ -596,8 +595,11 @@ func GetDiggerConfigForBranch(gh utils.GithubClientProvider, installationId int6
596595
var dependencyGraph graph.Graph[string, dg_configuration.Project]
597596

598597
err = utils.CloneGitRepoAndDoAction(cloneUrl, branch, "", *token, func(dir string) error {
599-
diggerYmlBytes, err := os.ReadFile(path.Join(dir, "digger.yml"))
600-
diggerYmlStr = string(diggerYmlBytes)
598+
diggerYmlStr, err = dg_configuration.ReadDiggerYmlFileContents(dir)
599+
if err != nil {
600+
log.Printf("could not load digger config: %v", err)
601+
return err
602+
}
601603
config, _, dependencyGraph, err = dg_configuration.LoadDiggerConfig(dir, true, changedFiles)
602604
if err != nil {
603605
log.Printf("Error loading digger config: %v", err)

backend/controllers/projects.go

+9-2
Original file line numberDiff line numberDiff line change
@@ -485,7 +485,6 @@ func (d DiggerController) SetJobStatusForProject(c *gin.Context) {
485485
err = CreateTerraformOutputsSummary(d.GithubClientProvider, batch)
486486
if err != nil {
487487
log.Printf("could not generate terraform plans summary: %v", err)
488-
c.JSON(http.StatusInternalServerError, gin.H{"error": "could not generate terraform plans summary"})
489488
}
490489

491490
err = AutomergePRforBatchIfEnabled(d.GithubClientProvider, batch)
@@ -660,10 +659,13 @@ func CreateTerraformOutputsSummary(gh utils.GithubClientProvider, batch *models.
660659
log.Printf("Error loading digger config from batch: %v", err)
661660
return fmt.Errorf("error loading digger config from batch: %v", err)
662661
}
663-
662+
663+
log.Printf("creating AI summary now")
664+
664665
config, _, err := digger_config.ConvertDiggerYamlToConfig(diggerConfigYml)
665666

666667
if batch.Status == orchestrator_scheduler.BatchJobSucceeded && config.Reporting.AiSummary == true {
668+
log.Printf("initiating ai summary creation")
667669
prService, err := GetPrServiceFromBatch(batch, gh)
668670
if err != nil {
669671
log.Printf("Error getting github service: %v", err)
@@ -678,6 +680,11 @@ func CreateTerraformOutputsSummary(gh utils.GithubClientProvider, batch *models.
678680
apiToken := os.Getenv("DIGGER_AI_SUMMARY_API_TOKEN")
679681

680682
jobs, err := models.DB.GetDiggerJobsForBatch(batch.ID)
683+
if err != nil {
684+
log.Printf("could not get jobs for batch: %v", err)
685+
return fmt.Errorf("could not get jobs for batch: %v", err)
686+
}
687+
681688
var terraformOutputs = ""
682689
for _, job := range jobs {
683690
var jobSpec orchestrator_scheduler.JobJson

libs/digger_config/digger_config.go

+11
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,17 @@ type FileSystemTerragruntDirWalker struct {
3030
type FileSystemModuleDirWalker struct {
3131
}
3232

33+
func ReadDiggerYmlFileContents(dir string) (string, error) {
34+
var diggerYmlBytes []byte
35+
diggerYmlBytes, err := os.ReadFile(path.Join(dir, "digger.yml"))
36+
if err != nil {
37+
// if file doesn't exist look for digger.yaml instead
38+
diggerYmlBytes, err = os.ReadFile(path.Join(dir, "digger.yaml"))
39+
}
40+
diggerYmlStr := string(diggerYmlBytes)
41+
return diggerYmlStr, nil
42+
}
43+
3344
func CheckOrCreateDiggerFile(dir string) error {
3445
// Check for digger.yml
3546
ymlPath := filepath.Join(dir, "digger.yml")

0 commit comments

Comments
 (0)