Compare commits

...

22 Commits

Author SHA1 Message Date
0ebafa2e99 Prevent archiving already archived files 2025-08-21 13:39:52 +02:00
d95182f187 Add some logs 2025-08-21 13:34:45 +02:00
2274829eda Configure log rotation for cleaner services
They were like 90G big...
2025-08-18 21:48:55 +02:00
526d9a2ff7 set -e 2025-08-07 11:13:49 +02:00
7b82af7006 chore: add deploy helper scripts and move interval to SCAN_INTERVAL env in nssm configs to standardize service setup 2025-08-07 11:13:23 +02:00
c496c5793b chore: add NSSM scripts for directory cleaners and remove old single-service script to standardize deployment across targets 2025-08-07 11:13:23 +02:00
c85208f884 chore: rename Go module to directory-cleaner and add NSSM scripts for
Windows service deployment
2025-08-07 11:00:16 +02:00
14a8858de6 Clean up a bit 2025-08-07 10:56:14 +02:00
9982d36aaf Hallucinate everything
Who knows if its better...
2025-08-07 10:55:47 +02:00
cd6017519e refactor: replace custom ignore logic with doublestar glob matching, improve logging, and update deps to use github.com/bmatcuk/doublestar/v4 while removing unused modules 2025-08-07 10:44:59 +02:00
37b08d27f5 feat: replace std log with cylogger, add flag-based init, enhance logs,
update Go version and deps, and streamline Docker build/deploy scripts
2025-08-07 10:31:28 +02:00
862f95f730 Update 2024-07-16 18:00:12 +02:00
763d8e4b81 Refactor other 2024-06-22 14:22:00 +02:00
4a30a15921 Refactor root scanning 2024-06-22 14:10:37 +02:00
b95d9f7ff0 Refactor thresholds to durations 2024-06-22 14:10:32 +02:00
6b63c3c020 Refactor vars to shorthands 2024-06-22 13:59:53 +02:00
8b89d13997 Refactor main 2024-06-22 13:58:51 +02:00
7c8c92be9a Cleanup 2024-06-22 13:45:20 +02:00
d9f64ac0ad Update dockerfile 2024-06-22 13:45:12 +02:00
dd819547ed Refactor file time handling and add USE_MODTIME flag 2024-02-02 11:32:12 +01:00
c57102011e Refactor file scanning and add file counting 2024-02-02 11:15:05 +01:00
0386c0dceb Update Dockerfile to create a minimal size image 2024-01-31 23:33:38 +01:00
12 changed files with 521 additions and 303 deletions

1
.gitignore vendored Normal file
View File

@@ -0,0 +1 @@
main.exe

View File

@@ -1 +0,0 @@
docker build -t directory-cleaner .

2
deploy.sh Normal file
View File

@@ -0,0 +1,2 @@
docker build -t docker.site.quack-lab.dev/directory-cleaner .
docker push docker.site.quack-lab.dev/directory-cleaner

4
deploy_nssm.sh Normal file
View File

@@ -0,0 +1,4 @@
sh nssm_stopall.sh
sh nssm_rmall.sh
sh nssm_deployall.sh
sh nssm_startall.sh

View File

@@ -1,30 +1,21 @@
# syntax=docker/dockerfile:1
FROM golang:1.22-rc-alpine
# Set destination for COPY
WORKDIR /app
# Download Go modules
COPY go.mod go.sum ./
RUN go mod download
# Copy the source code. Note the slash at the end, as explained in
# https://docs.docker.com/engine/reference/builder/#copy
# May not be enough for complex projects
COPY *.go ./
# Build
# Add GOOS=linux for linux
# GOOS=windows
# GOOS=darwin
RUN go build -o /main
# Optional:
# To bind to a TCP port, runtime parameters must be supplied to the docker command.
# But we can document in the Dockerfile what ports
# the application is going to listen on by default.
# https://docs.docker.com/engine/reference/builder/#expose
# EXPOSE 8080
# Run
FROM golang:1.21.6 as base
WORKDIR $GOPATH/src/app/
COPY . .
RUN go mod download
RUN go mod verify
RUN CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -o /main .
FROM scratch
COPY --from=base /usr/share/zoneinfo /usr/share/zoneinfo
COPY --from=base /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/
COPY --from=base /etc/passwd /etc/passwd
COPY --from=base /etc/group /etc/group
COPY --from=base /main .
CMD ["/main"]

19
go.mod
View File

@@ -1,7 +1,18 @@
module main
module directory-cleaner
go 1.21.6
go 1.23.6
require github.com/djherbis/times v1.6.0
require (
git.site.quack-lab.dev/dave/cylogger v1.3.0
github.com/bmatcuk/doublestar/v4 v4.9.1
github.com/djherbis/times v1.6.0
)
require golang.org/x/sys v0.0.0-20220615213510-4f61da869c0c // indirect
require (
github.com/google/go-cmp v0.5.9 // indirect
github.com/hexops/valast v1.5.0 // indirect
golang.org/x/mod v0.7.0 // indirect
golang.org/x/sys v0.3.0 // indirect
golang.org/x/tools v0.4.0 // indirect
mvdan.cc/gofumpt v0.4.0 // indirect
)

31
go.sum
View File

@@ -1,4 +1,33 @@
git.site.quack-lab.dev/dave/cylogger v1.3.0 h1:eTWPUD+ThVi8kGIsRcE0XDeoH3yFb5miFEODyKUdWJw=
git.site.quack-lab.dev/dave/cylogger v1.3.0/go.mod h1:wctgZplMvroA4X6p8f4B/LaCKtiBcT1Pp+L14kcS8jk=
github.com/bmatcuk/doublestar/v4 v4.9.1 h1:X8jg9rRZmJd4yRy7ZeNDRnM+T3ZfHv15JiBJ/avrEXE=
github.com/bmatcuk/doublestar/v4 v4.9.1/go.mod h1:xBQ8jztBU6kakFMg+8WGxn0c6z1fTSPVIjEY1Wr7jzc=
github.com/djherbis/times v1.6.0 h1:w2ctJ92J8fBvWPxugmXIv7Nz7Q3iDMKNx9v5ocVH20c=
github.com/djherbis/times v1.6.0/go.mod h1:gOHeRAz2h+VJNZ5Gmc/o7iD9k4wW7NMVqieYCY99oc0=
golang.org/x/sys v0.0.0-20220615213510-4f61da869c0c h1:aFV+BgZ4svzjfabn8ERpuB4JI4N6/rdy1iusx77G3oU=
github.com/frankban/quicktest v1.14.3 h1:FJKSZTDHjyhriyC81FLQ0LY93eSai0ZyR/ZIkd3ZUKE=
github.com/frankban/quicktest v1.14.3/go.mod h1:mgiwOwqx65TmIk1wJ6Q7wvnVMocbUorkibMOrVTHZps=
github.com/google/go-cmp v0.5.9 h1:O2Tfq5qg4qc4AmwVlvv0oLiVAGB7enBSJ2x2DqQFi38=
github.com/google/go-cmp v0.5.9/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY=
github.com/hexops/autogold v0.8.1 h1:wvyd/bAJ+Dy+DcE09BoLk6r4Fa5R5W+O+GUzmR985WM=
github.com/hexops/autogold v0.8.1/go.mod h1:97HLDXyG23akzAoRYJh/2OBs3kd80eHyKPvZw0S5ZBY=
github.com/hexops/gotextdiff v1.0.3 h1:gitA9+qJrrTCsiCl7+kh75nPqQt1cx4ZkudSTLoUqJM=
github.com/hexops/gotextdiff v1.0.3/go.mod h1:pSWU5MAI3yDq+fZBTazCSJysOMbxWL1BSow5/V2vxeg=
github.com/hexops/valast v1.5.0 h1:FBTuvVi0wjTngtXJRZXMbkN/Dn6DgsUsBwch2DUJU8Y=
github.com/hexops/valast v1.5.0/go.mod h1:Jcy1pNH7LNraVaAZDLyv21hHg2WBv9Nf9FL6fGxU7o4=
github.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=
github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk=
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
github.com/rogpeppe/go-internal v1.9.0/go.mod h1:WtVeX8xhTBvf0smdhujwtBcq4Qrzq/fJaraNFVN+nFs=
golang.org/x/mod v0.7.0 h1:LapD9S96VoQRhi/GrNTqeBJFrUjs5UHCAtTlgwA5oZA=
golang.org/x/mod v0.7.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
golang.org/x/sync v0.1.0 h1:wsuoTGHzEhffawBOhz5CYhcrV4IdKZbEyZjBMuTp12o=
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
golang.org/x/sys v0.0.0-20220615213510-4f61da869c0c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.3.0 h1:w8ZOecv6NaNa/zC8944JTU3vz4u6Lagfk4RPQxv92NQ=
golang.org/x/sys v0.3.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/tools v0.4.0 h1:7mTAgkunk3fr4GAloyyCasadO6h9zSsQZbwvcaIciV4=
golang.org/x/tools v0.4.0/go.mod h1:UE5sM2OK9E/d67R0ANs2xJizIymRP5gJU295PvKXxjQ=
mvdan.cc/gofumpt v0.4.0 h1:JVf4NN1mIpHogBj7ABpgOyZc65/UUOkKQFkoURsz4MM=
mvdan.cc/gofumpt v0.4.0/go.mod h1:PljLOHDeZqgS8opHRKLzp2It2VBuSdteAgqUfzMTxlQ=

617
main.go
View File

@@ -1,268 +1,349 @@
package main
import (
"log"
"os"
"path/filepath"
"regexp"
"strconv"
"strings"
"time"
"github.com/djherbis/times"
)
var timeUnits = map[string]int64{
"ms": 1,
"s": 1000,
"m": 1000 * 60,
"h": 1000 * 60 * 60,
"d": 1000 * 60 * 60 * 24,
"M": 1000 * 60 * 60 * 24 * 30,
"y": 1000 * 60 * 60 * 24 * 365,
}
var valueRegex, _ = regexp.Compile(`\d+`)
var unitRegex, _ = regexp.Compile(`[a-zA-Z]+`)
func parseDuration(date string) int64 {
var milliseconds int64 = 0
date = strings.TrimSpace(date)
var parts = strings.Split(date, "_")
for _, part := range parts {
part = strings.TrimSpace(part)
log.Printf("Parsing date part: %s\n", part)
var value = valueRegex.FindString(part)
var unit = unitRegex.FindString(part)
if value == "" || unit == "" {
log.Println("Invalid date part: " + part)
continue
}
if _, ok := timeUnits[unit]; !ok {
log.Println("Invalid date unit: " + unit)
continue
}
log.Printf("Part %s parsed as: Value: %s, Unit: %s\n", part, value, unit)
var valueMs, _ = strconv.ParseInt(value, 10, 16)
valueMs = valueMs * timeUnits[unit]
milliseconds += valueMs
log.Printf("Adding %dms to duration, now: %d\n", valueMs, milliseconds)
}
return milliseconds
}
func getEnv(key, def string) string {
var value, exists = os.LookupEnv(key)
if exists {
return value
}
return def
}
func scanRoot() {
log.Println("Scanning root directory...")
filepath.Walk(constants.ROOT, func(path string, info os.FileInfo, err error) error {
if err != nil {
log.Printf("Error scanning %s: %s\n", path, err)
return nil
}
path = filepath.ToSlash(path)
if path == constants.ROOT {
log.Printf("Skipping root directory %s...\n", path)
return nil
}
if info.IsDir() {
log.Printf("Skipping directory %s...\n", path)
return filepath.SkipDir
}
// We hope that IGNORED_DIRECTORIES is a small list, so we can afford to iterate over it
// In fact iteration should be faster for small lists rather than hashing
for _, ignoredDir := range constants.IGNORED_DIRECTORIES {
log.Println(constants.IGNORED_DIRECTORIES, len(constants.IGNORED_DIRECTORIES))
if strings.HasPrefix(path, ignoredDir) {
log.Printf("Ignoring directory %s\n", path)
return filepath.SkipDir
}
}
processFile(path, info)
return nil
})
}
func scanArchive() {
log.Println("Scanning archive...")
filepath.Walk(constants.ROOT_ARCHIVE, func(path string, info os.FileInfo, err error) error {
if err != nil {
log.Printf("Error scanning %s: %s\n", path, err)
return nil
}
path = filepath.ToSlash(path)
if path == constants.ROOT_ARCHIVE {
log.Printf("Skipping root directory %s...\n", path)
return nil
}
processArchiveFile(path, info)
return nil
})
}
func processFile(path string, info os.FileInfo) {
var now = time.Now().UnixMilli()
var fileATime int64 = times.Get(info).AccessTime().UnixMilli()
var accessTimeDelta = now - fileATime
log.Printf("File %s last accessed at %d, %dms ago\n", path, fileATime, accessTimeDelta)
if accessTimeDelta > constants.ARCHIVE_THRESHOLD {
log.Printf("File %s was accessed more than %dms ago, archiving...\n", path, constants.ARCHIVE_THRESHOLD)
archiveFile(path)
}
}
func processArchiveFile(path string, info os.FileInfo) {
var now = time.Now().UnixMilli()
var fileATime int64 = times.Get(info).AccessTime().UnixMilli()
var accessTimeDelta = now - fileATime
log.Printf("File %s last accessed at %d, %dms ago\n", path, fileATime, accessTimeDelta)
if accessTimeDelta > constants.DELETE_THRESHOLD {
log.Printf("File %s was accessed more than %dms ago, deleting...\n", path, constants.DELETE_THRESHOLD)
deleteFile(path)
}
}
func archiveFile(path string) {
// defer os.Exit(1)
var newPath = constants.ROOT_ARCHIVE + strings.Replace(path, constants.ROOT, "", 1)
log.Printf("Archiving file %s to %s...\n", path, newPath)
os.MkdirAll(filepath.Dir(newPath), os.ModePerm)
var err = os.Rename(path, newPath)
if err != nil {
log.Printf("Error archiving file %s: %s\n", path, err)
return
}
}
func deleteFile(path string) {
// defer os.Exit(1)
log.Printf("Deleting file %s...\n", path)
var err = os.Remove(path)
if err != nil {
log.Printf("Error deleting file %s: %s\n", path, err)
return
}
}
func cleanRoot() {
var files, err = os.ReadDir(constants.ROOT)
if err != nil {
log.Printf("Error reading root directory %s: %s\n", constants.ROOT, err)
return
}
for _, file := range files {
if !file.IsDir() {
continue
}
var empty, err = isDirEmpty(constants.ROOT + "/" + file.Name())
if err != nil {
log.Printf("Error checking if directory %s is empty: %s\n", file.Name(), err)
continue
}
log.Printf("Directory %s isempty: %t\n", file.Name(), empty)
if empty {
log.Printf("Deleting empty directory %s\n", file.Name())
var err = os.RemoveAll(constants.ROOT + "/" + file.Name())
if err != nil {
log.Printf("Error deleting empty directory %s: %s\n", file.Name(), err)
}
}
}
}
func isDirEmpty(dirPath string) (bool, error) {
var empty = true
var ferr error = nil
filepath.Walk(dirPath, func(path string, info os.FileInfo, err error) error {
if err != nil {
log.Printf("Error scanning %s: %s\n", path, err)
ferr = err
return nil
}
if !info.IsDir() {
empty = false
log.Printf("Directory %s is not empty, found %s\n", dirPath, path)
return filepath.SkipAll
}
return nil
})
return empty, ferr
}
type Constants struct {
ROOT string
ROOT_ARCHIVE string
IGNORED_DIRECTORIES []string
ARCHIVE_THRESHOLD int64
DELETE_THRESHOLD int64
SCAN_INTERVAL time.Duration
}
var constants = Constants{}
func main() {
log.SetFlags(0b111)
var ROOT = filepath.ToSlash(strings.TrimSpace(getEnv("ROOT", "/tmp")))
var ROOT_ARCHIVE = filepath.ToSlash(strings.TrimSpace(getEnv("ROOT_ARCHIVE", ROOT+"/archive")))
os.Mkdir(ROOT_ARCHIVE, os.ModePerm)
var IGNORED_DIRECTORIES = []string{}
var ignoredEnv = getEnv("IGNORED_DIRECTORIES", "")
if ignoredEnv != "" {
ignoredEnv = strings.TrimSpace(ignoredEnv)
IGNORED_DIRECTORIES = append(IGNORED_DIRECTORIES, strings.Split(ignoredEnv, ",")...)
}
IGNORED_DIRECTORIES = append(IGNORED_DIRECTORIES, ROOT_ARCHIVE)
for key, dir := range IGNORED_DIRECTORIES {
IGNORED_DIRECTORIES[key] = filepath.ToSlash(strings.TrimSpace(dir))
}
var ARCHIVE_THRESHOLD = parseDuration(getEnv("ARCHIVE_THRESHOLD", "1d"))
var DELETE_THRESHOLD = parseDuration(getEnv("DELETE_THRESHOLD", "12h"))
var SCAN_INTERVAL = time.Duration(parseDuration(getEnv("SCAN_INTERVAL", "1m")) * 1e6)
constants.ROOT = ROOT
constants.ROOT_ARCHIVE = ROOT_ARCHIVE
constants.IGNORED_DIRECTORIES = IGNORED_DIRECTORIES
constants.ARCHIVE_THRESHOLD = ARCHIVE_THRESHOLD
constants.DELETE_THRESHOLD = DELETE_THRESHOLD
constants.SCAN_INTERVAL = SCAN_INTERVAL
log.Println("Input args parsed as:")
log.Printf("ROOT: %s\n", ROOT)
log.Printf("ROOT_ARCHIVE: %s\n", ROOT_ARCHIVE)
log.Printf("IGNORED_DIRECTORIES: %s\n", IGNORED_DIRECTORIES)
log.Printf("ARCHIVE_THRESHOLD: %d\n", ARCHIVE_THRESHOLD)
log.Printf("DELETE_THRESHOLD: %d\n", DELETE_THRESHOLD)
log.Printf("SCAN_INTERVAL: %d\n", SCAN_INTERVAL.Milliseconds())
scanRoot()
scanArchive()
cleanRoot()
for {
log.Printf("Running at %d", time.Now().UnixMilli())
time.Sleep(SCAN_INTERVAL)
scanRoot()
scanArchive()
cleanRoot()
}
}
package main
import (
"flag"
"os"
"path/filepath"
"regexp"
"strconv"
"strings"
"time"
logger "git.site.quack-lab.dev/dave/cylogger"
"github.com/bmatcuk/doublestar/v4"
"github.com/djherbis/times"
)
type Constants struct {
ROOT string
ROOT_ARCHIVE string
IGNORED_DIRECTORIES []string
ARCHIVE_THRESHOLD int64 // ms
DELETE_THRESHOLD int64 // ms
SCAN_INTERVAL time.Duration
USE_MODTIME bool
}
var (
constants Constants
timeUnits = map[string]int64{"ms": 1, "s": 1000, "m": 60_000, "h": 3_600_000, "d": 86_400_000, "M": 2_592_000_000, "y": 31_536_000_000}
valueRegex = regexp.MustCompile(`\d+`)
unitRegex = regexp.MustCompile(`[a-zA-Z]+`)
numFilesArchived = 0
numFilesDeleted = 0
)
func parseDurationMS(expr string) int64 {
expr = strings.TrimSpace(expr)
if expr == "" {
return 0
}
var total int64
for _, p := range strings.Split(expr, "_") {
p = strings.TrimSpace(p)
if p == "" {
continue
}
v := valueRegex.FindString(p)
u := unitRegex.FindString(p)
if v == "" || u == "" {
logger.Warning("Invalid duration part: %q", p)
continue
}
unit, ok := timeUnits[u]
if !ok {
logger.Warning("Invalid duration unit: %q", u)
continue
}
n, err := strconv.ParseInt(v, 10, 64)
if err != nil {
logger.Warning("Invalid duration value: %q: %v", v, err)
continue
}
total += n * unit
}
return total
}
func getenv(key, def string) string {
if v, ok := os.LookupEnv(key); ok {
return v
}
return def
}
func loadConfig() Constants {
flag.Parse()
logger.InitFlag()
root := filepath.ToSlash(strings.TrimSpace(getenv("ROOT", "/tmp")))
archive := filepath.ToSlash(strings.TrimSpace(getenv("ROOT_ARCHIVE", root+"/archive")))
_ = os.MkdirAll(archive, os.ModePerm)
// TODO: This just doesn't really work or do what it says it does
// Fix it...
// Right now we don't care because we're not using it
// But we might want to...
ignored := []string{}
if ig := strings.TrimSpace(getenv("IGNORED_DIRECTORIES", "")); ig != "" {
for _, s := range strings.Split(ig, ",") {
if t := strings.TrimSpace(s); t != "" {
ignored = append(ignored, filepath.ToSlash(t))
}
}
}
archiveMs := parseDurationMS(getenv("ARCHIVE_THRESHOLD", "1d"))
deleteMs := parseDurationMS(getenv("DELETE_THRESHOLD", "12h"))
interval := time.Duration(parseDurationMS(getenv("SCAN_INTERVAL", "1m"))) * time.Millisecond
useMod := strings.EqualFold(strings.TrimSpace(getenv("USE_MODTIME", "false")), "true")
logger.Info("Config:")
logger.Info(" ROOT: %s", root)
logger.Info(" ROOT_ARCHIVE: %s", archive)
logger.Info(" IGNORED_DIRECTORIES: %v", ignored)
logger.Info(" ARCHIVE_THRESHOLD(ms): %d", archiveMs)
logger.Info(" DELETE_THRESHOLD(ms): %d", deleteMs)
logger.Info(" SCAN_INTERVAL(ms): %d", interval.Milliseconds())
logger.Info(" USE_MODTIME: %t", useMod)
return Constants{
ROOT: root,
ROOT_ARCHIVE: archive,
IGNORED_DIRECTORIES: ignored,
ARCHIVE_THRESHOLD: archiveMs,
DELETE_THRESHOLD: deleteMs,
SCAN_INTERVAL: interval,
USE_MODTIME: useMod,
}
}
func shouldIgnore(path string) bool {
log := logger.Default.WithPrefix("shouldIgnore").WithPrefix(path)
// Match against slash-normalized full path
path = filepath.ToSlash(path)
for _, pat := range constants.IGNORED_DIRECTORIES {
patLog := log.WithPrefix(pat)
ok, err := doublestar.Match(pat, path)
if err != nil {
patLog.Warning("Ignore pattern error %q vs %q: %v", pat, path, err)
continue
}
if ok {
patLog.Debug("ignore due to doublestar %q", pat)
return true
}
}
return false
}
func fileTime(info os.FileInfo) int64 {
t := times.Get(info)
if constants.USE_MODTIME {
return t.ModTime().UnixMilli()
}
// If AccessTime is not supported, it returns ModTime; times handles that internally.
return t.AccessTime().UnixMilli()
}
func archiveCandidate(path string, info os.FileInfo) {
log := logger.Default.WithPrefix("archiveCandidate")
if info.IsDir() {
return
}
now := time.Now().UnixMilli()
ft := fileTime(info)
log.Debug("now %s filetime %s", time.UnixMilli(now).Format(time.RFC3339), time.UnixMilli(ft).Format(time.RFC3339))
if now-ft <= constants.ARCHIVE_THRESHOLD {
log.Debug("skip")
return
}
rel, err := filepath.Rel(constants.ROOT, path)
if err != nil {
log.Warning("rel ROOT->%s: %v", path, err)
return
}
if strings.HasPrefix(rel, "archive") {
log.Debug("refusing to archive a file that's already archived")
return
}
log.Debug("rel %q", rel)
dst := filepath.Join(constants.ROOT_ARCHIVE, rel)
log.Debug("dst %q", dst)
if err := os.MkdirAll(filepath.Dir(dst), os.ModePerm); err != nil {
log.Error("mkdir %s: %v", filepath.Dir(dst), err)
return
}
if err := os.Rename(path, dst); err != nil {
log.Error("archive %s -> %s: %v", path, dst, err)
return
}
numFilesArchived++
log.Info("Archived: %s -> %s", path, dst)
}
func deleteCandidate(path string, info os.FileInfo) {
log := logger.Default.WithPrefix("deleteCandidate").WithPrefix(path)
if info.IsDir() {
return
}
now := time.Now().UnixMilli()
ft := fileTime(info)
log.Debug("now %s filetime %s", time.UnixMilli(now).Format(time.RFC3339), time.UnixMilli(ft).Format(time.RFC3339))
if now-ft <= constants.DELETE_THRESHOLD {
log.Debug("skip")
return
}
log.Debug("delete")
if err := os.Remove(path); err != nil {
log.Error("delete %s: %v", path, err)
return
}
numFilesDeleted++
log.Info("Deleted: %s", path)
}
func scanRoot() {
log := logger.Default.WithPrefix("scanRoot")
root := constants.ROOT
// doublestar.Glob with os.DirFS(root) returns relative paths
log.Debug("glob %s", root)
paths, err := doublestar.Glob(os.DirFS(root), "**")
if err != nil {
log.Error("glob %s: %v", root, err)
return
}
log.Info("Found %d files in root", len(paths))
for _, rel := range paths {
pathLog := log.WithPrefix(rel)
full := filepath.Join(root, rel)
pathLog.Debug("full %s", full)
if shouldIgnore(full) {
pathLog.Debug("ignore")
continue
}
info, err := os.Stat(full)
pathLog.Trace("stat %+v", info)
if err != nil {
log.Warning("stat %s: %v", full, err)
continue
}
archiveCandidate(full, info)
}
}
func scanArchive() {
log := logger.Default.WithPrefix("scanArchive")
base := constants.ROOT_ARCHIVE
paths, err := doublestar.Glob(os.DirFS(base), "**")
if err != nil {
log.Error("glob %s: %v", base, err)
return
}
for _, rel := range paths {
full := filepath.Join(base, rel)
info, err := os.Stat(full)
if err != nil {
log.Warning("stat %s: %v", full, err)
continue
}
deleteCandidate(full, info)
}
}
func cleanEmptyDirs() {
log := logger.Default.WithPrefix("cleanEmptyDirs")
root := constants.ROOT
paths, err := doublestar.Glob(os.DirFS(root), "**")
if err != nil {
log.Error("glob %s: %v", root, err)
return
}
// Remove empty dirs; iterate deepest first by sorting longer paths first
// Build a slice of directory paths
var dirs []string
for _, rel := range paths {
full := filepath.Join(root, rel)
info, err := os.Stat(full)
if err != nil || !info.IsDir() {
continue
}
if shouldIgnore(full) {
continue
}
dirs = append(dirs, full)
}
// Simple length-based reverse to try delete children before parents
for i := 0; i < len(dirs)-1; i++ {
for j := i + 1; j < len(dirs); j++ {
if len(dirs[i]) < len(dirs[j]) {
dirs[i], dirs[j] = dirs[j], dirs[i]
}
}
}
for _, d := range dirs {
empty, err := isDirEmpty(d)
if err != nil {
log.Warning("check empty %s: %v", d, err)
continue
}
if empty {
if err := os.Remove(d); err == nil {
log.Info("Removed empty dir: %s", d)
} else {
log.Warning("remove %s: %v", d, err)
}
}
}
}
func isDirEmpty(dir string) (bool, error) {
f, err := os.Open(dir)
if err != nil {
return false, err
}
defer f.Close()
// Read at most one entry; if none, it's empty
_, err = f.Readdirnames(1)
if err == os.ErrNotExist {
// For older Go versions, fall back
return true, nil
}
if err != nil {
// io.EOF indicates empty for Readdirnames
return true, nil
}
return false, nil
}
func doRun() {
scanRoot()
scanArchive()
cleanEmptyDirs()
logger.Info("Cycle summary: archived=%d deleted=%d", numFilesArchived, numFilesDeleted)
numFilesArchived, numFilesDeleted = 0, 0
}
func main() {
flag.Parse()
logger.InitFlag()
logger.Info("Starting directory cleaner")
constants = loadConfig()
logger.Info("Ready. First scan in %s", constants.SCAN_INTERVAL)
// Run immediately, then on interval
doRun()
t := time.NewTicker(constants.SCAN_INTERVAL)
defer t.Stop()
for range t.C {
logger.Info("Tick %d", time.Now().UnixMilli())
doRun()
}
}

70
nssm_deployall.sh Normal file
View File

@@ -0,0 +1,70 @@
#!/bin/bash
set -e
# tmp-cleaner service
nssm install DirectoryCleanerTmp C:\\Users\\Administrator\\go\\bin\\directory-cleaner.exe
nssm set DirectoryCleanerTmp AppEnvironmentExtra ARCHIVE_THRESHOLD=12h DELETE_THRESHOLD=2d SCAN_INTERVAL=20s
nssm set DirectoryCleanerTmp AppDirectory C:\\tmp
nssm set DirectoryCleanerTmp AppExit Default Restart
nssm set DirectoryCleanerTmp AppStdout C:\\tmp\\cleaner.log
nssm set DirectoryCleanerTmp AppStdoutCreationDisposition 2
nssm set DirectoryCleanerTmp AppStderr C:\\tmp\\cleaner.log
nssm set DirectoryCleanerTmp AppStderrCreationDisposition 2
nssm set DirectoryCleanerTmp AppRotateFiles 1
nssm set DirectoryCleanerTmp AppRotateOnline 1
nssm set DirectoryCleanerTmp AppRotateBytes 104857600
nssm set DirectoryCleanerTmp DisplayName DirectoryCleanerTmp
nssm set DirectoryCleanerTmp ObjectName LocalSystem
nssm set DirectoryCleanerTmp Start SERVICE_AUTO_START
nssm set DirectoryCleanerTmp Type SERVICE_WIN32_OWN_PROCESS
# tmp-cleaner-e service
nssm install DirectoryCleanerTmpE C:\\Users\\Administrator\\go\\bin\\directory-cleaner.exe
nssm set DirectoryCleanerTmpE AppEnvironmentExtra ARCHIVE_THRESHOLD=1d DELETE_THRESHOLD=2d SCAN_INTERVAL=20s
nssm set DirectoryCleanerTmpE AppDirectory E:\\tmp
nssm set DirectoryCleanerTmpE AppExit Default Restart
nssm set DirectoryCleanerTmpE AppStdout E:\\tmp\\cleaner.log
nssm set DirectoryCleanerTmpE AppStdoutCreationDisposition 2
nssm set DirectoryCleanerTmpE AppStderr E:\\tmp\\cleaner.log
nssm set DirectoryCleanerTmpE AppStderrCreationDisposition 2
nssm set DirectoryCleanerTmpE AppRotateFiles 1
nssm set DirectoryCleanerTmpE AppRotateOnline 1
nssm set DirectoryCleanerTmpE AppRotateBytes 104857600
nssm set DirectoryCleanerTmpE DisplayName DirectoryCleanerTmpE
nssm set DirectoryCleanerTmpE ObjectName LocalSystem
nssm set DirectoryCleanerTmpE Start SERVICE_AUTO_START
nssm set DirectoryCleanerTmpE Type SERVICE_WIN32_OWN_PROCESS
# downloads-cleaner service
nssm install DirectoryCleanerDownloads C:\\Users\\Administrator\\go\\bin\\directory-cleaner.exe
nssm set DirectoryCleanerDownloads AppEnvironmentExtra ARCHIVE_THRESHOLD=6h DELETE_THRESHOLD=3d SCAN_INTERVAL=20s
nssm set DirectoryCleanerDownloads AppDirectory "C:\\Users\\Administrator\\Downloads"
nssm set DirectoryCleanerDownloads AppExit Default Restart
nssm set DirectoryCleanerDownloads AppStdout "C:\\Users\\Administrator\\Downloads\\cleaner.log"
nssm set DirectoryCleanerDownloads AppStdoutCreationDisposition 2
nssm set DirectoryCleanerDownloads AppStderr "C:\\Users\\Administrator\\Downloads\\cleaner.log"
nssm set DirectoryCleanerDownloads AppStderrCreationDisposition 2
nssm set DirectoryCleanerDownloads AppRotateFiles 1
nssm set DirectoryCleanerDownloads AppRotateOnline 1
nssm set DirectoryCleanerDownloads AppRotateBytes 104857600
nssm set DirectoryCleanerDownloads DisplayName DirectoryCleanerDownloads
nssm set DirectoryCleanerDownloads ObjectName LocalSystem
nssm set DirectoryCleanerDownloads Start SERVICE_AUTO_START
nssm set DirectoryCleanerDownloads Type SERVICE_WIN32_OWN_PROCESS
# temp-cleaner service
nssm install DirectoryCleanerTemp C:\\Users\\Administrator\\go\\bin\\directory-cleaner.exe
nssm set DirectoryCleanerTemp AppEnvironmentExtra ARCHIVE_THRESHOLD=6h DELETE_THRESHOLD=1d SCAN_INTERVAL=20s
nssm set DirectoryCleanerTemp AppDirectory "C:\\Users\\Administrator\\AppData\\Local\\Temp"
nssm set DirectoryCleanerTemp AppExit Default Restart
nssm set DirectoryCleanerTemp AppStdout "C:\\Users\\Administrator\\AppData\\Local\\Temp\\cleaner.log"
nssm set DirectoryCleanerTemp AppStdoutCreationDisposition 2
nssm set DirectoryCleanerTemp AppStderr "C:\\Users\\Administrator\\AppData\\Local\\Temp\\cleaner.log"
nssm set DirectoryCleanerTemp AppStderrCreationDisposition 2
nssm set DirectoryCleanerTemp AppRotateFiles 1
nssm set DirectoryCleanerTemp AppRotateOnline 1
nssm set DirectoryCleanerTemp AppRotateBytes 104857600
nssm set DirectoryCleanerTemp DisplayName DirectoryCleanerTemp
nssm set DirectoryCleanerTemp ObjectName LocalSystem
nssm set DirectoryCleanerTemp Start SERVICE_AUTO_START
nssm set DirectoryCleanerTemp Type SERVICE_WIN32_OWN_PROCESS

10
nssm_rmall.sh Normal file
View File

@@ -0,0 +1,10 @@
#!/bin/bash
set -e
# Remove all services
services=$(nssm list | grep "DirectoryCleaner" | awk '{print $1}')
for service in $services; do
echo "Removing $service"
nssm remove "$service" confirm
done

10
nssm_startall.sh Normal file
View File

@@ -0,0 +1,10 @@
#!/bin/bash
set -e
# Start all services
services=$(nssm list | grep "DirectoryCleaner" | awk '{print $1}')
for service in $services; do
echo "Starting $service"
nssm start "$service"
done

10
nssm_stopall.sh Normal file
View File

@@ -0,0 +1,10 @@
#!/bin/bash
set -e
# Stop all services
services=$(nssm list | grep "DirectoryCleaner" | awk '{print $1}')
for service in $services; do
echo "Stopping $service"
nssm stop "$service"
done