mirror of
https://github.com/aljazceru/opencode.git
synced 2025-12-20 17:24:22 +01:00
f993541e0b
This release has a bunch of minor breaking changes if you are using opencode plugins or sdk 1. storage events have been removed (we might bring this back but had some issues) 2. concept of `app` is gone - there is a new concept called `project` and endpoints to list projects and get the current project 3. plugin receives `directory` which is cwd and `worktree` which is where the root of the project is if it's a git repo 4. the session.chat function has been renamed to session.prompt in sdk. it no longer requires model to be passed in (model is now an object) 5. every endpoint takes an optional `directory` parameter to operate as though opencode is running in that directory
127 lines
2.7 KiB
Go
127 lines
2.7 KiB
Go
package completions
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"sort"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/sst/opencode-sdk-go"
|
|
"github.com/sst/opencode/internal/app"
|
|
"github.com/sst/opencode/internal/styles"
|
|
"github.com/sst/opencode/internal/theme"
|
|
)
|
|
|
|
type filesContextGroup struct {
|
|
app *app.App
|
|
gitFiles []CompletionSuggestion
|
|
}
|
|
|
|
func (cg *filesContextGroup) GetId() string {
|
|
return "files"
|
|
}
|
|
|
|
func (cg *filesContextGroup) GetEmptyMessage() string {
|
|
return "no matching files"
|
|
}
|
|
|
|
func (cg *filesContextGroup) getGitFiles() []CompletionSuggestion {
|
|
items := make([]CompletionSuggestion, 0)
|
|
|
|
status, _ := cg.app.Client.File.Status(context.Background(), opencode.FileStatusParams{})
|
|
if status != nil {
|
|
files := *status
|
|
sort.Slice(files, func(i, j int) bool {
|
|
return files[i].Added+files[i].Removed > files[j].Added+files[j].Removed
|
|
})
|
|
|
|
for _, file := range files {
|
|
displayFunc := func(s styles.Style) string {
|
|
t := theme.CurrentTheme()
|
|
green := s.Foreground(t.Success()).Render
|
|
red := s.Foreground(t.Error()).Render
|
|
display := file.Path
|
|
if file.Added > 0 {
|
|
display += green(" +" + strconv.Itoa(int(file.Added)))
|
|
}
|
|
if file.Removed > 0 {
|
|
display += red(" -" + strconv.Itoa(int(file.Removed)))
|
|
}
|
|
return display
|
|
}
|
|
item := CompletionSuggestion{
|
|
Display: displayFunc,
|
|
Value: file.Path,
|
|
ProviderID: cg.GetId(),
|
|
RawData: file,
|
|
}
|
|
items = append(items, item)
|
|
}
|
|
}
|
|
|
|
return items
|
|
}
|
|
|
|
func (cg *filesContextGroup) GetChildEntries(
|
|
query string,
|
|
) ([]CompletionSuggestion, error) {
|
|
items := make([]CompletionSuggestion, 0)
|
|
|
|
query = strings.TrimSpace(query)
|
|
if query == "" {
|
|
items = append(items, cg.gitFiles...)
|
|
}
|
|
|
|
files, err := cg.app.Client.Find.Files(
|
|
context.Background(),
|
|
opencode.FindFilesParams{Query: opencode.F(query)},
|
|
)
|
|
if err != nil {
|
|
slog.Error("Failed to get completion items", "error", err)
|
|
return items, err
|
|
}
|
|
if files == nil {
|
|
return items, nil
|
|
}
|
|
|
|
for _, file := range *files {
|
|
exists := false
|
|
for _, existing := range cg.gitFiles {
|
|
if existing.Value == file {
|
|
if query != "" {
|
|
items = append(items, existing)
|
|
}
|
|
exists = true
|
|
}
|
|
}
|
|
if !exists {
|
|
displayFunc := func(s styles.Style) string {
|
|
// t := theme.CurrentTheme()
|
|
// return s.Foreground(t.Text()).Render(file)
|
|
return s.Render(file)
|
|
}
|
|
|
|
item := CompletionSuggestion{
|
|
Display: displayFunc,
|
|
Value: file,
|
|
ProviderID: cg.GetId(),
|
|
RawData: file,
|
|
}
|
|
items = append(items, item)
|
|
}
|
|
}
|
|
|
|
return items, nil
|
|
}
|
|
|
|
func NewFileContextGroup(app *app.App) CompletionProvider {
|
|
cg := &filesContextGroup{
|
|
app: app,
|
|
}
|
|
go func() {
|
|
cg.gitFiles = cg.getGitFiles()
|
|
}()
|
|
return cg
|
|
}
|