obitools4/pkg/obiseq/eval.go

package obiseq

import (
	"context"
	"fmt"

	log "github.com/sirupsen/logrus"
)

func Expression(expression string) func(*BioSequence) (interface{}, error) {

	exp, err := OBILang.NewEvaluable(expression)
	if err != nil {
		log.Fatalf("Error in the expression : %s", expression)
	}

	f := func(sequence *BioSequence) (interface{}, error) {
		return exp(context.Background(),
			map[string]interface{}{
				"annotations": sequence.Annotations(),
				"sequence":    sequence,
			},
		)
	}

	return f
}

func EditIdWorker(expression string) SeqWorker {
	e := Expression(expression)
	f := func(sequence *BioSequence) (BioSequenceSlice, error) {
		v, err := e(sequence)
		if err == nil {
			sequence.SetId(fmt.Sprintf("%v", v))
		} else {
			err = fmt.Errorf("Expression '%s' cannot be evaluated on sequence %s : %v",
				expression,
				sequence.Id(),
				err)
		}

		return BioSequenceSlice{sequence}, err
	}

	return f
}

func EditAttributeWorker(key string, expression string) SeqWorker {
	e := Expression(expression)
	f := func(sequence *BioSequence) (BioSequenceSlice, error) {
		v, err := e(sequence)
		if err == nil {
			sequence.SetAttribute(key, v)
		} else {
			err = fmt.Errorf("Expression '%s' cannot be evaluated on sequence %s : %v",
				expression,
				sequence.Id(),
				err)
		}

		return BioSequenceSlice{sequence}, err
	}

	return f
}