package main

import (
	"database/sql"
	"encoding/json"
	"errors"
	"fmt"
	"net/http"
	"net/url"
	"strings"

	"github.com/thermokarst/bactdb/Godeps/_workspace/src/github.com/jmoiron/modl"
)

var (
	ErrStrainNotFound   = errors.New("Strain not found")
	ErrStrainNotUpdated = errors.New("Strain not updated")
)

func init() {
	DB.AddTableWithName(StrainBase{}, "strains").SetKeys(true, "Id")
}

func (s *StrainBase) PreInsert(e modl.SqlExecutor) error {
	ct := currentTime()
	s.CreatedAt = ct
	s.UpdatedAt = ct
	return nil
}

func (s *StrainBase) PreUpdate(e modl.SqlExecutor) error {
	s.UpdatedAt = currentTime()
	return nil
}

type StrainService struct{}

type StrainBase struct {
	Id                  int64      `db:"id" json:"id"`
	SpeciesId           int64      `db:"species_id" json:"species"`
	StrainName          string     `db:"strain_name" json:"strainName"`
	TypeStrain          bool       `db:"type_strain" json:"typeStrain"`
	AccessionNumbers    NullString `db:"accession_numbers" json:"accessionNumbers"`
	Genbank             NullString `db:"genbank" json:"genbank"`
	WholeGenomeSequence NullString `db:"whole_genome_sequence" json:"wholeGenomeSequence"`
	IsolatedFrom        NullString `db:"isolated_from" json:"isolatedFrom"`
	Notes               NullString `db:"notes" json:"notes"`
	CreatedAt           NullTime   `db:"created_at" json:"createdAt"`
	UpdatedAt           NullTime   `db:"updated_at" json:"updatedAt"`
	DeletedAt           NullTime   `db:"deleted_at" json:"deletedAt"`
	CreatedBy           int64      `db:"created_by" json:"createdBy"`
	UpdatedBy           int64      `db:"updated_by" json:"updatedBy"`
	DeletedBy           NullInt64  `db:"deleted_by" json:"deletedBy"`
}

type Strain struct {
	*StrainBase
	Measurements      NullSliceInt64 `db:"measurements" json:"measurements"`
	Characteristics   NullSliceInt64 `db:"characteristics" json:"characteristics"`
	TotalMeasurements int64          `db:"total_measurements" json:"totalMeasurements"`
	SortOrder         int64          `db:"sort_order" json:"sortOrder"`
	CanEdit           bool           `db:"-" json:"canEdit"`
}

type Strains []*Strain

type StrainMeta struct {
	CanAdd bool `json:"canAdd"`
}

type StrainPayload struct {
	Strain          *Strain          `json:"strain"`
	Species         *ManySpecies     `json:"species"`
	Characteristics *Characteristics `json:"characteristics"`
	Measurements    *Measurements    `json:"measurements"`
	Meta            *StrainMeta      `json:"meta"`
}

type StrainsPayload struct {
	Strains         *Strains         `json:"strains"`
	Species         *ManySpecies     `json:"species"`
	Characteristics *Characteristics `json:"characteristics"`
	Measurements    *Measurements    `json:"measurements"`
	Meta            *StrainMeta      `json:"meta"`
}

func (s *StrainPayload) marshal() ([]byte, error) {
	return json.Marshal(s)
}

func (s *StrainsPayload) marshal() ([]byte, error) {
	return json.Marshal(s)
}

func (s StrainService) unmarshal(b []byte) (entity, error) {
	var sj StrainPayload
	err := json.Unmarshal(b, &sj)
	return &sj, err
}

func (s StrainBase) SpeciesName() string {
	var species SpeciesBase
	if err := DBH.Get(&species, s.SpeciesId); err != nil {
		return ""
	}
	return species.SpeciesName
}

func (s StrainService) list(val *url.Values, claims *Claims) (entity, *appError) {
	if val == nil {
		return nil, ErrMustProvideOptionsJSON
	}
	var opt ListOptions
	if err := schemaDecoder.Decode(&opt, *val); err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	strains, err := listStrains(opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	species_opt, err := speciesOptsFromStrains(opt)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	species, err := listSpecies(*species_opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	characteristics_opt, err := characteristicsOptsFromStrains(opt)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	characteristics, err := listCharacteristics(*characteristics_opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	characteristic_ids := []int64{}
	for _, c := range *characteristics {
		characteristic_ids = append(characteristic_ids, c.Id)
	}

	strain_ids := []int64{}
	for _, s := range *strains {
		strain_ids = append(strain_ids, s.Id)
	}

	measurement_opt := MeasurementListOptions{
		ListOptions: ListOptions{
			Genus: opt.Genus,
		},
		Strains:         strain_ids,
		Characteristics: characteristic_ids,
	}

	measurements, err := listMeasurements(measurement_opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	payload := StrainsPayload{
		Strains:         strains,
		Species:         species,
		Measurements:    measurements,
		Characteristics: characteristics,
		Meta: &StrainMeta{
			CanAdd: canAdd(claims),
		},
	}

	return &payload, nil
}

func (s StrainService) get(id int64, genus string, claims *Claims) (entity, *appError) {
	strain, err := getStrain(id, genus, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	species, err := getSpecies(strain.SpeciesId, genus, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	opt := ListOptions{Genus: genus, Ids: []int64{id}}
	characteristics_opt, err := characteristicsOptsFromStrains(opt)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	characteristics, err := listCharacteristics(*characteristics_opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	characteristic_ids := []int64{}
	for _, c := range *characteristics {
		characteristic_ids = append(characteristic_ids, c.Id)
	}

	measurement_opt := MeasurementListOptions{
		ListOptions: ListOptions{
			Genus: genus,
		},
		Strains:         []int64{id},
		Characteristics: characteristic_ids,
	}

	measurements, err := listMeasurements(measurement_opt, claims)
	if err != nil {
		return nil, newJSONError(err, http.StatusInternalServerError)
	}

	var many_species ManySpecies = []*Species{species}

	payload := StrainPayload{
		Strain:          strain,
		Species:         &many_species,
		Characteristics: characteristics,
		Measurements:    measurements,
		Meta: &StrainMeta{
			CanAdd: canAdd(claims),
		},
	}

	return &payload, nil
}

func (s StrainService) update(id int64, e *entity, genus string, claims *Claims) *appError {
	payload := (*e).(*StrainPayload)
	payload.Strain.UpdatedBy = claims.Sub
	payload.Strain.Id = id

	count, err := DBH.Update(payload.Strain.StrainBase)
	if err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}
	if count != 1 {
		return newJSONError(ErrStrainNotUpdated, http.StatusBadRequest)
	}

	strain, err := getStrain(id, genus, claims)
	if err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}

	species, err := getSpecies(strain.SpeciesId, genus, claims)
	if err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}

	var many_species ManySpecies = []*Species{species}

	payload.Strain = strain
	payload.Species = &many_species
	payload.Meta = &StrainMeta{
		CanAdd: canAdd(claims),
	}

	return nil
}

func (s StrainService) create(e *entity, genus string, claims *Claims) *appError {
	payload := (*e).(*StrainPayload)
	payload.Strain.CreatedBy = claims.Sub
	payload.Strain.UpdatedBy = claims.Sub

	if err := DBH.Insert(payload.Strain.StrainBase); err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}

	strain, err := getStrain(payload.Strain.Id, genus, claims)
	if err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}

	species, err := getSpecies(strain.SpeciesId, genus, claims)
	if err != nil {
		return newJSONError(err, http.StatusInternalServerError)
	}

	var many_species ManySpecies = []*Species{species}

	payload.Strain = strain
	payload.Species = &many_species
	payload.Meta = &StrainMeta{
		CanAdd: canAdd(claims),
	}

	return nil
}

func listStrains(opt ListOptions, claims *Claims) (*Strains, error) {
	var vals []interface{}

	q := `SELECT st.*, array_agg(m.id) AS measurements,
		array_agg(DISTINCT m.characteristic_id) AS characteristics,
		COUNT(m) AS total_measurements,
		rank() OVER (ORDER BY sp.species_name ASC, st.type_strain ASC, st.strain_name ASC) AS sort_order
		FROM strains st
		INNER JOIN species sp ON sp.id=st.species_id
		INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=LOWER($1)
		LEFT OUTER JOIN measurements m ON m.strain_id=st.id`
	vals = append(vals, opt.Genus)

	if len(opt.Ids) != 0 {
		var conds []string
		s := "st.id IN ("
		for i, id := range opt.Ids {
			s = s + fmt.Sprintf("$%v,", i+2) // start param index at 2
			vals = append(vals, id)
		}
		s = s[:len(s)-1] + ")"
		conds = append(conds, s)
		q += " WHERE (" + strings.Join(conds, ") AND (") + ")"
	}

	q += " GROUP BY st.id, st.species_id, sp.species_name;"

	strains := make(Strains, 0)
	err := DBH.Select(&strains, q, vals...)
	if err != nil {
		return nil, err
	}

	for _, s := range strains {
		s.CanEdit = canEdit(claims, s.CreatedBy)
	}

	return &strains, nil
}

func getStrain(id int64, genus string, claims *Claims) (*Strain, error) {
	var strain Strain
	q := `SELECT st.*, array_agg(DISTINCT m.id) AS measurements,
		array_agg(DISTINCT m.characteristic_id) AS characteristics,
		COUNT(m) AS total_measurements, 0 AS sort_order
		FROM strains st
		INNER JOIN species sp ON sp.id=st.species_id
		INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=LOWER($1)
		LEFT OUTER JOIN measurements m ON m.strain_id=st.id
		WHERE st.id=$2
		GROUP BY st.id;`
	if err := DBH.SelectOne(&strain, q, genus, id); err != nil {
		if err == sql.ErrNoRows {
			return nil, ErrStrainNotFound
		}
		return nil, err
	}

	strain.CanEdit = canEdit(claims, strain.CreatedBy)

	return &strain, nil
}

func speciesOptsFromStrains(opt ListOptions) (*ListOptions, error) {
	relatedSpeciesIds := make([]int64, 0)

	if opt.Ids == nil || len(opt.Ids) == 0 {
		q := `SELECT DISTINCT st.species_id
			FROM strains st
			INNER JOIN species sp ON sp.id=st.species_id
			INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=LOWER($1);`
		if err := DBH.Select(&relatedSpeciesIds, q, opt.Genus); err != nil {
			return nil, err
		}
	} else {
		var vals []interface{}
		var count int64 = 1
		q := fmt.Sprintf("SELECT DISTINCT species_id FROM strains WHERE %s;", valsIn("id", opt.Ids, &vals, &count))
		if err := DBH.Select(&relatedSpeciesIds, q, vals...); err != nil {
			return nil, err
		}
	}

	return &ListOptions{Genus: opt.Genus, Ids: relatedSpeciesIds}, nil
}

func characteristicsOptsFromStrains(opt ListOptions) (*ListOptions, error) {
	relatedCharacteristicsIds := make([]int64, 0)

	if opt.Ids == nil || len(opt.Ids) == 0 {
		q := `SELECT DISTINCT m.characteristic_id
				FROM measurements m
				INNER JOIN strains st ON st.id=m.strain_id
				INNER JOIN species sp ON sp.id=st.species_id
				INNER JOIN genera g ON g.id=sp.genus_id AND LOWER(g.genus_name)=LOWER($1);`
		if err := DBH.Select(&relatedCharacteristicsIds, q, opt.Genus); err != nil {
			return nil, err
		}
	} else {
		var vals []interface{}
		var count int64 = 1
		q := fmt.Sprintf("SELECT DISTINCT characteristic_id FROM measurements WHERE %s;", valsIn("strain_id", opt.Ids, &vals, &count))
		if err := DBH.Select(&relatedCharacteristicsIds, q, vals...); err != nil {
			return nil, err
		}
	}

	return &ListOptions{Genus: opt.Genus, Ids: relatedCharacteristicsIds}, nil
}