package obitax

import (
	"git.metabarcoding.org/lecasofts/go/obitools/pkg/obiseq"
	"git.metabarcoding.org/lecasofts/go/obitools/pkg/obiutils"
	log "github.com/sirupsen/logrus"
)

func (taxonomy *Taxonomy) MakeSetTaxonAtRankWorker(rank string) obiseq.SeqWorker {

	if !obiutils.Contains(taxonomy.RankList(), rank) {
		log.Fatalf("%s is not a valid rank (allowed ranks are %v)",
			rank,
			taxonomy.RankList())
	}

	w := func(sequence *obiseq.BioSequence) *obiseq.BioSequence {
		taxonomy.SetTaxonAtRank(sequence, rank)
		return sequence
	}

	return w
}

func (taxonomy *Taxonomy) MakeSetSpeciesWorker() obiseq.SeqWorker {

	w := func(sequence *obiseq.BioSequence) *obiseq.BioSequence {
		taxonomy.SetSpecies(sequence)
		return sequence
	}

	return w
}

func (taxonomy *Taxonomy) MakeSetGenusWorker() obiseq.SeqWorker {

	w := func(sequence *obiseq.BioSequence) *obiseq.BioSequence {
		taxonomy.SetGenus(sequence)
		return sequence
	}

	return w
}

func (taxonomy *Taxonomy) MakeSetFamilyWorker() obiseq.SeqWorker {

	w := func(sequence *obiseq.BioSequence) *obiseq.BioSequence {
		taxonomy.SetFamily(sequence)
		return sequence
	}

	return w
}