// ************************
// ************************
// 日期：2022年9月14日
// 作者：ZeroKong/东方零食
// ************************
// ************************
package main

import (
	"container/list"
	"encoding/csv"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"
)

// 结构体
type Data struct { //带结构标签，反引号来包围字符串
	Date     string              `json:"date"`
	Author   string              `json:"author"`
	Email    string              `json:"email"`
	ItAssets map[string]ItAssets `json:"itAssets"`
}

type ItAssets struct {
	// 位置、部门、负责人、IP地址、系统名称、备注、整改方案字段名称
	Address    string `json:"位置"`
	Department string `json:"部门"`
	Leader     string `json:"负责人"`
	IPAddress  string `json:"IP地址"`
	System     string `json:"系统名称"`
	Remark     string `json:"备注"`
	Plan       string `json:"整改方案"`
}

func main() {
	// 打开文件
	jsonFile, err := os.Open("data.json")
	// 如果打开失败，则终止程序
	if err != nil {
		// 输出错误信息
		fmt.Println("error opening json file")
		return
	}

	// 关闭文件
	defer func(jsonFile *os.File) {
		// 如果关闭失败，则终止程序
		err := jsonFile.Close()
		if err != nil {
			// 输出错误信息
			fmt.Println("error closing json file")
		}
	}(jsonFile)

	// 读取文件内容
	jsonData, err := ioutil.ReadAll(jsonFile)
	// 如果读取失败，则终止程序
	if err != nil {
		// 输出错误信息
		fmt.Println("error reading json file")
		return
	}

	// 将json数据解析到结构体
	var data Data
	err = json.Unmarshal(jsonData, &data)
	if err != nil {
		// 输出错误信息
		fmt.Println("error unmarshalling json data")
		return
	}

	var cvslist list.List
	//遍历扫描报告目录下的文件
	//此处需要优化，不判断文件夹
	files, _ := os.ReadDir("./扫描报告")
	for _, file := range files {
		//去除".doc"字符
		fileName := file.Name()
		fileName = fileName[:len(fileName)-4]
		// map判断key是否存在
		if _, ok := data.ItAssets[fileName]; ok {
			// 存在
			var report []byte
			report, err = json.Marshal(data.ItAssets[fileName])
			cvslist.PushBack(string(report))
			fmt.Println(string(report))
			//判断部门\负责人\系统名称目录是否存在不存在则创建目录
			if _, err := os.Stat("./扫描报告/" + data.ItAssets[fileName].Department + "/" + data.ItAssets[fileName].Leader + "/" + data.ItAssets[fileName].System); os.IsNotExist(err) {
				err := os.MkdirAll("./扫描报告/"+data.ItAssets[fileName].Department+"/"+data.ItAssets[fileName].Leader+"/"+data.ItAssets[fileName].System, os.ModePerm)
				if err != nil {
					return
				}
			}
			//移动file.Name()到指定目录
			err = os.Rename("./扫描报告/"+file.Name(), "./扫描报告/"+string(data.ItAssets[fileName].Department)+"/"+string(data.ItAssets[fileName].Leader)+"/"+string(data.ItAssets[fileName].System)+"/"+file.Name())
			if err != nil {
				fmt.Println(err)
			}
		} else {
			// 不存在
			//因为在mac下编译有个隐藏文件先忽略，后续使用执行文件后删除判断即可
			if fileName == ".DS_S" {
				fmt.Println("文件名为.DS_S")
			} else {
				fmt.Println(fileName)
				//判断部门\负责人\系统名称目录是否存在不存在则创建目录
				if _, err := os.Stat("./扫描报告/无法核实"); os.IsNotExist(err) {
					err := os.MkdirAll("./扫描报告/无法核实", os.ModePerm)
					if err != nil {
						return
					}
				}
				//移动file.Name()到指定目录
				err = os.Rename("./扫描报告/"+file.Name(), "./扫描报告/无法核实/"+file.Name())
				if err != nil {
					fmt.Println(err)
				}
				cvslist.PushBack("{\"位置\":\"\",\"部门\":\"\",\"负责人\":\"\",\"IP地址\":\"" + fileName + "\",\"系统名称\":\"\",\"备注\":\"\",\"整改方案\":\"\"}")
			}
		}
	}

	//将cvslist每一行json中的位置、部门、负责人、IP地址、系统名称、备注、整改方案写入csv文件
	//创建csv文件
	csvFile, err := os.Create("./扫描报告.csv")
	if err != nil {
		fmt.Println(err)
		return
	}
	defer csvFile.Close()

	//写入UTF-8 BOM
	csvFile.WriteString("\xEF\xBB\xBF")
	//写入文件头
	csvWriter := csv.NewWriter(csvFile)
	csvWriter.Write([]string{"位置", "部门", "负责人", "IP地址", "系统名称", "备注", "整改方案"})
	//遍历cvslist
	for e := cvslist.Front(); e != nil; e = e.Next() {
		var data02 ItAssets
		err := json.Unmarshal([]byte(e.Value.(string)), &data02)
		if err != nil {
			fmt.Println(err)
			return
		}

		csvWriter.Write([]string{data02.Address, data02.Department, data02.Leader, data02.IPAddress, data02.System, data02.Remark, data02.Plan})
	}
	csvWriter.Flush()

}