subtree(users/wpcarro): docking briefcase at '24f5a642'
git-subtree-dir: users/wpcarro git-subtree-mainline:464bbcb15cgit-subtree-split:24f5a642afChange-Id: I6105b3762b79126b3488359c95978cadb3efa789
This commit is contained in:
commit
019f8fd211
766 changed files with 175420 additions and 0 deletions
82
users/wpcarro/tools/symlinkManager/main.go
Normal file
82
users/wpcarro/tools/symlinkManager/main.go
Normal file
|
|
@ -0,0 +1,82 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"utils"
|
||||
)
|
||||
|
||||
var hostnames = map[string]string{
|
||||
os.Getenv("DESKTOP"): "desktop",
|
||||
os.Getenv("LAPTOP"): "work_laptop",
|
||||
}
|
||||
|
||||
func main() {
|
||||
audit := flag.Bool("audit", false, "Output all symlinks that would be deleted. This is the default behavior. This option is mutually exclusive with the --seriously option.")
|
||||
seriously := flag.Bool("seriously", false, "Actually delete the symlinks. This option is mutually exclusive with the --audit option.")
|
||||
repoName := flag.String("repo-name", "briefcase", "The name of the repository.")
|
||||
deviceOnly := flag.Bool("device-only", false, "Only output the device-specific dotfiles.")
|
||||
flag.Parse()
|
||||
|
||||
if !*audit && !*seriously {
|
||||
log.Fatal(errors.New("Either -audit or -seriously needs to be set."))
|
||||
}
|
||||
if *audit == *seriously {
|
||||
log.Fatal(errors.New("Arguments -audit and -seriously are mutually exclusive"))
|
||||
}
|
||||
|
||||
home, err := os.UserHomeDir()
|
||||
utils.FailOn(err)
|
||||
count := 0
|
||||
|
||||
err = filepath.Walk(home, func(path string, info os.FileInfo, err error) error {
|
||||
if utils.IsSymlink(info.Mode()) {
|
||||
dest, err := os.Readlink(path)
|
||||
utils.FailOn(err)
|
||||
|
||||
var predicate func(string) bool
|
||||
|
||||
if *deviceOnly {
|
||||
predicate = func(dest string) bool {
|
||||
var hostname string
|
||||
hostname, err = os.Hostname()
|
||||
utils.FailOn(err)
|
||||
seeking, ok := hostnames[hostname]
|
||||
if !ok {
|
||||
log.Fatal(fmt.Sprintf("Hostname \"%s\" not supported in the hostnames map.", hostname))
|
||||
}
|
||||
return strings.Contains(dest, *repoName) && strings.Contains(dest, seeking)
|
||||
}
|
||||
} else {
|
||||
predicate = func(dest string) bool {
|
||||
return strings.Contains(dest, *repoName)
|
||||
}
|
||||
}
|
||||
|
||||
if predicate(dest) {
|
||||
if *audit {
|
||||
fmt.Printf("%s -> %s\n", path, dest)
|
||||
} else if *seriously {
|
||||
fmt.Printf("rm %s\n", path)
|
||||
err = os.Remove(path)
|
||||
utils.FailOn(err)
|
||||
}
|
||||
count += 1
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
utils.FailOn(err)
|
||||
if *audit {
|
||||
fmt.Printf("Would have deleted %d symlinks.\n", count)
|
||||
} else if *seriously {
|
||||
fmt.Printf("Successfully deleted %d symlinks.\n", count)
|
||||
}
|
||||
|
||||
os.Exit(0)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue