44 Commits

Author SHA1 Message Date
33b3a3d2b6 Improve globbing to better handle both patterns and non patterns 2025-04-13 19:47:52 +02:00
78536c3e19 Implement doublestar wildcards 2025-04-13 19:28:57 +02:00
3a5a333c62 Add LogSuccess 2025-03-11 21:37:55 +01:00
5a2520e3b1 Fix deleting by not deleting at all 2025-03-10 19:08:07 +01:00
12d71dba1c Fix flag inheritance for wildcard links 2025-03-10 19:02:44 +01:00
d94f8db27a Stop logging W as an L 2025-03-10 18:58:34 +01:00
913a279011 Redo old format
Why did I even remove it?
2025-03-10 18:55:25 +01:00
af956110be Implement undoing 2025-03-10 18:53:41 +01:00
8c5d783d2c More betterify logging 2025-03-10 18:49:50 +01:00
6359705714 Fix up logging a little 2025-03-10 18:36:11 +01:00
62bfb91246 Implement wildcards 2025-03-10 18:17:37 +01:00
083d42a9dd No longer support old format 2025-03-10 18:10:59 +01:00
71ea17122c Clean house 2025-03-10 18:08:23 +01:00
83477d5f18 Maybe fix the directory crawler 2025-02-26 11:17:51 +01:00
125bf78c16 Implement yaml format 2025-02-26 11:15:56 +01:00
edaa699c20 Ignore lines with # 2025-02-02 21:19:47 +01:00
5b2da09eb2 Add empty go.sum 2025-02-02 21:19:44 +01:00
1c23ad0cfd Fix issue with defaulting to reading from file while providing other args 2024-11-10 15:58:52 +01:00
653e883742 Fix issue with incorrectly parsing ~ 2024-10-18 20:19:31 +02:00
41bac18525 Replace ~ only if the path begins with ~ 2024-10-08 14:11:38 +02:00
02106824fd Discover go install 2024-10-03 10:53:16 +02:00
cd42bc1fb8 Fix issue with non absolute paths not being absoluted 2024-09-23 16:16:47 +02:00
41123846d1 Convert all paths to absolute
To fix issue where folders were being linked as files
2024-09-20 18:24:10 +02:00
205f8314d6 Improve filepath resolution for relative paths 2024-09-11 19:43:43 +02:00
290e6fdaba Now force forces creation 2024-09-11 19:37:07 +02:00
d98ecd787a Add colors and refine logs a little 2024-09-11 19:35:33 +02:00
1a6992e2a7 Remove sync 2024-09-11 19:32:02 +02:00
fc59878389 Implement deleting real actual files 2024-09-11 19:32:01 +02:00
ff1af19088 Ensure target exists by creating directories 2024-09-11 18:53:22 +02:00
e6bb1f0c53 Update 2024-08-16 14:57:20 +02:00
e149103d21 Print "hard" property on instruction 2024-08-16 09:46:15 +02:00
ebccc49d34 Make forced hard links overwrite target if it has the same name as source
Or something like that... Maybe...
2024-08-16 09:45:05 +02:00
55dc061c31 Default file to "sync" 2024-08-16 09:39:14 +02:00
2c49b65502 Move hard out of flag args and into command 2024-08-16 09:38:35 +02:00
1f21965288 Update sync file 2024-08-15 22:46:34 +02:00
b8e7bc3576 Add support for hard links 2024-08-15 22:44:03 +02:00
PhatPhuckDave
595a11552c Trim instruction before parse 2024-07-21 13:44:26 +02:00
2a7740d8d7 Refactor reading args 2024-07-03 13:31:12 +02:00
7580ca5399 Enable build for linux 2024-07-03 13:31:03 +02:00
58cce74ce8 Remove escapes from code "blocks" 2024-07-01 21:06:01 +02:00
e022a838ba Solve the race condition when recursively reading files, hopefully 2024-07-01 21:00:38 +02:00
0a627ae9ca Add readme 2024-07-01 20:38:38 +02:00
d72644aec3 Code format
I think? Don't know what changed
2024-07-01 20:30:19 +02:00
eeb8dac3a0 Add insane ramblings 2024-07-01 20:28:49 +02:00
14 changed files with 1224 additions and 507 deletions

7
.gitignore vendored
View File

@@ -1,2 +1,5 @@
*.exe
main.exe *.exe
cln
cln.log
.qodo

89
README.md Normal file
View File

@@ -0,0 +1,89 @@
# synclib
A small Go tool for creating symbolic links
Created out of infuriating difficulty of creating symbolic links on windows
## Instruction Formats
The tool supports two formats for defining symbolic links:
### 1. CSV Format (Legacy)
Simple comma-separated values with the format: `<source>,<destination>[,force][,hard][,delete]`
For example:
```
source_path,target_path
source_path,target_path,true
source_path,target_path,true,true
source_path,target_path,true,true,true
```
Or with named flags:
```
source_path,target_path,force=true,hard=true,delete=true
source_path,target_path,f=true,h=true,d=true
```
### 2. YAML Format (Recommended)
A more readable format using YAML:
```yaml
links:
- source: ~/Documents/config.ini
target: ~/.config/app/config.ini
force: true
- source: ~/Pictures
target: ~/Documents/Pictures
hard: true
force: true
- source: ~/Scripts/script.sh
target: ~/bin/script.sh
delete: true
```
Alternatively, you can use an array directly:
```yaml
- source: ~/Documents/config.ini
target: ~/.config/app/config.ini
force: true
- source: ~/Pictures
target: ~/Documents/Pictures
hard: true
```
## Input Methods
The tool supports input of these instructions through:
- Stdin
- `echo "this,that" | sync`
- Run arguments
- `sync this,that foo,bar "foo 2","C:/bar"`
- Files
- `sync -f <file>` (CSV format)
- `sync -f <file.yaml>` or `sync -f <file.yml>` (YAML format)
- Where the file contains instructions, one per line for CSV or structured YAML
- Directories
- `sync -r <directory>`
- This mode will look for "sync", "sync.yaml", or "sync.yml" files recursively in directories and run their instructions
## Options
- `force: true` - Overwrite an existing symbolic link at the target location
- `hard: true` - Create a hard link instead of a symbolic link
- `delete: true` - Delete a non-symlink file at the target location (implies `force: true`)
## Use case
I have a lot of folders (documents, projects, configurations) backed up via Seafile and to have the software using those folders find them at their usual location I'm creating soft symbolic links from the seafile drive to their original location
It would be problematic to have to redo all (or some part) of these symlinks when reinstalling the OS or having something somewhere explode (say software uninstalled) so I have all the instructions in sync files in individual folders in the seafile drive
Which means I can easily back up my configuration and `sync -r ~/Seafile` to symlink it where it belongs

99
colors.go Normal file
View File

@@ -0,0 +1,99 @@
package main
import (
"fmt"
"math/rand/v2"
)
const (
// Reset
Reset = "\033[0m" // Text Reset
// Regular Colors
Black = "\033[0;30m" // Black
Red = "\033[0;31m" // Red
Green = "\033[0;32m" // Green
Yellow = "\033[0;33m" // Yellow
Blue = "\033[0;34m" // Blue
Purple = "\033[0;35m" // Purple
Cyan = "\033[0;36m" // Cyan
White = "\033[0;37m" // White
// Bold
BBlack = "\033[1;30m" // Black
BRed = "\033[1;31m" // Red
BGreen = "\033[1;32m" // Green
BYellow = "\033[1;33m" // Yellow
BBlue = "\033[1;34m" // Blue
BPurple = "\033[1;35m" // Purple
BCyan = "\033[1;36m" // Cyan
BWhite = "\033[1;37m" // White
// Underline
UBlack = "\033[4;30m" // Black
URed = "\033[4;31m" // Red
UGreen = "\033[4;32m" // Green
UYellow = "\033[4;33m" // Yellow
UBlue = "\033[4;34m" // Blue
UPurple = "\033[4;35m" // Purple
UCyan = "\033[4;36m" // Cyan
UWhite = "\033[4;37m" // White
// Background
On_Black = "\033[40m" // Black
On_Red = "\033[41m" // Red
On_Green = "\033[42m" // Green
On_Yellow = "\033[43m" // Yellow
On_Blue = "\033[44m" // Blue
On_Purple = "\033[45m" // Purple
On_Cyan = "\033[46m" // Cyan
On_White = "\033[47m" // White
// High Intensty
IBlack = "\033[0;90m" // Black
IRed = "\033[0;91m" // Red
IGreen = "\033[0;92m" // Green
IYellow = "\033[0;93m" // Yellow
IBlue = "\033[0;94m" // Blue
IPurple = "\033[0;95m" // Purple
ICyan = "\033[0;96m" // Cyan
IWhite = "\033[0;97m" // White
// Bold High Intensty
BIBlack = "\033[1;90m" // Black
BIRed = "\033[1;91m" // Red
BIGreen = "\033[1;92m" // Green
BIYellow = "\033[1;93m" // Yellow
BIBlue = "\033[1;94m" // Blue
BIPurple = "\033[1;95m" // Purple
BICyan = "\033[1;96m" // Cyan
BIWhite = "\033[1;97m" // White
// High Intensty backgrounds
On_IBlack = "\033[0;100m" // Black
On_IRed = "\033[0;101m" // Red
On_IGreen = "\033[0;102m" // Green
On_IYellow = "\033[0;103m" // Yellow
On_IBlue = "\033[0;104m" // Blue
On_IPurple = "\033[10;95m" // Purple
On_ICyan = "\033[0;106m" // Cyan
On_IWhite = "\033[0;107m" // White
)
// The acceptable range is [16, 231] but here we remove some very dark colors
// That make text unreadable on a dark terminal
// See https://www.hackitu.de/termcolor256/
var colors = []int{22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 57, 62, 63, 64, 65, 67, 68, 69, 70, 71, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 148, 149, 150, 151, 152, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 184, 185, 186, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 226, 227, 228, 229, 230}
var colorsIndex int = -1
var shuffled bool
func GenerateRandomAnsiColor() string {
if !shuffled {
rand.Shuffle(len(colors), func(i int, j int) {
colors[i], colors[j] = colors[j], colors[i]
})
shuffled = true
}
colorsIndex++
return fmt.Sprintf("\033[1;4;38;5;%dm", colors[colorsIndex%len(colors)])
}

View File

@@ -1 +1,2 @@
go build main && cp main.exe "/c/Program Files/Git/usr/bin/cln.exe" GOOS=windows GOARCH=amd64 go build -o main.exe main && cp main.exe "/c/Program Files/Git/usr/bin/cln.exe"
GOOS=linux GOARCH=amd64 go build -o main_linux main

6
go.mod
View File

@@ -1,3 +1,7 @@
module main module cln
go 1.21.7 go 1.21.7
require gopkg.in/yaml.v3 v3.0.1
require github.com/bmatcuk/doublestar/v4 v4.8.1

6
go.sum Normal file
View File

@@ -0,0 +1,6 @@
github.com/bmatcuk/doublestar/v4 v4.8.1 h1:54Bopc5c2cAvhLRAzqOGCYHYyhcDHsFF4wWIR5wKP38=
github.com/bmatcuk/doublestar/v4 v4.8.1/go.mod h1:xBQ8jztBU6kakFMg+8WGxn0c6z1fTSPVIjEY1Wr7jzc=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

View File

@@ -1,132 +1,382 @@
package main package main
import ( import (
"fmt" "fmt"
"log" "os"
"os" "path/filepath"
"regexp" "strings"
"strconv"
"strings" "github.com/bmatcuk/doublestar/v4"
) "gopkg.in/yaml.v3"
)
type LinkInstruction struct {
Source string type LinkInstruction struct {
Target string Source string `yaml:"source"`
Force bool Target string `yaml:"target"`
} Force bool `yaml:"force,omitempty"`
Hard bool `yaml:"hard,omitempty"`
func (instruction *LinkInstruction) String() string { Delete bool `yaml:"delete,omitempty"`
return fmt.Sprintf("%s%s%s%s%s%s%s%s%s", SourceColor, instruction.Source, DefaultColor, deliminer, TargetColor, instruction.Target, DefaultColor, deliminer, strconv.FormatBool(instruction.Force)) }
}
type YAMLConfig struct {
func ParseInstruction(line string) (LinkInstruction, error) { Links []LinkInstruction `yaml:"links"`
parts := strings.Split(line, deliminer) }
instruction := LinkInstruction{}
func (instruction *LinkInstruction) Tidy() {
if len(parts) < 2 { instruction.Source = strings.ReplaceAll(instruction.Source, "\"", "")
return instruction, fmt.Errorf("invalid format - not enough parameters (must have at least source and target)") instruction.Source = strings.ReplaceAll(instruction.Source, "\\", "/")
} instruction.Source = strings.TrimSpace(instruction.Source)
instruction.Source = parts[0] instruction.Target = strings.ReplaceAll(instruction.Target, "\"", "")
instruction.Target = parts[1] instruction.Target = strings.ReplaceAll(instruction.Target, "\\", "/")
instruction.Force = false instruction.Target = strings.TrimSpace(instruction.Target)
if len(parts) > 2 { }
res, _ := regexp.MatchString("^(?i)T|TRUE$", parts[2])
instruction.Force = res func (instruction *LinkInstruction) String() string {
} var flags []string
if instruction.Force {
instruction.Source, _ = ConvertHome(instruction.Source) flags = append(flags, "force=true")
instruction.Target, _ = ConvertHome(instruction.Target) }
if instruction.Hard {
instruction.Source = NormalizePath(instruction.Source) flags = append(flags, "hard=true")
instruction.Target = NormalizePath(instruction.Target) }
if instruction.Delete {
return instruction, nil flags = append(flags, "delete=true")
} }
func (instruction *LinkInstruction) RunSync() error { flagsStr := ""
if !FileExists(instruction.Source) { if len(flags) > 0 {
return fmt.Errorf("instruction source %s%s%s does not exist", SourceColor, instruction.Source, DefaultColor) flagsStr = " [" + strings.Join(flags, ", ") + "]"
} }
if AreSame(instruction.Source, instruction.Target) { return fmt.Sprintf("%s → %s%s",
log.Printf("Source %s%s%s and target %s%s%s are the same, %snothing to do...%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor, PathColor, DefaultColor) FormatSourcePath(instruction.Source),
return nil FormatTargetPath(instruction.Target),
} flagsStr)
}
if FileExists(instruction.Target) {
if instruction.Force { func (instruction *LinkInstruction) Undo() {
isSymlink, err := IsSymlink(instruction.Target) if !FileExists(instruction.Target) {
if err != nil { LogInfo("%s does not exist, skipping", FormatTargetPath(instruction.Target))
return fmt.Errorf("could not determine whether %s%s%s is a sym link or not, stopping; err: %s%+v%s", TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor) return
} }
if isSymlink { isSymlink, err := IsSymlink(instruction.Target)
log.Printf("Removing symlink at %s%s%s", TargetColor, instruction.Target, DefaultColor) if err != nil {
err = os.Remove(instruction.Target) LogError("could not determine whether %s is a sym link or not, stopping; err: %v",
if err != nil { FormatTargetPath(instruction.Target), err)
return fmt.Errorf("failed deleting %s%s%s due to %s%+v%s", TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor) return
} }
} else {
return fmt.Errorf("refusing to delte actual (non symlink) file %s%s%s", TargetColor, instruction.Target, DefaultColor) if isSymlink {
} LogInfo("Removing symlink at %s", FormatTargetPath(instruction.Target))
} else { err = os.Remove(instruction.Target)
return fmt.Errorf("target %s%s%s exists - handle manually or set the 'forced' flag (3rd field)", TargetColor, instruction.Target, DefaultColor) if err != nil {
} LogError("could not remove symlink at %s; err: %v",
} FormatTargetPath(instruction.Target), err)
}
err := os.Symlink(instruction.Source, instruction.Target) LogSuccess("Removed symlink at %s", FormatTargetPath(instruction.Target))
if err != nil { } else {
return fmt.Errorf("failed creating symlink between %s%s%s and %s%s%s with error %s%+v%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor) LogInfo("%s is not a symlink, skipping", FormatTargetPath(instruction.Target))
} }
log.Printf("Created symlink between %s%s%s and %s%s%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor) }
return nil func ParseInstruction(line, workdir string) (LinkInstruction, error) {
} line = strings.TrimSpace(line)
if strings.HasPrefix(line, "#") {
func (instruction *LinkInstruction) RunAsync(status chan (error)) { return LinkInstruction{}, fmt.Errorf("comment line")
defer close(status) }
if !FileExists(instruction.Source) {
status <- fmt.Errorf("instruction source %s%s%s does not exist", SourceColor, instruction.Source, DefaultColor) parts := strings.Split(line, deliminer)
return instruction := LinkInstruction{}
}
if len(parts) < 2 {
if AreSame(instruction.Source, instruction.Target) { return instruction, fmt.Errorf("invalid format - not enough parameters (must have at least source and target)")
status <- fmt.Errorf("source %s%s%s and target %s%s%s are the same, %snothing to do...%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor, PathColor, DefaultColor) }
return
} instruction.Source = strings.TrimSpace(parts[0])
instruction.Target = strings.TrimSpace(parts[1])
if FileExists(instruction.Target) {
if instruction.Force { for i := 2; i < len(parts); i++ {
isSymlink, err := IsSymlink(instruction.Target) flagPart := strings.TrimSpace(parts[i])
if err != nil {
status <- fmt.Errorf("could not determine whether %s%s%s is a sym link or not, stopping; err: %s%+v%s", TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor) // Support for legacy format (backward compatibility)
return if !strings.Contains(flagPart, "=") {
} // Legacy format: positional boolean flags
switch i {
if isSymlink { case 2: // Force flag (3rd position)
log.Printf("Removing symlink at %s%s%s", TargetColor, instruction.Target, DefaultColor) instruction.Force = isTrue(flagPart)
err = os.Remove(instruction.Target) case 3: // Hard flag (4th position)
if err != nil { instruction.Hard = isTrue(flagPart)
status <- fmt.Errorf("failed deleting %s%s%s due to %s%+v%s", TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor) case 4: // Delete flag (5th position)
return instruction.Delete = isTrue(flagPart)
} if instruction.Delete {
} else { instruction.Force = true // Delete implies Force
status <- fmt.Errorf("refusing to delte actual (non symlink) file %s%s%s", TargetColor, instruction.Target, DefaultColor) }
return }
} continue
} else { }
status <- fmt.Errorf("target %s%s%s exists - handle manually or set the 'forced' flag (3rd field)", TargetColor, instruction.Target, DefaultColor)
return // New format: named flags (name=value)
} nameValue := strings.SplitN(flagPart, "=", 2)
} if len(nameValue) != 2 {
// Skip malformed flags
err := os.Symlink(instruction.Source, instruction.Target) continue
if err != nil { }
status <- fmt.Errorf("failed creating symlink between %s%s%s and %s%s%s with error %s%+v%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor, ErrorColor, err, DefaultColor)
return flagName := strings.ToLower(strings.TrimSpace(nameValue[0]))
} flagValue := strings.TrimSpace(nameValue[1])
log.Printf("Created symlink between %s%s%s and %s%s%s", SourceColor, instruction.Source, DefaultColor, TargetColor, instruction.Target, DefaultColor)
switch flagName {
status <- nil case "force", "f":
} instruction.Force = isTrue(flagValue)
case "hard", "h":
instruction.Hard = isTrue(flagValue)
case "delete", "d":
instruction.Delete = isTrue(flagValue)
if instruction.Delete {
instruction.Force = true // Delete implies Force
}
}
}
instruction.Tidy()
instruction.Source, _ = ConvertHome(instruction.Source)
instruction.Target, _ = ConvertHome(instruction.Target)
instruction.Source = NormalizePath(instruction.Source, workdir)
instruction.Target = NormalizePath(instruction.Target, workdir)
return instruction, nil
}
func isTrue(value string) bool {
value = strings.ToLower(strings.TrimSpace(value))
return value == "true" || value == "t" || value == "yes" || value == "y" || value == "1"
}
func (instruction *LinkInstruction) RunAsync(status chan (error)) {
defer close(status)
if undo {
instruction.Undo()
return
}
if !FileExists(instruction.Source) {
status <- fmt.Errorf("instruction source %s does not exist", FormatSourcePath(instruction.Source))
return
}
if !instruction.Force && AreSame(instruction.Source, instruction.Target) {
//status <- fmt.Errorf("source %s and target %s are the same, nothing to do...",
// FormatSourcePath(instruction.Source),
// FormatTargetPath(instruction.Target))
LogInfo("Source %s and target %s are the same, nothing to do...",
FormatSourcePath(instruction.Source),
FormatTargetPath(instruction.Target))
return
}
if FileExists(instruction.Target) {
if instruction.Force {
isSymlink, err := IsSymlink(instruction.Target)
if err != nil {
status <- fmt.Errorf("could not determine whether %s is a sym link or not, stopping; err: %v",
FormatTargetPath(instruction.Target), err)
return
}
if instruction.Hard {
info, err := os.Stat(instruction.Target)
if err != nil {
status <- fmt.Errorf("could not stat %s, stopping; err: %v",
FormatTargetPath(instruction.Target), err)
return
}
if info.Mode().IsRegular() && info.Name() == filepath.Base(instruction.Source) {
LogTarget("Overwriting existing file %s", instruction.Target)
err := os.Remove(instruction.Target)
if err != nil {
status <- fmt.Errorf("could not remove existing file %s; err: %v",
FormatTargetPath(instruction.Target), err)
return
}
}
}
if isSymlink {
LogTarget("Removing symlink at %s", instruction.Target)
err = os.Remove(instruction.Target)
if err != nil {
status <- fmt.Errorf("failed deleting %s due to %v",
FormatTargetPath(instruction.Target), err)
return
}
} else {
if !instruction.Delete {
status <- fmt.Errorf("refusing to delte actual (non symlink) file %s",
FormatTargetPath(instruction.Target))
return
}
LogImportant("Deleting (!!!) %s", instruction.Target)
err = os.RemoveAll(instruction.Target)
if err != nil {
status <- fmt.Errorf("failed deleting %s due to %v",
FormatTargetPath(instruction.Target), err)
return
}
}
} else {
status <- fmt.Errorf("target %s exists - handle manually or set the 'forced' flag (3rd field)",
FormatTargetPath(instruction.Target))
return
}
}
targetDir := filepath.Dir(instruction.Target)
if _, err := os.Stat(targetDir); os.IsNotExist(err) {
err = os.MkdirAll(targetDir, 0755)
if err != nil {
status <- fmt.Errorf("failed creating directory %s due to %v",
FormatTargetPath(targetDir), err)
return
}
}
var err error
if instruction.Hard {
err = os.Link(instruction.Source, instruction.Target)
} else {
err = os.Symlink(instruction.Source, instruction.Target)
}
if err != nil {
status <- fmt.Errorf("failed creating symlink between %s and %s with error %v",
FormatSourcePath(instruction.Source),
FormatTargetPath(instruction.Target),
err)
return
}
LogSuccess("Created symlink between %s and %s",
FormatSourcePath(instruction.Source),
FormatTargetPath(instruction.Target))
status <- nil
}
func ParseYAMLFile(filename, workdir string) ([]LinkInstruction, error) {
data, err := os.ReadFile(filename)
if err != nil {
return nil, fmt.Errorf("error reading YAML file: %w", err)
}
// First try to parse as a list of link instructions
var config YAMLConfig
err = yaml.Unmarshal(data, &config)
if err != nil || len(config.Links) == 0 {
// If that fails, try parsing as a direct list of instructions
var instructions []LinkInstruction
err = yaml.Unmarshal(data, &instructions)
if err != nil {
return nil, fmt.Errorf("error parsing YAML: %w", err)
}
config.Links = instructions
}
expanded := []LinkInstruction{}
for _, link := range config.Links {
LogSource("Expanding pattern source %s in YAML file %s", link.Source, filename)
newlinks, err := ExpandPattern(link.Source, workdir, link.Target)
if err != nil {
return nil, fmt.Errorf("error expanding pattern: %w", err)
}
// "Clone" the original link instruction for each expanded link
for i := range newlinks {
newlinks[i].Delete = link.Delete
newlinks[i].Hard = link.Hard
newlinks[i].Force = link.Force
}
LogInfo("Expanded pattern %s in YAML file %s to %d links",
FormatSourcePath(link.Source), FormatSourcePath(filename), len(newlinks))
expanded = append(expanded, newlinks...)
}
for i := range expanded {
link := &expanded[i]
link.Tidy()
link.Source, _ = ConvertHome(link.Source)
link.Target, _ = ConvertHome(link.Target)
link.Source = NormalizePath(link.Source, workdir)
link.Target = NormalizePath(link.Target, workdir)
// If Delete is true, Force must also be true
if link.Delete {
link.Force = true
}
}
return expanded, nil
}
func ExpandPattern(source, workdir, target string) (links []LinkInstruction, err error) {
static, pattern := doublestar.SplitPattern(source)
if static == "" {
static = workdir
}
LogInfo("Static part: %s", static)
LogInfo("Pattern part: %s", pattern)
files, err := doublestar.Glob(os.DirFS(static), pattern)
if err != nil {
return nil, fmt.Errorf("error expanding pattern: %w", err)
}
targetIsFile := false
if info, err := os.Stat(target); err == nil && !info.IsDir() {
targetIsFile = true
}
for _, file := range files {
if info, err := os.Stat(file); err == nil && info.IsDir() {
// We don't care about matched directories
// We want files within them
if len(files) == 1 {
// Special case: if there is only one file, and it's a directory
// This should only ever happen if our source is a path (and not a glob!)
// And our target is a path, a directory
// ...but it will also happen if the source IS a glob and it happens to match ONE directory
// I think that should happen rarely enough to not be an issue...
links = append(links, LinkInstruction{
Source: filepath.Join(static, file),
Target: target,
})
continue
}
LogInfo("Skipping directory %s", file)
continue
}
var targetPath string
if targetIsFile && len(files) == 1 {
// Special case: target is a file, and glob matches exactly one file.
// Use target directly (don't append filename).
targetPath = target
} else {
// Default: append filename to target dir.
targetPath = filepath.Join(target, file)
}
links = append(links, LinkInstruction{
Source: filepath.Join(static, file),
Target: targetPath,
})
}
LogInfo("Expanded pattern %s to %d links", FormatSourcePath(source), len(links))
return
}
func IsYAMLFile(filename string) bool {
ext := strings.ToLower(filepath.Ext(filename))
return ext == ".yaml" || ext == ".yml"
}

99
logger.go Normal file
View File

@@ -0,0 +1,99 @@
package main
import (
"fmt"
"log"
)
// Message type prefixes
const (
InfoPrefix = "INFO"
ErrorPrefix = "ERROR"
WarningPrefix = "WARN"
SourcePrefix = "SOURCE"
TargetPrefix = "TARGET"
PathPrefix = "PATH"
ImportantPrefix = "IMPORTANT"
SuccessPrefix = "DONE"
)
// LogInfo logs an informational message
func LogInfo(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s[%s]%s %s", BGreen, InfoPrefix, Reset, message)
}
// LogSuccess logs a success message
func LogSuccess(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s%s[%s]%s %s", BBlue, On_Blue, SuccessPrefix, Reset, message)
}
// LogSource logs a message about a source file/path with proper coloring
func LogSource(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s[%s]%s %s%s%s", BPurple, SourcePrefix, Reset, SourceColor, message, Reset)
}
// LogTarget logs a message about a target file/path with proper coloring
func LogTarget(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s[%s]%s %s%s%s", BYellow, TargetPrefix, Reset, TargetColor, message, Reset)
}
// LogPath logs a message about a path with proper coloring
func LogPath(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s[%s]%s %s%s%s", BGreen, PathPrefix, Reset, PathColor, message, Reset)
}
// LogImportant logs a message that needs attention with proper coloring
func LogImportant(format string, args ...interface{}) {
message := fmt.Sprintf(format, args...)
log.Printf("%s[%s]%s %s%s%s", BRed, ImportantPrefix, Reset, ImportantColor, message, Reset)
}
// LogError logs an error message with proper coloring that won't be cut off
func LogError(format string, args ...interface{}) {
// Format the message first with normal text (no red coloring)
message := fmt.Sprintf(format, args...)
// The Error prefix itself is bold red on a light background for maximum visibility
prefix := fmt.Sprintf("%s%s[%s]%s ", BRed, On_White, ErrorPrefix, Reset)
// The message is in default color (no red), only the [ERROR] prefix is colored
log.Printf("%s%s", prefix, message)
}
// FormatSourcePath returns a source path with proper coloring
func FormatSourcePath(path string) string {
return fmt.Sprintf("%s%s%s", SourceColor, path, Reset)
}
// FormatTargetPath returns a target path with proper coloring
func FormatTargetPath(path string) string {
return fmt.Sprintf("%s%s%s", TargetColor, path, Reset)
}
// FormatPathValue returns a path with proper coloring
func FormatPathValue(path string) string {
return fmt.Sprintf("%s%s%s", PathColor, path, Reset)
}
// FormatErrorValue returns an error value without any additional formatting
// Since error messages are no longer red, we don't need special formatting
func FormatErrorValue(err error) string {
if err == nil {
return ""
}
// Just return the error string with no color formatting
return fmt.Sprintf("%v", err)
}
// FormatErrorMessage formats an error message with no additional color formatting,
// while preserving the special formatting of embedded source/target/path elements.
func FormatErrorMessage(format string, args ...interface{}) string {
// This just formats the message with no additional color formatting
// The path formatting will still be preserved
return fmt.Sprintf(format, args...)
}

530
main.go
View File

@@ -1,238 +1,292 @@
package main package main
import ( import (
"bufio" "bufio"
"flag" "flag"
"io" "io"
"log" "log"
"os" "os"
"regexp" "path/filepath"
"sync" "regexp"
"sync/atomic" "sync"
) "sync/atomic"
)
const deliminer = ","
const ( const deliminer = ","
Black = "\033[30m" const SourceColor = Purple
Red = "\033[31m" const TargetColor = Yellow
Green = "\033[32m" const ErrorColor = Red
Yellow = "\033[33m" const ImportantColor = BRed
Blue = "\033[34m" const DefaultColor = Reset
Magenta = "\033[35m" const PathColor = Green
Cyan = "\033[36m"
White = "\033[37m" var FileRegex, _ = regexp.Compile(`sync\.ya?ml$`)
) var programName = os.Args[0]
const SourceColor = Magenta var undo = false
const TargetColor = Yellow
const ErrorColor = Red func main() {
const DefaultColor = White recurse := flag.String("r", "", "recurse into directories")
const PathColor = Green file := flag.String("f", "", "file to read instructions from")
debug := flag.Bool("d", false, "debug")
var DirRegex, _ = regexp.Compile(`^(.+?)[/\\]sync$`) undoF := flag.Bool("u", false, "undo")
var FileRegex, _ = regexp.Compile(`^sync$`) flag.Parse()
var programName = os.Args[0] undo = *undoF
func main() { if *debug {
recurse := flag.String("r", "", "recurse into directories") log.SetFlags(log.Lmicroseconds | log.Lshortfile)
file := flag.String("f", "", "file to read instructions from") logFile, err := os.Create(programName + ".log")
debug := flag.Bool("d", false, "debug") if err != nil {
flag.Parse() LogError("Error creating log file: %v", err)
os.Exit(1)
if *debug { }
log.SetFlags(log.Lmicroseconds | log.Lshortfile) logger := io.MultiWriter(os.Stdout, logFile)
logFile, err := os.Create("main.log") log.SetOutput(logger)
if err != nil { } else {
log.Printf("Error creating log file: %v", err) log.SetFlags(log.Lmicroseconds)
os.Exit(1) }
}
logger := io.MultiWriter(os.Stdout, logFile) instructions := make(chan *LinkInstruction, 1000)
log.SetOutput(logger) status := make(chan error)
} else {
log.SetFlags(log.Lmicroseconds) // Check input sources in priority order
} switch {
case *recurse != "":
log.Printf("Recurse: %s", *recurse) LogInfo("Recurse: %s", *recurse)
log.Printf("File: %s", *file) go ReadFromFilesRecursively(*recurse, instructions, status)
instructions := make(chan LinkInstruction, 1000) case *file != "":
status := make(chan error) LogInfo("File: %s", *file)
if *recurse != "" { go ReadFromFile(*file, instructions, status, true)
go ReadFromFilesRecursively(*recurse, instructions, status)
} else if *file != "" { case len(flag.Args()) > 0:
go ReadFromFile(*file, instructions, status, true) LogInfo("Reading from command line arguments")
} else if len(os.Args) > 1 { go ReadFromArgs(instructions, status)
go ReadFromArgs(instructions, status)
} else { // case IsPipeInput():
go ReadFromStdin(instructions, status) // LogInfo("Reading from stdin pipe")
} // go ReadFromStdin(instructions, status)
go func() { default:
for { if _, err := os.Stat("sync"); err == nil {
err, ok := <-status LogInfo("Using default sync file")
if !ok { go ReadFromFile("sync", instructions, status, true)
break } else if _, err := os.Stat("sync.yaml"); err == nil {
} LogInfo("Using default sync.yaml file")
if err != nil { go ReadFromFile("sync.yaml", instructions, status, true)
log.Println(err) } else if _, err := os.Stat("sync.yml"); err == nil {
} LogInfo("Using default sync.yml file")
} go ReadFromFile("sync.yml", instructions, status, true)
}() } else {
LogInfo("No input provided")
var instructionsDone int32 LogInfo("Provide input as: ")
var wg sync.WaitGroup LogInfo("Arguments - %s <source>,<target>,<force?>", programName)
for { LogInfo("File - %s -f <file>", programName)
instruction, ok := <-instructions LogInfo("YAML File - %s -f <file.yaml>", programName)
if !ok { LogInfo("Folder (finding sync files in folder recursively) - %s -r <folder>", programName)
log.Printf("No more instructions to process") LogInfo("stdin - (cat <file> | %s)", programName)
break os.Exit(1)
} }
log.Printf("Processing: %s", instruction.String()) }
status := make(chan error)
go instruction.RunAsync(status) go func() {
wg.Add(1) for {
err := <-status err, ok := <-status
if err != nil { if !ok {
log.Printf("Failed parsing instruction %s%s%s due to %s%+v%s", SourceColor, instruction.String(), DefaultColor, ErrorColor, err, DefaultColor) break
} }
atomic.AddInt32(&instructionsDone, 1) if err != nil {
wg.Done() LogError("%v", err)
} }
wg.Wait() }
log.Println("All done") }()
if instructionsDone == 0 {
log.Printf("No input provided") var instructionsDone int32 = 0
log.Printf("Provide input as: ") var wg sync.WaitGroup
log.Printf("Arguments - %s <source>,<target>,<force?>", programName) for {
log.Printf("File - %s -f <file>", programName) instruction, ok := <-instructions
log.Printf("Folder (finding sync files in folder recursively) - %s -r <folder>", programName) if !ok {
log.Printf("stdin - (cat <file> | %s)", programName) LogInfo("No more instructions to process")
os.Exit(1) break
} }
} LogInfo("Processing: %s", instruction.String())
status := make(chan error)
func ReadFromFilesRecursively(input string, output chan LinkInstruction, status chan error) { go instruction.RunAsync(status)
defer close(output) wg.Add(1)
defer close(status) err := <-status
if err != nil {
input = NormalizePath(input) LogError("Failed processing instruction: %v", err)
log.Printf("Reading input from files recursively starting in %s%s%s", PathColor, input, DefaultColor) }
atomic.AddInt32(&instructionsDone, 1)
files := make(chan string, 128) wg.Done()
recurseStatus := make(chan error) }
go GetSyncFilesRecursively(input, files, recurseStatus) wg.Wait()
go func() { if instructionsDone == 0 {
for { LogInfo("No instructions were processed")
err, ok := <-recurseStatus os.Exit(1)
if !ok { }
break LogInfo("All done")
} }
if err != nil {
log.Printf("Failed to get sync files recursively: %s%+v%s", ErrorColor, err, DefaultColor) func IsPipeInput() bool {
status <- err info, err := os.Stdin.Stat()
} if err != nil {
} return false
}() }
return info.Mode()&os.ModeNamedPipe != 0
var wg sync.WaitGroup }
for {
file, ok := <-files func ReadFromFilesRecursively(input string, output chan *LinkInstruction, status chan error) {
if !ok { defer close(output)
log.Printf("No more files to process") defer close(status)
break
} workdir, _ := os.Getwd()
wg.Add(1) input = NormalizePath(input, workdir)
go func() { LogInfo("Reading input from files recursively starting in %s", FormatPathValue(input))
defer wg.Done()
log.Println(file) files := make(chan string, 128)
file = NormalizePath(file) fileStatus := make(chan error)
log.Printf("Processing file: %s%s%s", PathColor, file, DefaultColor) var wg sync.WaitGroup
go GetSyncFilesRecursively(input, files, fileStatus)
// This "has" to be done because instructions are resolved in relation to cwd go func() {
fileDir := DirRegex.FindStringSubmatch(file) wg.Wait()
if fileDir == nil { close(files)
log.Printf("Failed to extract directory from %s%s%s", SourceColor, file, DefaultColor) }()
return
} go func() {
log.Printf("Changing directory to %s%s%s (for %s%s%s)", PathColor, fileDir[1], DefaultColor, PathColor, file, DefaultColor) for {
err := os.Chdir(fileDir[1]) err, ok := <-fileStatus
if err != nil { if !ok {
log.Printf("Failed to change directory to %s%s%s: %s%+v%s", SourceColor, fileDir[1], DefaultColor, ErrorColor, err, DefaultColor) break
return }
} if err != nil {
LogError("Failed to get sync files recursively: %v", err)
ReadFromFile(file, output, status, false) status <- err
}() }
} }
wg.Wait() }()
}
func ReadFromFile(input string, output chan LinkInstruction, status chan error, doclose bool) { for {
if doclose { file, ok := <-files
defer close(output) if !ok {
defer close(status) LogInfo("No more files to process")
} break
}
input = NormalizePath(input) wg.Add(1)
log.Printf("Reading input from file: %s%s%s", PathColor, input, DefaultColor) go func() {
file, err := os.Open(input) defer wg.Done()
if err != nil { LogInfo(file)
log.Fatalf("Failed to open file %s%s%s: %s%+v%s", SourceColor, input, DefaultColor, ErrorColor, err, DefaultColor) file = NormalizePath(file, workdir)
return LogInfo("Processing file: %s", FormatPathValue(file))
}
defer file.Close() // This "has" to be done because instructions are resolved in relation to cwd
fileDir := FileRegex.FindStringSubmatch(file)
scanner := bufio.NewScanner(file) if fileDir == nil {
for scanner.Scan() { LogError("Failed to extract directory from %s", FormatSourcePath(file))
line := scanner.Text() return
instruction, err := ParseInstruction(line) }
if err != nil { LogInfo("Changing directory to %s (for %s)",
log.Printf("Error parsing line: %s'%s'%s, error: %s%+v%s", SourceColor, line, DefaultColor, ErrorColor, err, DefaultColor) FormatPathValue(fileDir[1]),
continue FormatPathValue(file))
} err := os.Chdir(fileDir[1])
log.Printf("Read instruction: %s", instruction.String()) if err != nil {
output <- instruction LogError("Failed to change directory to %s: %v",
} FormatSourcePath(fileDir[1]), err)
} return
func ReadFromArgs(output chan LinkInstruction, status chan error) { }
defer close(output) ReadFromFile(file, output, status, false)
defer close(status) // Don't return directory, stay where we are
os.Chdir(workdir)
log.Printf("Reading input from args") }()
for _, arg := range os.Args[1:] { }
instruction, err := ParseInstruction(arg) }
if err != nil {
log.Printf("Error parsing arg: %s'%s'%s, error: %s%+v%s", SourceColor, arg, DefaultColor, ErrorColor, err, DefaultColor) func ReadFromFile(input string, output chan *LinkInstruction, status chan error, doclose bool) {
continue if doclose {
} defer close(output)
output <- instruction defer close(status)
} }
}
func ReadFromStdin(output chan LinkInstruction, status chan error) { input = NormalizePath(input, filepath.Dir(input))
defer close(output) LogInfo("Reading input from file: %s", FormatPathValue(input))
defer close(status)
// Check if this is a YAML file
log.Printf("Reading input from stdin") if IsYAMLFile(input) {
LogInfo("Parsing as YAML file")
info, err := os.Stdin.Stat() instructions, err := ParseYAMLFile(input, filepath.Dir(input))
if err != nil { if err != nil {
log.Fatalf("Failed to stat stdin: %s%+v%s", ErrorColor, err, DefaultColor) LogError("Failed to parse YAML file %s: %v",
status <- err FormatSourcePath(input), err)
return status <- err
} return
if info.Mode()&os.ModeNamedPipe != 0 { }
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() { for _, instruction := range instructions {
line := scanner.Text() instr := instruction // Create a copy to avoid reference issues
instruction, err := ParseInstruction(line) LogInfo("Read YAML instruction: %s", instr.String())
if err != nil { output <- &instr
log.Printf("Error parsing line: %s'%s'%s, error: %s%+v%s", SourceColor, line, DefaultColor, ErrorColor, err, DefaultColor) }
continue return
} }
output <- instruction
} // Handle CSV format (legacy)
if err := scanner.Err(); err != nil { file, err := os.Open(input)
log.Fatalf("Error reading from stdin: %s%+v%s", ErrorColor, err, DefaultColor) if err != nil {
status <- err log.Fatalf("Failed to open file %s%s%s: %s%+v%s",
return SourceColor, input, DefaultColor, ErrorColor, err, DefaultColor)
} return
} }
} defer file.Close()
scanner := bufio.NewScanner(file)
for scanner.Scan() {
line := scanner.Text()
instruction, err := ParseInstruction(line, filepath.Dir(input))
if err != nil {
log.Printf("Error parsing line: %s'%s'%s, error: %s%+v%s",
SourceColor, line, DefaultColor, ErrorColor, err, DefaultColor)
continue
}
log.Printf("Read instruction: %s", instruction.String())
output <- &instruction
}
}
func ReadFromArgs(output chan *LinkInstruction, status chan error) {
defer close(output)
defer close(status)
workdir, _ := os.Getwd()
LogInfo("Reading input from args")
for _, arg := range flag.Args() {
instruction, err := ParseInstruction(arg, workdir)
if err != nil {
LogError("Error parsing arg '%s': %v", arg, err)
continue
}
output <- &instruction
}
}
func ReadFromStdin(output chan *LinkInstruction, status chan error) {
defer close(output)
defer close(status)
workdir, _ := os.Getwd()
LogInfo("Reading input from stdin")
scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() {
line := scanner.Text()
instruction, err := ParseInstruction(line, workdir)
if err != nil {
LogError("Error parsing line '%s': %v", line, err)
continue
}
output <- &instruction
}
if err := scanner.Err(); err != nil {
LogError("Error reading from stdin: %v", err)
status <- err
return
}
}

1
sync
View File

@@ -1 +0,0 @@
test,testdir/test3

6
sync.yaml Normal file
View File

@@ -0,0 +1,6 @@
- source: A/**/*
target: B
- source: A/go.mod
target: B/go.mod
- source: A
target: B/foo

26
sync.yaml.example Normal file
View File

@@ -0,0 +1,26 @@
# Example sync.yaml file
# You can use this format to define symbolic links
# Each link specifies source, target, and optional flags
links:
- source: ~/Documents/config.ini
target: ~/.config/app/config.ini
# This will create a symbolic link, overwriting any existing symlink
force: true
- source: ~/Pictures
target: ~/Documents/Pictures
# This will create a hard link instead of a symbolic link
hard: true
force: true
- source: ~/Scripts/script.sh
target: ~/bin/script.sh
# This will delete a non-symlink file at the target location
# 'delete: true' implies 'force: true'
delete: true
# Alternative format:
# Instead of using the 'links' property, you can define an array directly:
# - source: ~/Documents/config.ini
# target: ~/.config/app/config.ini
# force: true

26
synclib.log Normal file
View File

@@ -0,0 +1,26 @@
18:24:37.967093 Using default sync.yaml file
18:24:37.967093 Input 'sync.yaml' is not absolute, prepending work dir '.'
18:24:37.967593 Reading input from file: C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml
18:24:37.967593 Parsing as YAML file
18:24:37.967593 Failed to parse YAML file C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml: error unmarshaling YAML: yaml: unmarshal errors:
line 1: cannot unmarshal !!seq into main.YAMLConfig
18:24:37.967593 No more instructions to process
18:24:37.968092 No instructions were processed
18:27:59.691333 Using default sync.yaml file
18:27:59.691333 Input 'sync.yaml' is not absolute, prepending work dir '.'
18:27:59.691834 Reading input from file: C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml
18:27:59.691834 Parsing as YAML file
18:27:59.692335 Expanding wildcard source \* in YAML file C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml
18:27:59.692335 Expanded wildcard source \* to 0 links
18:27:59.692836 Expanded wildcard source \* in YAML file C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml to 0 links
18:27:59.692836 No more instructions to process
18:27:59.692836 No instructions were processed
18:28:04.075821 Using default sync.yaml file
18:28:04.076320 Input 'sync.yaml' is not absolute, prepending work dir '.'
18:28:04.076320 Reading input from file: C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml
18:28:04.076320 Parsing as YAML file
18:28:04.076320 Expanding wildcard source \* in YAML file C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml
18:28:04.076821 Expanded wildcard source \* to 0 links
18:28:04.076821 Expanded wildcard source \* in YAML file C:/Users/Administrator/Seafile/Projects-Go/GoProjects/synclib/sync.yaml to 0 links
18:28:04.076821 No more instructions to process
18:28:04.076821 No instructions were processed

319
util.go
View File

@@ -1,132 +1,187 @@
package main package main
import ( import (
"fmt" "fmt"
"log" "os"
"os" "path/filepath"
"path/filepath" "strings"
"strings" "sync"
"sync" "sync/atomic"
"sync/atomic" "time"
"time" )
)
func IsSymlink(path string) (bool, error) {
func IsSymlink(path string) (bool, error) { fileInfo, err := os.Lstat(path)
fileInfo, err := os.Lstat(path) if err != nil {
if err != nil { return false, err
return false, err }
}
// os.ModeSymlink is a bitmask that identifies the symlink mode.
// os.ModeSymlink is a bitmask that identifies the symlink mode. // If the file mode & os.ModeSymlink is non-zero, the file is a symlink.
// If the file mode & os.ModeSymlink is non-zero, the file is a symlink. return fileInfo.Mode()&os.ModeSymlink != 0, nil
return fileInfo.Mode()&os.ModeSymlink != 0, nil }
}
func FileExists(path string) bool {
func FileExists(path string) bool { _, err := os.Lstat(path)
_, err := os.Lstat(path) return err == nil
return err == nil }
}
func NormalizePath(input, workdir string) string {
func NormalizePath(input string) string { input = filepath.Clean(input)
workingdirectory, _ := os.Getwd() input = filepath.ToSlash(input)
input = strings.ReplaceAll(input, "\\", "/") input = strings.ReplaceAll(input, "\"", "")
input = strings.ReplaceAll(input, "\"", "")
if !filepath.IsAbs(input) {
if !filepath.IsAbs(input) { LogInfo("Input '%s' is not absolute, prepending work dir '%s'", input, workdir)
input = workingdirectory + "/" + input var err error
} input = filepath.Join(workdir, input)
input, err = filepath.Abs(input)
return filepath.Clean(input) if err != nil {
} LogError("Failed to get absolute path for %s: %v", FormatSourcePath(input), err)
return input
func AreSame(lhs string, rhs string) bool { }
lhsinfo, err := os.Stat(lhs) }
if err != nil {
return false input = filepath.Clean(input)
} input = filepath.ToSlash(input)
rhsinfo, err := os.Stat(rhs) return input
if err != nil { }
return false
} func AreSame(lhs string, rhs string) bool {
lhsinfo, err := os.Stat(lhs)
return os.SameFile(lhsinfo, rhsinfo) if err != nil {
} return false
}
func ConvertHome(input string) (string, error) { rhsinfo, err := os.Stat(rhs)
if strings.Contains(input, "~") { if err != nil {
homedir, err := os.UserHomeDir() return false
if err != nil { }
return input, fmt.Errorf("unable to convert ~ to user directory with error %+v", err)
} return os.SameFile(lhsinfo, rhsinfo)
}
return strings.Replace(input, "~", homedir, 1), nil
} func ConvertHome(input string) (string, error) {
return input, nil if strings.HasPrefix(input, "~/") {
} homedir, err := os.UserHomeDir()
if err != nil {
func GetSyncFilesRecursively(input string, output chan string, status chan error) { return input, fmt.Errorf("unable to convert ~ to user directory with error %+v", err)
defer close(output) }
defer close(status)
return strings.Replace(input, "~", homedir, 1), nil
var filesProcessed int32 }
var foldersProcessed int32 return input, nil
progressTicker := time.NewTicker(200 * time.Millisecond) }
defer progressTicker.Stop()
func GetSyncFilesRecursively(input string, output chan string, status chan error) {
var wg sync.WaitGroup defer close(output)
var initial sync.Once defer close(status)
wg.Add(1)
directories := make(chan string, 100000) var filesProcessed int32
workerPool := make(chan struct{}, 4000) var foldersProcessed int32
directories <- input var activeWorkers int32
go func() { progressTicker := time.NewTicker(200 * time.Millisecond)
for { defer progressTicker.Stop()
fmt.Printf("\rFiles processed: %d; Folders processed: %d; Workers: %d; Directory Stack Size: %d;", filesProcessed, foldersProcessed, len(workerPool), len(directories))
<-progressTicker.C done := make(chan struct{})
} defer close(done)
}()
directories := make(chan string, 100000)
log.Printf("%+v", len(workerPool)) workerPool := make(chan struct{}, 4000)
go func() { directories <- input
for directory := range directories {
workerPool <- struct{}{} go func() {
wg.Add(1) for {
go func(directory string) { select {
atomic.AddInt32(&foldersProcessed, 1) case <-progressTicker.C:
defer wg.Done() dirCount := len(directories)
defer func() { <-workerPool }() workers := atomic.LoadInt32(&activeWorkers)
fmt.Printf("\rFiles processed: %d; Folders processed: %d; Active workers: %d; Directory queue: %d",
files, err := os.ReadDir(directory) atomic.LoadInt32(&filesProcessed),
if err != nil { atomic.LoadInt32(&foldersProcessed),
log.Printf("Error reading directory %s: %+v", directory, err) workers,
return dirCount)
} case <-done:
// Final progress update
for _, file := range files { fmt.Printf("\nFiles processed: %d; Folders processed: %d; Completed successfully\n",
// log.Printf("Processing file %s", file.Name()) atomic.LoadInt32(&filesProcessed),
if file.IsDir() { atomic.LoadInt32(&foldersProcessed))
directories <- filepath.Join(directory, file.Name()) return
} else { }
// log.Println(file.Name(), DirRegex.MatchString(file.Name())) }
if FileRegex.MatchString(file.Name()) { }()
// log.Printf("Writing")
output <- filepath.Join(directory, file.Name()) allDone := make(chan struct{})
}
atomic.AddInt32(&filesProcessed, 1) go func() {
} // WTF is this waitgroup?
} // Nowhere is it added...
// log.Printf("Done reading directory %s", directory) var wg sync.WaitGroup
initial.Do(func() { go func() {
// Parallelism is very difficult... for {
time.Sleep(250 * time.Millisecond) if atomic.LoadInt32(&activeWorkers) == 0 && len(directories) == 0 {
wg.Done() time.Sleep(10 * time.Millisecond)
}) if atomic.LoadInt32(&activeWorkers) == 0 && len(directories) == 0 {
}(directory) close(allDone)
} return
}() }
}
wg.Wait() time.Sleep(50 * time.Millisecond)
log.Printf("Files processed: %d; Folders processed: %d", filesProcessed, foldersProcessed) }
} }()
for {
select {
case directory, ok := <-directories:
if !ok {
wg.Wait()
return
}
atomic.AddInt32(&activeWorkers, 1)
go func(dir string) {
workerPool <- struct{}{}
atomic.AddInt32(&foldersProcessed, 1)
processDirectory(dir, directories, output, &filesProcessed)
<-workerPool
atomic.AddInt32(&activeWorkers, -1)
}(directory)
}
}
}()
<-allDone
if atomic.LoadInt32(&filesProcessed) > 0 {
LogInfo("Files processed: %d; Folders processed: %d",
atomic.LoadInt32(&filesProcessed),
atomic.LoadInt32(&foldersProcessed))
}
}
func processDirectory(directory string, directories chan<- string, output chan<- string, filesProcessed *int32) {
files, err := os.ReadDir(directory)
if err != nil {
LogError("Error reading directory %s: %v", directory, err)
return
}
for _, file := range files {
if file.IsDir() {
directories <- filepath.Join(directory, file.Name())
} else {
if IsYAMLSyncFile(file.Name()) {
output <- filepath.Join(directory, file.Name())
}
atomic.AddInt32(filesProcessed, 1)
}
}
}
func IsYAMLSyncFile(filename string) bool {
return filename == "sync.yaml" || filename == "sync.yml"
}