package qm

import "github.com/nullbio/sqlboiler/boil"

// QueryMod to modify the query object
type QueryMod func(q *boil.Query)

// Apply the query mods to the Query object
func Apply(q *boil.Query, mods ...QueryMod) {
	for _, mod := range mods {
		mod(q)
	}
}

// SQL allows you to execute a plain SQL statement
func SQL(sql string, args ...interface{}) QueryMod {
	return func(q *boil.Query) {
		boil.SetSQL(q, sql, args...)
	}
}

// Or surrounds where clauses to join them with OR as opposed to AND
func Or(whereMods ...QueryMod) QueryMod {
	return func(q *boil.Query) {
		if len(whereMods) < 2 {
			panic("Or requires at least two arguments")
		}

		for _, w := range whereMods {
			w(q)
			boil.SetLastWhereAsOr(q)
		}
	}
}

// Limit the number of returned rows
func Limit(limit int) QueryMod {
	return func(q *boil.Query) {
		boil.SetLimit(q, limit)
	}
}

// Offset into the results
func Offset(offset int) QueryMod {
	return func(q *boil.Query) {
		boil.SetOffset(q, offset)
	}
}

// InnerJoin on another table
func InnerJoin(clause string, args ...interface{}) QueryMod {
	return func(q *boil.Query) {
		boil.AppendInnerJoin(q, clause, args...)
	}
}

// Select specific columns opposed to all columns
func Select(columns ...string) QueryMod {
	return func(q *boil.Query) {
		boil.AppendSelect(q, columns...)
	}
}

// Where allows you to specify a where clause for your statement
func Where(clause string, args ...interface{}) QueryMod {
	return func(q *boil.Query) {
		boil.AppendWhere(q, clause, args...)
	}
}

// GroupBy allows you to specify a group by clause for your statement
func GroupBy(clause string) QueryMod {
	return func(q *boil.Query) {
		boil.ApplyGroupBy(q, clause)
	}
}

// OrderBy allows you to specify a order by clause for your statement
func OrderBy(clause string) QueryMod {
	return func(q *boil.Query) {
		boil.ApplyOrderBy(q, clause)
	}
}

// Having allows you to specify a having clause for your statement
func Having(clause string) QueryMod {
	return func(q *boil.Query) {
		boil.ApplyHaving(q, clause)
	}
}

// From allows to specify the table for your statement
func From(from string) QueryMod {
	return func(q *boil.Query) {
		boil.AppendFrom(q, from)
	}
}

// Count turns the query into a counting calculation
func Count() QueryMod {
	return func(q *boil.Query) {
		boil.SetCount(q)
	}
}

// Avg turns the query into an average calculation
func Avg() QueryMod {
	return func(q *boil.Query) {
		boil.SetAvg(q)
	}
}

// Min turns the query into a minimum calculation
func Min() QueryMod {
	return func(q *boil.Query) {
		boil.SetMin(q)
	}
}

// Max turns the query into a maximum calculation
func Max() QueryMod {
	return func(q *boil.Query) {
		boil.SetMax(q)
	}
}

// Sum turns the query into a sum calculation
func Sum() QueryMod {
	return func(q *boil.Query) {
		boil.SetSum(q)
	}
}