Implemented a basic scanner
All checks were successful
/ test (push) Successful in 17s

This commit is contained in:
Eduard Urbach 2025-06-19 15:17:50 +02:00
parent df92f55df6
commit 154893d9f7
Signed by: akyoto
GPG key ID: 49226B848C78F6C8
19 changed files with 410 additions and 11 deletions

View file

@ -1,4 +0,0 @@
package scanner
// Result contains everything the compiler needs to start a build.
type Result struct{}

View file

@ -2,9 +2,23 @@ package scanner
import (
"git.urbach.dev/cli/q/src/build"
"git.urbach.dev/cli/q/src/fs"
)
// Scan scans all the files included in the build.
func Scan(b *build.Build) Result {
return Result{}
func Scan(b *build.Build) (<-chan *fs.File, <-chan error) {
s := scanner{
files: make(chan *fs.File),
errors: make(chan error),
build: b,
}
go func() {
s.queue(b.Files...)
s.group.Wait()
close(s.files)
close(s.errors)
}()
return s.files, s.errors
}

View file

@ -4,10 +4,54 @@ import (
"testing"
"git.urbach.dev/cli/q/src/build"
"git.urbach.dev/cli/q/src/fs"
"git.urbach.dev/cli/q/src/scanner"
"git.urbach.dev/go/assert"
)
func TestScan(t *testing.T) {
b := build.New("../../examples/hello")
scanner.Scan(b)
func TestScanDirectory(t *testing.T) {
b := build.New("testdata")
files, errors := scanner.Scan(b)
err := consume(t, files, errors)
assert.Nil(t, err)
}
func TestScanFile(t *testing.T) {
b := build.New("testdata/file.q")
files, errors := scanner.Scan(b)
err := consume(t, files, errors)
assert.Nil(t, err)
}
func TestScanNotExisting(t *testing.T) {
b := build.New("_")
files, errors := scanner.Scan(b)
err := consume(t, files, errors)
assert.NotNil(t, err)
}
func consume(t *testing.T, files <-chan *fs.File, errors <-chan error) error {
var lastError error
for files != nil || errors != nil {
select {
case file, ok := <-files:
if !ok {
files = nil
continue
}
t.Log(file)
case err, ok := <-errors:
if !ok {
errors = nil
continue
}
lastError = err
}
}
return lastError
}

147
src/scanner/scanner.go Normal file
View file

@ -0,0 +1,147 @@
package scanner
import (
"os"
"path/filepath"
"strings"
"sync"
"git.urbach.dev/cli/q/src/build"
"git.urbach.dev/cli/q/src/fs"
)
// scanner is used to scan files before the actual compilation step.
type scanner struct {
files chan *fs.File
errors chan error
build *build.Build
queued sync.Map
group sync.WaitGroup
}
// queue scans the list of files.
func (s *scanner) queue(files ...string) {
for _, file := range files {
stat, err := os.Stat(file)
if err != nil {
s.errors <- err
return
}
if stat.IsDir() {
s.queueDirectory(file, "main")
} else {
s.queueFile(file, "main")
}
}
}
// queueDirectory queues an entire directory to be scanned.
func (s *scanner) queueDirectory(directory string, pkg string) {
_, loaded := s.queued.LoadOrStore(directory, nil)
if loaded {
return
}
err := fs.Walk(directory, func(name string) {
if !strings.HasSuffix(name, ".q") {
return
}
tmp := name[:len(name)-2]
for {
underscore := strings.LastIndexByte(tmp, '_')
if underscore == -1 {
break
}
condition := tmp[underscore+1:]
switch condition {
case "linux":
if s.build.OS != build.Linux {
return
}
case "mac":
if s.build.OS != build.Mac {
return
}
case "unix":
if s.build.OS != build.Linux && s.build.OS != build.Mac {
return
}
case "windows":
if s.build.OS != build.Windows {
return
}
case "x86":
if s.build.Arch != build.X86 {
return
}
case "arm":
if s.build.Arch != build.ARM {
return
}
default:
return
}
tmp = tmp[:underscore]
}
fullPath := filepath.Join(directory, name)
s.queueFile(fullPath, pkg)
})
if err != nil {
s.errors <- err
}
}
// queueFile queues a single file to be scanned.
func (s *scanner) queueFile(file string, pkg string) {
_, loaded := s.queued.LoadOrStore(file, nil)
if loaded {
return
}
s.group.Add(1)
go func() {
defer s.group.Done()
err := s.scanFile(file, pkg)
if err != nil {
s.errors <- err
}
}()
}
// scanFile scans a single file.
func (s *scanner) scanFile(path string, pkg string) error {
contents, err := os.ReadFile(path)
if err != nil {
return err
}
file := &fs.File{
Path: path,
Package: pkg,
Bytes: contents,
}
s.files <- file
return nil
}

0
src/scanner/testdata/file.q vendored Normal file
View file

0
src/scanner/testdata/file_arm.q vendored Normal file
View file

0
src/scanner/testdata/file_custom.q vendored Normal file
View file

0
src/scanner/testdata/file_linux.q vendored Normal file
View file

0
src/scanner/testdata/file_mac.q vendored Normal file
View file

0
src/scanner/testdata/file_unix.q vendored Normal file
View file

0
src/scanner/testdata/file_windows.q vendored Normal file
View file

0
src/scanner/testdata/file_x86.q vendored Normal file
View file