List filename and write to text file

listfilename.go

package main

import (
	"fmt"
	"os"
	"path/filepath"
	"sort"
)

func main() {

	var files []string

	var rootDir []string
	rootDir = append(rootDir, "~/Downloads")
	rootDir = append(rootDir, "~/Picture")

	for _, rDir := range rootDir {
		err := filepath.Walk(rDir, func(path string, info os.FileInfo, err error) error {

			if info.IsDir() {
				return nil
			}

			// if !strings.HasSuffix(info.Name(), ".mkv") {
			// 	return nil
			// }

			files = append(files, info.Name())
			return nil
		})
		if err != nil {
			panic(err)
		}
	}

	sort.Strings(files)

	f, err := os.Create("filename.txt")
	if err != nil {
		fmt.Println(err)
		return
	}

	for _, file := range files {
		// fmt.Println(file)
		_, err := f.WriteString(file + "\n")
		if err != nil {
			fmt.Println(err)
			f.Close()
			return
		}
	}

	err = f.Close()

}