feat: filesize

README.md

@@ -16,7 +16,7 @@ tmpbin:
   treshday: 30 # files not modified for more than this long will be moved into tmpbin, 30 days as default
   deleteday: 30 # files in tmpbin for more than this long will be deleted, 30 days as default
 ```
-4. Set your rules in `rules.yml`: (Currently support 10 rules atmost)
+4. Set your rules in `rules.yml`: (Currently support 10 rules utmost)
 ```yml
 rule1:
   enable: true
@@ -25,6 +25,9 @@ rule1:
   - ".*?.doc"
   - ".*?.docx"
   - ".*?.pdf"
+  threshday: 7
+  maxsize: # MB
+  minsize: # MB
 rule2:
 ...
 ```

rules.yml

@@ -5,4 +5,16 @@ rule1:
   - ".*?.doc"
   - ".*?.docx"
   - ".*?.pdf"
-  threshday: 30
+  threshday: 7
+  maxsize:
+  minsize:
+rule2:
+  enable: true
+  name: films
+  regex:
+  - ".*?.mp4"
+  - ".*?.avi"
+  - ".*?.flv"
+  threshday: 7
+  maxsize:
+  minsize: 300

utils/classify.go

@@ -77,11 +77,7 @@ func doRule(rule *Rule, path string, isVerbose bool) {
 	}
 
 	// Moving files to rule dir
-	for count, file := range dir {
-		if count > 10 {
-			break
-		}
-
+	for _, file := range dir {
 		// jump rule dir
 		if file.Name() == rule.Name || file.IsDir() {
 			continue
@@ -91,19 +87,21 @@ func doRule(rule *Rule, path string, isVerbose bool) {
 			re := regexp.MustCompile(pattern)
 			match := re.MatchString(file.Name())
 			if match {
-				modTime, strerr := GetFileModTime(path + rule.Name + file.Name())
+				modTime, strerr := GetFileModTime(path + file.Name())
 				if strerr == "" {
 					if isVerbose {
-						fmt.Printf("%c[0;34m%s%c[0m %c[0;32m%s%c[0m\n", 0x1B, file.Name(), 0x1B, 0x1B, modTime, 0x1B)
+						fmt.Printf("%c[0;34m%s%c[0m %c[0;32m%s%c[0m %d\n", 0x1B, file.Name(), 0x1B, 0x1B, modTime, 0x1B, file.Size())
 					}
 					// If file reaches deleteday
 					if time.Now().Unix()-modTime.Unix() >= int64(rule.Thresh*86400) {
-						if isVerbose {
-							fmt.Printf("%c[0;34m%s%c[0m matches %c[0;33m%s%c[0m\n", 0x1B, file.Name(), 0x1B, 0x1B, rule.Name, 0x1B)
+						if (rule.Maxsize <= 0 || file.Size() < (int64)(rule.Maxsize)*1024*1024) && file.Size() > (int64)(rule.Minsize)*1024*1024 {
+							if isVerbose {
+								fmt.Printf("%c[0;34m%s%c[0m matches %c[0;33m%s%c[0m\n", 0x1B, file.Name(), 0x1B, 0x1B, rule.Name, 0x1B)
+							}
+							src := path + file.Name()
+							des := path + rule.Name + file.Name()
+							MoveAll(file, src, des)
 						}
-						src := path + file.Name()
-						des := path + rule.Name + file.Name()
-						MoveAll(file, src, des)
 					}
 				} else {
 					fmt.Printf("Error while scanning %c[0;34m%s%c[0m :", 0x1B, file.Name(), 0x1B)

utils/tmpbin.go

@@ -61,10 +61,7 @@ func Manage(conf *Conf) {
 	if dir == nil {
 		return
 	}
-	for count, file := range dir {
-		if count > 10 {
-			break
-		}
+	for _, file := range dir {
 		modTime, strerr := GetFileModTime(conf.Targetdir + conf.Tmpbin.Name + file.Name())
 		if strerr == "" {
 			if conf.Verbose {
@@ -77,7 +74,12 @@ func Manage(conf *Conf) {
 				fmt.Printf("Deleting %c[0;34m%s%c[0m\n", 0x1B, file.Name(), 0x1B)
 				//}
 				src := conf.Targetdir + conf.Tmpbin.Name + file.Name()
-				err = os.Remove(src)
+				if file.IsDir() {
+					err = os.RemoveAll(src)
+				} else {
+					err = os.Remove(src)
+				}
+
 				if err != nil {
 					fmt.Printf("Error while deleting %c[0;34m%s%c[0m :", 0x1B, file.Name(), 0x1B)
 					fmt.Printf("\t%c[0;31m%s%c[0m\n", 0x1B, err, 0x1B)

utils/yaml.go

@@ -15,10 +15,12 @@ type Conf struct {
 
 // Rule : a template struct for a rule
 type Rule struct {
-	Enable bool     `yaml:"enable"`
-	Name   string   `yaml:"name"`
-	Regex  []string `yaml:"regex"`
-	Thresh int      `yaml:"threshday"`
+	Enable  bool     `yaml:"enable"`
+	Name    string   `yaml:"name"`
+	Regex   []string `yaml:"regex"`
+	Thresh  int      `yaml:"threshday"`
+	Maxsize int      `yaml:"maxsize"`
+	Minsize int      `yaml:"minsize"`
 }
 
 // Rules :