112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
|
// Copyright 2019 The UNICORE Authors. All rights reserved.
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file
|
||
|
|
//
|
||
|
|
// Author: Gaulthier Gain <gaulthier.gain@uliege.be>
|
||
|
|
|
||
|
|
package crawlertool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"path/filepath"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
u "tools/srcs/common"
|
||
|
|
)
|
||
|
|
|
||
|
|
// RunCrawler allows to run the crawler analyser tool (which is out of the
|
||
|
|
// UNICORE toolchain).
|
||
|
|
func RunCrawler() {
|
||
|
|
|
||
|
|
mapLabel := make(map[string]string)
|
||
|
|
mapConfig := make(map[string][]string)
|
||
|
|
|
||
|
|
// Init and parse local arguments
|
||
|
|
args := new(u.Arguments)
|
||
|
|
p, err := args.InitArguments()
|
||
|
|
if err != nil {
|
||
|
|
u.PrintErr(err)
|
||
|
|
}
|
||
|
|
if err := parseLocalArguments(p, args); err != nil {
|
||
|
|
u.PrintErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Used to select all libraries (even those below another Config fields)
|
||
|
|
fullSelect := *args.BoolArg[fullLibsArg]
|
||
|
|
|
||
|
|
var path string
|
||
|
|
if len(*args.StringArg[repoArg]) > 0 {
|
||
|
|
// Only one folder
|
||
|
|
path = *args.StringArg[repoArg]
|
||
|
|
u.PrintInfo("Parse folder: " + path)
|
||
|
|
if err := searchConfigUK(path, fullSelect, mapConfig, mapLabel); err != nil {
|
||
|
|
u.PrintErr()
|
||
|
|
}
|
||
|
|
|
||
|
|
} else if len(*args.StringArg[libsArg]) > 0 {
|
||
|
|
|
||
|
|
// Several folders within a list
|
||
|
|
lines, err := u.ReadLinesFile(*args.StringArg[libsArg])
|
||
|
|
if err != nil {
|
||
|
|
u.PrintErr(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process Config.uk of each process
|
||
|
|
for _, line := range lines {
|
||
|
|
path = strings.TrimSuffix(line, "\n")
|
||
|
|
u.PrintInfo("Parse folder: " + path)
|
||
|
|
if err := searchConfigUK(path, fullSelect, mapConfig, mapLabel); err != nil {
|
||
|
|
u.PrintErr(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
u.PrintErr("You must specify either -r (--repository) or -l (libs)")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Generate the out folder
|
||
|
|
outFolder := *args.StringArg[outputArg]
|
||
|
|
if outFolder[len(outFolder)-1:] != string(os.PathSeparator) {
|
||
|
|
outFolder += string(os.PathSeparator)
|
||
|
|
}
|
||
|
|
|
||
|
|
outputPath := outFolder +
|
||
|
|
"output_" + time.Now().Format("20060102150405")
|
||
|
|
|
||
|
|
// Create the dependencies graph
|
||
|
|
u.GenerateGraph("Unikraft Crawler", outputPath, mapConfig,
|
||
|
|
mapLabel)
|
||
|
|
|
||
|
|
u.PrintOk(".dot file is saved: " + outputPath)
|
||
|
|
}
|
||
|
|
|
||
|
|
// searchConfigUK performs a look-up to find "Config.uk" files.
|
||
|
|
//
|
||
|
|
// It returns an error if any, otherwise it returns nil.
|
||
|
|
func searchConfigUK(path string, fullSelect bool,
|
||
|
|
mapConfig map[string][]string, mapLabel map[string]string) error {
|
||
|
|
|
||
|
|
err := filepath.Walk(path, func(path string, info os.FileInfo,
|
||
|
|
err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Consider only CONFIGUK files
|
||
|
|
if !info.IsDir() && info.Name() == u.CONFIGUK {
|
||
|
|
lines, err := u.ReadLinesFile(path)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
u.ProcessConfigUK(lines, fullSelect, mapConfig, mapLabel)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|