90 lines
2.3 KiB
Go
90 lines
2.3 KiB
Go
// Copyright 2021 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 (
|
||
"crypto/md5"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"io"
|
||
"net/http"
|
||
"net/url"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type HuyiClient struct {
|
||
appId string
|
||
appKey string
|
||
template string
|
||
}
|
||
|
||
func GetHuyiClient(appId string, appKey string, template string) (*HuyiClient, error) {
|
||
return &HuyiClient{
|
||
appId: appId,
|
||
appKey: appKey,
|
||
template: template,
|
||
}, nil
|
||
}
|
||
|
||
func GetMd5String(s string) string {
|
||
h := md5.New()
|
||
h.Write([]byte(s))
|
||
return hex.EncodeToString(h.Sum(nil))
|
||
}
|
||
|
||
func (hc *HuyiClient) SendMessage(param map[string]string, targetPhoneNumber ...string) error {
|
||
code, ok := param["code"]
|
||
if !ok {
|
||
return fmt.Errorf("missing parameter: code")
|
||
}
|
||
|
||
if len(targetPhoneNumber) == 0 {
|
||
return fmt.Errorf("missin parer: trgetPhoneNumber")
|
||
}
|
||
|
||
_now := strconv.FormatInt(time.Now().Unix(), 10)
|
||
smsContent := fmt.Sprintf(hc.template, code)
|
||
v := url.Values{}
|
||
v.Set("account", hc.appId)
|
||
v.Set("content", smsContent)
|
||
v.Set("time", _now)
|
||
passwordStr := hc.appId + hc.appKey + "%s" + smsContent + _now
|
||
for _, mobile := range targetPhoneNumber {
|
||
password := fmt.Sprintf(passwordStr, mobile)
|
||
v.Set("password", GetMd5String(password))
|
||
v.Set("mobile", mobile)
|
||
|
||
body := strings.NewReader(v.Encode()) // encode form data
|
||
client := &http.Client{}
|
||
req, _ := http.NewRequest("POST", "http://106.ihuyi.com/webservice/sms.php?method=Submit&format=json", body)
|
||
|
||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded; param=value")
|
||
|
||
resp, err := client.Do(req) // request remote
|
||
if err != nil {
|
||
return err
|
||
}
|
||
defer resp.Body.Close() // ！ close ReadCloser
|
||
_, err = io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
|
||
return nil
|
||
}
|