wehub-resource-sync
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package query
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/kashav/fsql/transform"
|
|
)
|
|
|
|
// Modifier represents an attribute modifier.
|
|
type Modifier struct {
|
|
Name string
|
|
Arguments []string
|
|
}
|
|
|
|
func (m *Modifier) String() string {
|
|
return fmt.Sprintf("%s(%s)", m.Name, strings.Join(m.Arguments, ", "))
|
|
}
|
|
|
|
// applyModifiers iterates through each SELECT attribute for this query
|
|
// and applies the associated modifier to the attribute's output value.
|
|
func (q *Query) applyModifiers(path string, info os.FileInfo) (map[string]interface{}, error) {
|
|
results := make(map[string]interface{}, len(q.Attributes))
|
|
|
|
for _, attribute := range q.Attributes {
|
|
value, err := transform.DefaultFormatValue(attribute, path, info)
|
|
if err != nil {
|
|
return map[string]interface{}{}, err
|
|
}
|
|
|
|
if _, ok := q.Modifiers[attribute]; !ok {
|
|
results[attribute] = value
|
|
continue
|
|
}
|
|
|
|
for _, m := range q.Modifiers[attribute] {
|
|
value, err = transform.Format(&transform.FormatParams{
|
|
Attribute: attribute,
|
|
Path: path,
|
|
Info: info,
|
|
Value: value,
|
|
Name: m.Name,
|
|
Args: m.Arguments,
|
|
})
|
|
if err != nil {
|
|
return map[string]interface{}{}, err
|
|
}
|
|
}
|
|
|
|
results[attribute] = value
|
|
}
|
|
|
|
return results, nil
|
|
}
|