// Copyright 2023 The Casdoor Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package go_sms_sender

import (
	"bytes"
	"encoding/xml"
	"errors"
	"fmt"
	"io"
	"net/http"
)

type NetgsmClient struct {
	accessId   string
	accessKey  string
	sign       string
	template   string
	httpClient *http.Client
}

type NetgsmResponse struct {
	Code  string `xml:"main>code"`
	JobID string `xml:"main>jobID"`
	Error string `xml:"main>error"`
}

func GetNetgsmClient(accessId, accessKey, sign, template string) (*NetgsmClient, error) {
	return &NetgsmClient{
		accessId:   accessId,
		accessKey:  accessKey,
		sign:       sign,
		template:   template,
		httpClient: &http.Client{},
	}, nil
}

func (c *NetgsmClient) SendMessage(param map[string]string, targetPhoneNumber ...string) error {
	if len(targetPhoneNumber) == 0 {
		return fmt.Errorf("missing parameter: targetPhoneNumber")
	}

	for _, phoneNumber := range targetPhoneNumber {
		data := fmt.Sprintf(`
<mainbody>
   <header>
       <usercode>%s</usercode>
       <password>%s</password>
       <msgheader>%s</msgheader>
   </header>
   <body>
       <msg>
           <![CDATA[%s]]>
       </msg>
       <no>%s</no>
   </body>
</mainbody>`, c.accessId, c.accessKey, c.sign, c.template, phoneNumber)

		headers := map[string]string{
			"Content-Type": "application/xml",
		}

		respBody, err := c.postXML("https://api.netgsm.com.tr/sms/send/otp", data, headers)
		if err != nil {
			return err
		}

		var netgsmResponse NetgsmResponse
		if err := xml.Unmarshal([]byte(respBody), &netgsmResponse); err != nil {
			return err
		}

		if netgsmResponse.Code != "0" {
			return errors.New(netgsmResponse.Error)
		}
	}
	return nil
}

func (c *NetgsmClient) postXML(url, xmlData string, headers map[string]string) (string, error) {
	req, err := http.NewRequest("POST", url, bytes.NewBuffer([]byte(xmlData)))
	if err != nil {
		return "", err
	}

	for key, value := range headers {
		req.Header.Set(key, value)
	}

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	respBody, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}

	return string(respBody), nil
}