Skip to content

Commit

Permalink
Add .txt file output format
Browse files Browse the repository at this point in the history
  • Loading branch information
pcj committed Sep 26, 2024
1 parent db1d55f commit 04c0e6f
Showing 1 changed file with 24 additions and 3 deletions.
27 changes: 24 additions & 3 deletions language/scala/imports.go
Original file line number Diff line number Diff line change
@@ -1,8 +1,12 @@
package scala

import (
"fmt"
"log"
"os"
"path/filepath"

"github.com/bazelbuild/bazel-gazelle/label"
scpb "github.com/stackb/scala-gazelle/build/stack/gazelle/scala/cache"
"github.com/stackb/scala-gazelle/pkg/protobuf"
)
Expand All @@ -13,11 +17,28 @@ func (sl *scalaLang) writeResolvedImportsMapFile(filename string) error {
}

for _, sym := range sl.globalScope.GetSymbols("") {
imports.Imports[sym.Name] = sym.Label.String()
dep := "NO_LABEL"
if sym.Label != label.NoLabel {
dep = sym.Label.String()
}
imports.Imports[sym.Name] = dep
}

if err := protobuf.WriteFile(filename, imports); err != nil {
return err
if filepath.Ext(filename) == ".txt" {
f, err := os.Create(filename)
if err != nil {
return fmt.Errorf("create: %w", err)
}
for k, v := range imports.Imports {
fmt.Fprintln(f, k, v)
}
if err := f.Close(); err != nil {
return fmt.Errorf("close: %w", err)
}
} else {
if err := protobuf.WriteFile(filename, imports); err != nil {
return err
}
}

log.Printf("Wrote scala-gazelle import map %s", filename)
Expand Down

0 comments on commit 04c0e6f

Please sign in to comment.