complete main function

Author: gtindo

.gitignore

@@ -1,2 +1,3 @@
 organizer
-tests
+tests
+test_1

archiver.go

@@ -8,3 +8,5 @@ const FLATE = "flate"
 const GZIP = "gzip"
 const LZW = "lzw"
 const ZLIB = "zlib"
+
+// for future releases

main.go

@@ -2,33 +2,54 @@ package main
 
 import (
 	"fmt"
+	"os"
+	"strings"
 )
 
+const VALIDATION_ERROR = -1
+
 func main() {
-	InitArgs()
+	var check bool
+	var msg string
 
-	fmt.Println(Dirs)
+	InitArgs()
 
+	dirs := strings.Split(Dirs, ", ")
+
+	// directories validation
+	check, msg = AreDirectories(dirs)
+	if !check {
+		fmt.Println(msg)
+		os.Exit(VALIDATION_ERROR)
+	}
+
+	// lang validation
+	check, msg = isLangCorrect(Lang)
+	if !check {
+		fmt.Println(msg)
+		os.Exit(VALIDATION_ERROR)
+	}
+
+	// method validation
+	check, msg = isMethodCorrect(Method)
+	if !check {
+		fmt.Println(msg)
+		os.Exit(VALIDATION_ERROR)
+	}
+
+	// TODO for future release
+	// Archive validation
+	// Compress validation
+
+	for _, dir := range dirs {
+		err := MoveFiles(dir, OrganizeDir(
+			dir,
+			strings.Split(IgnoresFiles, ", "),
+			strings.Split(IgnoresExt, ", "),
+			Lang, Method))
+
+		if err != nil {
+			fmt.Println("An error occured while processing ", err)
+		}
+	}
 }
-
-// organize folder by types of file (extensions)
-// images, musics, videos, docs, packages, execs, books,
-// archives, fonts,
-
-// organize folder by alphabetical order
-
-// Organize by extension
-
-// By name looking same
-
-// By last modification date
-
-// archive folders, select archive type
-
-// work on multiple folders simultanouely
-
-// ignore some extensions
-
-// ignore some files
-
-// lang choice

organizers.go

@@ -1,7 +1,8 @@
 package main
 
 import (
-	"os"
+	"fmt"
+	"io/ioutil"
 	"strings"
 )
 
@@ -12,22 +13,37 @@ const DATE_M = "date"
 const EXTENSION_M = "ext"
 const LOOKING_SAME = "same" // For future release
 
-func OrganizeDir(filesInfos []os.FileInfo, ignore []string, lang string, method string) map[string][]string {
+func OrganizeDir(
+	dirPath string,
+	ignoreFiles []string,
+	ignoresExt []string,
+	lang string,
+	method string) map[string][]string {
+
+	filesInfos, err := ioutil.ReadDir(dirPath)
+	if err != nil {
+		fmt.Println("Error while reading ", dirPath)
+		return nil
+	}
+
 	var ext string
 	var fileType FileType
 	var ftName string
 	var uName string
 	var firstL string
 	var dateStr string
-	var isIgnore bool
+	var isIgnoreF bool
+	var isIgnoreE bool
 
 	organizer := make(map[string][]string)
 
 	for _, info := range filesInfos {
-		isIgnore = Contains(info.Name(), ignore)
+		ext = GetFileExtension(info.Name())
+		isIgnoreF = Contains(info.Name(), ignoreFiles)
+		isIgnoreE = Contains(ext, ignoresExt)
+
+		if !info.IsDir() && !isIgnoreF && !isIgnoreE {
 
-		if !info.IsDir() && !isIgnore {
-			ext = GetFileExtension(info.Name())
 			fileType = GetFileType(ext)
 			ftName = fileType.val(lang)
 

parser.go

@@ -36,29 +36,27 @@ func InitArgs() {
 		Example : -method type
 		`)
 
-	flag.StringVar(&IgnoresFiles, "-ignore-file", "",
-		`
-		Ignores some files while organising directories
+	flag.StringVar(&IgnoresFiles, "ef", "",
+		`Exclude some files while organising directories
 		Example for one file : --ignore-file file.txt
 		Example for mulitple files : --ignore-file "file1.txt, file2.txt, file3.txt"
 	`)
 
-	flag.StringVar(&IgnoresExt, "-ignore-ext", "",
-		`
-		Ignores files with given extensions
+	flag.StringVar(&IgnoresExt, "ex", "",
+		`Excludes files with given extensions
 		Example for one extension : --ignore-ext txt
 		Example for mulitple extensions : --ignore-file "mp3, txt, json"
 	`)
 
 	flag.StringVar(&Archive, "archive", "zip",
-		`
+		`(Not working yet)
 		Archive created folders with specified method
 		Suported methods are zip and tar
 		Example : -archive zip
 	`)
 
 	flag.StringVar(&Compress, "compress", "gzip",
-		`
+		`(Not working yet)
 		Compress created folders with specified algorithm
 		Suported algorithms are : bzip2, flate, gzip, lzw, zlib 
 		Example : -archive gzip

utils.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"bytes"
 	"os"
 	"os/exec"
 	"path/filepath"
@@ -9,7 +10,7 @@ import (
 )
 
 const OS_WINDOWS = "windows"
-const BYTE_PERMS = 664
+const BYTE_PERMS = 0775
 
 // Return extension of a file given his name as parameter
 func GetFileExtension(fileName string) string {
@@ -34,7 +35,7 @@ func Contains(val string, tab []string) bool {
 }
 
 func mv(filePath, destinationPath string) error {
-	oSys := strings.Split(runtime.GOARCH, "/")[0]
+	oSys := strings.Split(runtime.GOOS, "/")[0]
 	cmd := new(exec.Cmd)
 
 	if oSys == OS_WINDOWS {
@@ -43,6 +44,11 @@ func mv(filePath, destinationPath string) error {
 		cmd = exec.Command("mv", filePath, destinationPath)
 	}
 
+	var out bytes.Buffer
+	var stderr bytes.Buffer
+	cmd.Stdout = &out    // standard output
+	cmd.Stderr = &stderr // errors output
+
 	err := cmd.Run()
 
 	return err