sort.go

package main

import (
    "os"
    "log"
    "errors"
    "path/filepath"
    "mime"
    "sync"
    "strings"
    "regexp"
)

var wg sync.WaitGroup

func Sort(path string) error {
    
    
    handle, err := os.Lstat(path)
    
    if !handle.IsDir() {
        log.Print("It's not a directory")
        return errors.New("error")
    }
    
    directories := []string{"Images", "Music", "Videos", "Documents"}
    
    for _, subfolder := range directories {
        os.Mkdir(filepath.Join(path, subfolder), 0777)
    }
    
    dir, err := os.Open(path)
    fi, err := dir.Readdir(100)
    for _, file := range fi {
        wg.Add(1)
        go mapToDir(path, file.Name())
    }
        
    wg.Wait()
    return err
    
}

func SortWithRegexp(path string, pattern string) error {
    reg, err := regexp.Compile(pattern)
    if err != nil {
        log.Println("Bad regex")
    }
    
    log.Println("Function works")
    
    handle, err := os.Lstat(path)
    
    if !handle.IsDir() {
        log.Print("It's not a directory")
        return errors.New("error")
    }
    
    os.Mkdir(filepath.Join(path, "Grouped"), 0777)
    
    dir, err := os.Open(path)
    fi, err:= dir.Readdir(100)
    for _, file := range fi {
        filematch := reg.MatchString(file.Name())
        if filematch == true {
            MoveFile(path, file.Name(), "Grouped")
        }
    }
    return nil
}
    
    
    
    
    

func mapToDir(base, name string) error {
    defer wg.Done()
    
    docFormats := []string{".doc", ".txt", ".pdf", ".djvu", ".odt", ".rtf", ".docx", ".html"}
    
    ext := filepath.Ext(name)
    
    if sliceContains(docFormats, ext) {
        //Move to Documents subfolder
        MoveFile(base, name, "Documents")
    }
    
    
    mediatype := mime.TypeByExtension(ext)
    if strings.Contains(mediatype, "image") {
        MoveFile(base, name, "Images")
    }
    
    if strings.Contains(mediatype, "audio") {
        MoveFile(base, name, "Music")
    }
    
    if strings.Contains(mediatype, "video") {
        MoveFile(base, name, "Videos")
    }
    
    
    
    return nil
}

//Creates if directory doesn't already exist
//Will need this later, when we implement smart subfolders
//func CreateINE(dir string){
//    //Should lock other goroutines
//    _, err := os.Stat(dir)
//    
//    
//    if os.IsNotExist(err) {
//        os.Mkdir(dir, 0777)
//    }
//    
//}

func sliceContains(formats []string, ext string) bool {
    for _, format := range formats {
        if ext == format {
            return true
        }
    }
    return false
}

func MoveFile(base, name, subfolder string) {
    fulldirpath := filepath.Join(base, name)
    fulltargetpath := filepath.Join(base, subfolder, name)
    os.Rename(fulldirpath, fulltargetpath)
}