package main

import (

	"bytes"

	"fmt"

	"io/ioutil"

	"log"

	"os"

	"os/exec"

	"strings"

)

func main() {

	// 读取当前目录中的所有文件和子目录

	path := "E:/video"

	files, err := ioutil.ReadDir(path)

	if err != nil {

		panic(err)

	}

	// 获取文件，并输出它们的名字

	for _, file := range files {

		realPath := path+"/"+file.Name()

		name := file.Name()

		nameArr := strings.Split(name,".")

		//fmt.Println(nameArr)

		ext := nameArr[len(nameArr)-1]

		if ext == "mp4" || ext == "flv" {

			fmt.Println(name)

			translate(realPath,name,ext )

		}

	}

}

func translate(inputPath string,fileName string,ext string  ){

	var out bytes.Buffer

	fmt.Print("默认输出音频路径: d:/audio\n")

	outputName := strings.Replace(fileName,ext , "aac", 1)

	fmt.Println(outputName)

	outPutDir := "d:/audio"

	mkdirOutputDir(outPutDir)

	outputPath := outPutDir+"/"+outputName

	fileExist := FileExist(outputPath)

	if fileExist == true {

		deleteFile(outputPath)

	}

	cmdArguments := []string{"-i", inputPath, outputPath}

	cmd := exec.Command("ffmpeg", cmdArguments...)

	cmd.Stdout = &out

	err := cmd.Run()

	if err != nil {

		log.Fatal(err)

	} else{

		fmt.Println("...... 转换成功.......")

	}

}

func deleteFile(fileName string) {

	err := os.Remove(fileName)

	if err != nil {

		//fmt.Println("删除已经存在的文件 "+fileName+" 失败")

	} else {

		//fmt.Println("删除已经存在的文件 "+fileName+" 成功")

	}

}

func mkdirOutputDir(dir string)  {

	exist, err := PathExists(dir)

	if err != nil {

		fmt.Printf("get dir error![%v]\n", err)

		return

	}

	if exist {

		//fmt.Printf("已经存在目录 [%v]\n", dir)

	} else {

		//fmt.Printf("不存在目录[%v]\n", dir)

		// 创建文件夹

		err := os.Mkdir(dir, os.ModePerm)

		if err != nil {

			//fmt.Printf("创建目录失败[%v]\n", err)

		} else {

			//fmt.Printf("创建目录成功\n")

		}

	}

}

// 判断文件夹是否存在

func PathExists(path string) (bool, error) {

	_, err := os.Stat(path)

	if err == nil {

		return true, nil

	}

	if os.IsNotExist(err) {

		return false, nil

	}

	return false, err

}

func FileExist(path string) bool {

	_, err := os.Lstat(path)

	return !os.IsNotExist(err)

}