package influx

import (
	"context"
	"fmt"
	"net/url"
	"strings"
)

const (
	bucket string = "influxBucket"
)

const (
	FieldCPrifix string = "c"
	FieldIPrefix string = "i"

	// FieldP  string = "p"
	// FieldQ  string = "q"
	// FieldS  string = "s"
	// FieldPF string = "pf"

	// FieldF  string = "f"
	FieldDF string = "df"

	// FieldUABPrefix string = "uab"
	// FieldUBCPrefix string = "ubc"
	// FieldUCAPrefix string = "uca"

	FieldSuffixAMP = "_amp"
	FieldSuffixPA  = "_pa"
	FieldSuffixRMS = "_rms"
)

func (client *influxClient) GetSSUPointLastLimit(ctx context.Context, req *Request, limit int) ([]TV, error) {
	sql := fmt.Sprintf("select last(%s) as %s from %s where station='%s' and device='%s';",
		req.SubPos, req.SubPos, req.Measure, req.Station, req.MainPos)
	if limit > 1 {
		sql = fmt.Sprintf("select %s from %s where station='%s' and device='%s' and time>%dms order by time desc limit %d;",
			req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin, limit) // begin = time.Now().UnixMilli()-int64(limit*20+20)
	}

	reqData := url.Values{
		"db": {bucket},
		"q":  {sql},
	}

	return client.getTVsResp(ctx, reqData, req.RespType)
}

func (client *influxClient) GetSSUPointsLastLimit(ctx context.Context, req *Request, limit int) (map[string][]TV, error) {
	sql := ""
	if limit == 1 {
		fields := strings.Split(req.SubPos, ",")
		for i, field := range fields {
			fields[i] = "last(" + field + ") as " + field
		}
		sql = fmt.Sprintf("select %s from %s where station='%s' and device='%s';",
			strings.Join(fields, ","), req.Measure, req.Station, req.MainPos)
	} else {
		sql = fmt.Sprintf("select %s from %s where station='%s' and device='%s' and time>%dms order by time desc limit %d;",
			req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin, limit) // begin = time.Now().UnixMilli()-int64(limit*20+20)
	}

	reqData := url.Values{
		"db": {bucket},
		"q":  {sql},
	}

	return client.getF2TVsResp(ctx, reqData, req.RespType)
}

func (client *influxClient) GetSSUPointData(ctx context.Context, req *Request) ([]TV, error) {
	sql := fmt.Sprintf("select %s from %s where station='%s' and device='%s' and time>=%dms and time<%dms;",
		req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin, req.End)

	if req.Operate != "" && req.Step != "" && req.Default != "" {
		sql = fmt.Sprintf("select %s(%s) as %s from %s where station='%s' and device='%s' and time>=%dms and time<%dms group by time(%s) fill(%s);",
			req.Operate, req.SubPos, req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin, req.End, req.Step, req.Default)
	}

	reqData := url.Values{
		"db": {bucket},
		"q":  {sql},
	}

	return client.getTVsResp(ctx, reqData, req.RespType)
}

func (client *influxClient) GetSSUPointAfterOne(ctx context.Context, req *Request) ([]TV, error) {
	sql := fmt.Sprintf("select %s from %s where station='%s' and device='%s' and time>=%dms limit 1;",
		req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin)

	reqData := url.Values{
		"db": {bucket},
		"q":  {sql},
	}

	return client.getTVsResp(ctx, reqData, req.RespType)
}

func (client *influxClient) GetSSUPointBeforeOne(ctx context.Context, req *Request) ([]TV, error) {
	reqData := url.Values{
		"db": {bucket},
		"q": {fmt.Sprintf("select %s from %s where station='%s' and device='%s' and time>%dms and time<=%dms order by time desc limit 1;",
			req.SubPos, req.Measure, req.Station, req.MainPos, req.Begin, req.End)}, // begin = req.End-20-20
	}

	return client.getTVsResp(ctx, reqData, req.RespType)
}

func (client *influxClient) WriteLinesData(ctx context.Context, data []byte) error {
	return client.writeLinesData(ctx, bucket, data, true)
}