kefu/wechathook/service/tool.go

package service

import (
	"bytes"
	"encoding/json"
	"errors"
	"fmt"
	"io"
	"io/ioutil"
	"math/rand"
	"net/http"
	"os"
	"os/exec"
	"path/filepath"
	"regexp"
	"strconv"
	"strings"
	"time"
)

// PostJSON 发送HTTP POST请求并发送JSON数据
func PostJSON(url string, jsonData interface{}) ([]byte, error) {
	// 将JSON数据转换为字节切片
	jsonBytes, err := json.Marshal(jsonData)
	if err != nil {
		return nil, err
	}

	// 创建一个HTTP请求
	req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonBytes))
	if err != nil {
		return nil, err
	}
	req.Header.Set("Content-Type", "application/json")

	// 发送HTTP请求并获取响应
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	// 读取响应的内容
	respBody, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		return nil, err
	}

	return respBody, nil
}

/*
*
去除html标签，过滤html标签
*/
func TrimHtml(src string) string {
	//将HTML标签全转换成小写
	re, _ := regexp.Compile("\\<[\\S\\s]+?\\>")
	src = re.ReplaceAllStringFunc(src, strings.ToLower)
	//去除STYLE
	re, _ = regexp.Compile("\\<style[\\S\\s]+?\\</style\\>")
	src = re.ReplaceAllString(src, "")
	//去除SCRIPT
	re, _ = regexp.Compile("\\<script[\\S\\s]+?\\</script\\>")
	src = re.ReplaceAllString(src, "")
	//去除所有尖括号内的HTML代码，并换成换行符
	re, _ = regexp.Compile("\\<[\\S\\s]+?\\>")
	src = re.ReplaceAllString(src, "\n")
	//去除连续的换行符
	re, _ = regexp.Compile("\\s{2,}")
	src = re.ReplaceAllString(src, "\n")
	return strings.TrimSpace(src)
}

// GetIDCardFromText 从文本中提取身份证号
func GetIDCardFromText(str string) (string, error) {
	// 正则表达式，匹配18位或15位的身份证号
	// 18位身份证号格式: XXXXXXYYYYMMDDXXXXXX
	// 15位身份证号格式: XXXXXXYYYYMMDDXXX
	// 其中X表示数字
	var idCardRegex = regexp.MustCompile(`\d{17}[\dXx]|\d{15}`)
	matches := idCardRegex.FindAllString(str, -1)

	if len(matches) == 0 {
		return "", errors.New("没有找到身份证号")
	}

	// 如果有多个身份证号，返回第一个找到的
	// 或者根据具体需求可以返回所有匹配的身份证号
	return matches[0], nil
}

// 根据输入的字符串参数决定返回的随机数。如果参数包含 -，则将其解析为范围，并返回该范围内的随机整数。如果参数不包含 -，则将其解析为整数并直接返回
func GetRandomValue(input string) (int, error) {
	if strings.Contains(input, "-") {
		ranges := strings.Split(input, "-")
		if len(ranges) != 2 {
			return 0, fmt.Errorf("invalid input format")
		}
		min, err := strconv.Atoi(ranges[0])
		if err != nil {
			return 0, fmt.Errorf("invalid minimum value")
		}
		max, err := strconv.Atoi(ranges[1])
		if err != nil {
			return 0, fmt.Errorf("invalid maximum value")
		}
		if min >= max {
			return 0, fmt.Errorf("minimum value must be less than maximum value")
		}
		return rand.Intn(max-min+1) + min, nil
	} else {
		val, err := strconv.Atoi(input)
		if err != nil {
			return 0, fmt.Errorf("invalid integer value")
		}
		return val, nil
	}
}

// isWindowsPath 判断字符串是否是有效的Windows文件路径
func IsWindowsPath(path string) bool {
	// 检查路径是否以驱动器号开头，如 "C:\"
	if len(path) > 2 && path[1] == ':' && path[2] == filepath.Separator {
		// 检查路径中是否包含反斜杠
		for i := 3; i < len(path); i++ {
			if path[i] == filepath.Separator {
				return true
			}
		}
	}
	return false
}

// isImagePath 判断字符串是否可能是一个图片路径
func IsImagePath(path string) bool {
	// 定义常见的图片文件扩展名
	imageExtensions := []string{".jpg", ".jpeg", ".png", ".gif", ".bmp"}

	// 获取路径的文件名部分
	fileName := filepath.Base(path)

	// 检查文件名是否以这些扩展名之一结尾
	for _, ext := range imageExtensions {
		if strings.ToLower(strings.TrimPrefix(fileName, ext)) == fileName {
			return true
		}
	}

	// 如果没有匹配的扩展名，则不是图片路径
	return false
}

// 正则提取
func GetOneStringByRegex(str, rule string) (string, error) {
	reg, err := regexp.Compile(rule)
	if reg == nil || err != nil {
		return "", errors.New("正则MustCompile错误:" + err.Error())
	}
	//提取关键信息
	result := reg.FindStringSubmatch(str)
	if len(result) < 1 {
		return "", errors.New("没有获取到子字符串")
	}
	return result[1], nil
}
func ParseImgMessage(str string) string {
	var url string
	url, _ = GetOneStringByRegex(str, "img\\[(.*?)\\]")
	return url
}

// 下载一个网络文件并保存到本地
func DownloadFile(url string, filename string) error {
	// 创建一个HTTP客户端
	client := &http.Client{}
	// 发送GET请求
	req, err := http.NewRequest("GET", url, nil)
	if err != nil {
		return err
	}

	// 发送请求并获取响应
	resp, err := client.Do(req)
	if err != nil {
		return err
	}
	defer resp.Body.Close()

	// 检查HTTP响应状态码
	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("服务器返回状态码: %d", resp.StatusCode)
	}

	// 创建文件并写入响应体内容
	out, err := os.Create(filename)
	if err != nil {
		return err
	}
	defer out.Close()

	// 将响应体内容写入文件
	_, err = io.Copy(out, resp.Body)
	if err != nil {
		return err
	}

	return nil
}

// 定义一个结构体来存储图片的路径和描述
type MarkdownImage struct {
	Path string
	Desc string
}

// 定义一个函数来提取Markdown中的图片路径和描述
func ExtractMarkdownImagePaths(markdown string) []MarkdownImage {
	// 正则表达式匹配Markdown图片路径和描述
	// 组1: 图片描述，组2: 图片路径
	re := regexp.MustCompile(`!\[(.*?)\]\((.*?)\)`)

	// 使用FindAllStringSubmatch查找所有匹配的图片路径和描述
	matches := re.FindAllStringSubmatch(markdown, -1)

	// 初始化一个切片来存储MarkdownImage结构体
	images := make([]MarkdownImage, len(matches))

	// 提取图片路径和描述
	for i, match := range matches {
		if len(match) < 3 { // 确保有足够的匹配组
			continue
		}
		images[i] = MarkdownImage{
			Path: match[2], // 索引2是图片路径
			Desc: match[1], // 索引1是图片描述
		}
	}

	return images
}

// 获取当前日期格式化为年月日的字符串
func GetCurrentDateDir() string {
	currentTime := time.Now() // 获取当前时间
	// 格式化为年月日，例如 "20240624"
	dateStr := currentTime.Format("20060102")
	return dateStr
}

// 创建按照当前年月日命名的目录
func CreateDateDir(baseDir, dateStr string) error {
	dirPath := filepath.Join(baseDir, dateStr) // 构建目录路径
	if _, err := os.Stat(dirPath); os.IsNotExist(err) {
		// 如果目录不存在，则创建目录
		err := os.MkdirAll(dirPath, 0755)
		if err != nil {
			return err
		}
	}
	return nil
}

// 从URL中提取文件名和扩展名
func ExtractFilenameAndExt(url string) (string, string) {
	// 提取URL中的最后一个路径部分作为文件名
	filename := filepath.Base(url)
	// 分割文件名和扩展名
	ext := filepath.Ext(filename)
	// 如果没有扩展名，返回空字符串
	if ext == filename {
		ext = ""
	} else {
		// 移除点号，只保留扩展名
		ext = ext[1:]
	}
	return filename, ext
}

// 下载图片到按照当前年月日创建的目录
func DownloadImageToDailyDir(url, baseResourceDir, filename string) (string, error) {
	// 从URL中提取文件名和扩展名
	fileNameWithoutExt, ext := ExtractFilenameAndExt(url)

	// 如果提供的filename为空，则使用从URL中提取的文件名
	if filename == "" {
		filename = fileNameWithoutExt
	}
	// 如果文件名没有扩展名，则添加从URL中提取的扩展名
	if !strings.Contains(filename, "."+ext) {
		filename += "." + ext
	}

	// 获取当前日期目录名
	dateDir := GetCurrentDateDir()
	// 创建基础目录和日期目录
	if err := CreateDateDir(baseResourceDir, dateDir); err != nil {
		return "", err
	}

	// 构建文件路径
	filePath := filepath.Join(baseResourceDir, dateDir, filename)

	// 下载图片
	resp, err := http.Get(url)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	// 检查响应状态码
	if resp.StatusCode != http.StatusOK {
		return "", fmt.Errorf("http status code: %d", resp.StatusCode)
	}

	// 创建文件并打开文件句柄
	file, err := os.Create(filePath)
	if err != nil {
		return "", err
	}
	defer file.Close()

	// 将响应体的内容写入文件
	_, err = io.Copy(file, resp.Body)
	if err != nil {
		return "", err
	}

	return filePath, nil
}

// 判断文件文件夹不存在
func IsFileNotExist(path string) (bool, error) {
	_, err := os.Stat(path)
	if os.IsNotExist(err) {
		return true, nil
	}
	return false, err
}

// 获取程序执行目录
func GetRunPath2() string {
	file, _ := exec.LookPath(os.Args[0])
	path, _ := filepath.Abs(file)
	index := strings.LastIndex(path, string(os.PathSeparator))
	ret := path[:index]
	return ret
}

// 获取程序根目录
func GetRootPath() string {
	rootPath, _ := os.Getwd()
	if notExist, _ := IsFileNotExist(rootPath); notExist {
		rootPath = GetRunPath2()
		if notExist, _ := IsFileNotExist(rootPath); notExist {
			rootPath = "."
		}
	}
	return rootPath
}