package sms

import (
	"encoding/json"
	"errors"
)

type smsResponse struct {
	Status                  string `json:"status"`
	Msg                     string `json:"msg"`
	SendId                  string `json:"send_id"`
	Fee                     int    `json:"fee"`
	SmsCredits              string `json:"sms_credits"`
	TransactionalSmsCredits string `json:"transactional_sms_credits"`
}

func XSend(to, tempId string, data map[string]string, appId, appKey string) error {
	config := map[string]string{
		"appid":    appId,
		"appkey":   appKey,
		"signType": "md5",
	}

	s := CreateXsend(config)
	ret := s.SetTo(to).SetProject(tempId).AddVars(data).Xsend()

	retData := smsResponse{}
	err := json.Unmarshal([]byte(ret), &retData)
	if err != nil {
		return err
	}
	if retData.Status == "error" {
		return errors.New(retData.Msg)
	}
	return nil
}

func MultiXSend(to []string, tempId string, data map[string]string, appId, appKey string) ([]smsResponse, []string, error) {
	config := map[string]string{
		"appid":    appId,
		"appkey":   appKey,
		"signType": "md5",
	}

	s := CreateMultiXsend(config)

	for _, phone := range to {
		msg := CreateMulti()
		msg.SetTo(phone).AddVars(data)
		s.AddMulti(msg.Get())
	}

	ret := s.SetProject(tempId).MultiXsend()

	var retData []smsResponse
	failData := make([]string, len(retData))
	err := json.Unmarshal([]byte(ret), &retData)
	if err != nil {
		return nil, failData, err
	}
	for _, value := range retData {
		if value.Status == "error" {
			failData = append(failData, value.SendId)
		}
	}

	return retData, failData, nil
}