package main

import (
	// 导入包，导入前缀为下划线，则init函数被执行，然后注册驱动。
	"fmt"
	"fstool/db"
	"fstool/util"
	"os"
	"path/filepath"
)

func main() {
	// 获取信息
	args := os.Args
	// no input
	if args == nil || len(args) < 2 {
		fmt.Printf("input your real path, like /home etc.\n")
		os.Exit(1)
	}

	// 第一个是输入的参数
	// 路径
	path := args[1]
	//fmt.Printf("your path is :%s\n", path)

	// open datebase
	db.Init()
	// 创建数据表
	db.CreateTable()
	//遍历
	ftree, err := util.GetDirFilePaths(path, false)
	if err != nil {
		fmt.Printf("scan error is :%v", err)
	}
	//计算hash，然后存放到数据库
	for _, v := range ftree {
		//fmt.Printf("%v\n", v)
		// 计算结果
		//fmt.Printf("%v,%s\n", filepath.Dir(v), filepath.Base(v))
		relpath, err := filepath.Rel(path, v)
		if err != nil {
			fmt.Printf("relpath error :%v", err)
		}
		//fmt.Printf("relpath:%s\n", filepath.Dir(relpath))
		// 拼装文件信息插入数据库
		var stf db.StFileInfo
		stf.Fhash = util.CalacHash(v)
		stf.Fname = filepath.Base(v)

		// path
		rlpath := filepath.ToSlash(relpath)
		fmt.Printf("relpath:%s\n", filepath.Dir(rlpath))
		rlpath = filepath.Join(rlpath, "/")
		stf.Fpath = filepath.Dir(rlpath)

		// fmt.Printf("relpath:%s\n", filepath.Dir(rlpath))
		// insert into db
		// fmt.Printf("stf :%v\n", stf)
		db.InsertStf(stf)
	}
	// fmt.Printf("scan is :%v", ftree)
	// connectDB()
	// insert(1, 0.81192)
}