initial commit

kpurdon · kpurdon · commit 82ec4ff5476a · 2015-06-18T17:31:15.000-06:00
diff --git a/golang/README.md b/golang/README.md
@@ -0,0 +1,25 @@
+gomr: go [m]utli-[r]epository manager
+-----
+
+gomr is a golang command-line tool used to execute arbitrary commands on a set of tagged directories.
+
+## Features
+
+* Configuration is stored in the registered directory (allows for version-controlled configuration)
+* Any arbitrary command can be run, no ties to version-control specific commands.
+
+
+## Installation
+
+I'm not distributing binaries yet (TODO) for now checkout and `go install` to build the binary.
+
+## Command Help
+
+For help run:
+* `gomr --help`
+* `gomr register --help`
+* `gomr run --help`
+
+## Ports
+
+Pymr has been ported to ruby as [rumr](https://github.com/kpurdon/rumr) and python as [pymr](https://github.com/kpurdon/pymr).
diff --git a/golang/gomr.go b/golang/gomr.go
@@ -0,0 +1,60 @@
+package main
+
+import (
+	"github.com/codegangsta/cli"
+	"os"
+)
+
+func main() {
+	app := cli.NewApp()
+	app.Name = "gomr"
+	app.Author = "Kyle W. Purdon"
+	app.Version = "0.0.1"
+	app.Usage = "multi-repository manager in go"
+	app.Commands = []cli.Command{
+		{
+			Name:   "register",
+			Usage:  "register a directory",
+			Action: register,
+			Flags: []cli.Flag{
+				cli.StringFlag{
+					Name:  "directory, d",
+					Value: "./",
+					Usage: "directory to tag",
+				},
+				cli.StringFlag{
+					Name:  "tag, t",
+					Value: "default",
+					Usage: "tag to add for directory",
+				},
+				cli.BoolFlag{
+					Name:  "append",
+					Usage: "append the tag to an existing registered directory",
+				},
+			},
+		},
+		{
+			Name:   "run",
+			Usage:  "run a command",
+			Action: run,
+			Flags: []cli.Flag{
+				cli.StringFlag{
+					Name:  "basepath, b",
+					Value: "./",
+					Usage: "path to begin the recursive search",
+				},
+				cli.StringFlag{
+					Name:  "tag, t",
+					Value: "default",
+					Usage: "tag to add for directory",
+				},
+				cli.BoolFlag{
+					Name:  "dryrun, d",
+					Usage: "print (dont execute) the commands that will be run",
+				},
+			},
+		},
+	}
+
+	app.Run(os.Args)
+}
diff --git a/golang/register.go b/golang/register.go
@@ -0,0 +1,50 @@
+package main
+
+import (
+	"fmt"
+	"github.com/codegangsta/cli"
+	"github.com/robfig/config"
+	"os"
+	"path"
+	"strings"
+)
+
+func AppendIfMissing(slice []string, i string) []string {
+	for _, ele := range slice {
+		if ele == i {
+			return slice
+		}
+	}
+	return append(slice, i)
+}
+
+func register(ctx *cli.Context) {
+
+	fn := path.Join(ctx.String("directory"), ".gomr")
+
+	newTags := strings.Split(ctx.String("tag"), ",")
+
+	if ctx.Bool("append") {
+		if _, err := os.Stat(fn); err == nil {
+			cfg, _ := config.ReadDefault(".gomr")
+			curTags, _ := cfg.String("gomr", "tags")
+
+			for _, tag := range strings.Split(curTags, ",") {
+				newTags = AppendIfMissing(newTags, tag)
+			}
+		} else {
+			err := "append used, existing file not found."
+			fmt.Fprintf(os.Stderr, "error: %v\n", err)
+			os.Exit(1)
+		}
+	}
+
+	outTags := strings.Join(newTags, ",")
+
+	outCfg := config.NewDefault()
+	outCfg.AddSection("gomr")
+	outCfg.AddOption("gomr", "tags", outTags)
+	outCfg.WriteFile(fn, 0644, "gomr configuration file")
+
+	return
+}
diff --git a/golang/run.go b/golang/run.go
@@ -0,0 +1,57 @@
+package main
+
+import (
+	"fmt"
+	"github.com/codegangsta/cli"
+	"github.com/robfig/config"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"strings"
+)
+
+func RunGomr(ctx *cli.Context) filepath.WalkFunc {
+	return func(path string, f os.FileInfo, err error) error {
+
+		tag := ctx.String("tag")
+		dryrun := ctx.Bool("dryrun")
+
+		if len(ctx.Args()) == 0 {
+			panic("no command given")
+		}
+
+		command := ctx.Args()[0]
+
+		if strings.Contains(path, ".gomr") {
+
+			cfg, _ := config.ReadDefault(path)
+			gomrTags, _ := cfg.String("gomr", "tags")
+
+			if strings.Contains(gomrTags, tag) {
+				if dryrun {
+					fmt.Printf("Would run %s in %s\n", command, filepath.Dir(path))
+				} else {
+					os.Chdir(filepath.Dir(path))
+					cmd := exec.Command("sh", "-c", command)
+					stdout, err := cmd.Output()
+
+					if err != nil {
+						panic(err.Error())
+					}
+
+					println(string(stdout))
+				}
+			}
+
+		}
+		return nil
+	}
+}
+
+func run(ctx *cli.Context) {
+
+	root := ctx.String("basepath")
+	filepath.Walk(root, RunGomr(ctx))
+
+	return
+}
diff --git a/python/README.md b/python/README.md
@@ -0,0 +1,31 @@
+pymr: [Py]thon [m]utli-[r]epository Tool
+-----
+
+PyMR is a python command-line tool used to execute arbitrary commands on a set of tagged directories.
+
+## Features
+
+* Configuration is stored in the registered directory (allows for version-controlled configuration)
+* Any arbitrary command can be run, no ties to version-control specific commands.
+
+## Where is PyMR
+
+PyMR source is available on [Github](https://github.com/kpurdon/pymr) and is released to [PyPI](https://pypi.python.org/pypi/pymr).
+
+## Installation
+
+From pypi: `pip install pymr`
+
+From source:
+* `python setup.py install`
+
+## Command Help
+
+For help run:
+* `pymr --help`
+* `pymr register --help`
+* `pymr run --help`
+
+## Ports
+
+Pymr has been ported to ruby as [rumr](https://github.com/kpurdon/rumr) and go as [gomr](https://github.com/kpurdon/gomr).
diff --git a/python/pymr/__init__.py b/python/pymr/__init__.py
diff --git a/python/pymr/__init__.pyc b/python/pymr/__init__.pyc
diff --git a/python/pymr/pymr.py b/python/pymr/pymr.py
@@ -0,0 +1,61 @@
+import os
+import pickle
+import fnmatch
+from subprocess import call
+
+import pathlib
+import click
+
+
+@click.group()
+def pymr():
+    pass
+
+
+@pymr.command()
+@click.option('--directory', '-d', default='./')
+@click.option('--tag', '-t', multiple=True)
+@click.option('--append', is_flag=True)
+def register(directory, tag, append):
+    '''
+    register a directory
+    '''
+
+    pymr_file = os.path.join(directory, '.pymr')
+    new_tags = tag
+
+    if append:
+        if os.path.exists(pymr_file):
+            cur_tags = pickle.load(open(pymr_file))
+            new_tags = tuple(set(new_tags + cur_tags))
+
+    pickle.dump(new_tags, open(pymr_file, 'wb'))
+
+
+@pymr.command()
+@click.argument('command')
+@click.option('--basepath', '-b', default='./')
+@click.option('--tag', '-t')
+@click.option('--dryrun', is_flag=True)
+def run(command, basepath, tag, dryrun):
+    '''
+    run a given command in matching sub-directories
+    '''
+
+    for fn in pathlib.Path(basepath).glob('**/.pymr'):
+
+        with fn.open(mode='rb') as f:
+            cur_tags = pickle.load(f)
+
+        parent_dir = str(fn.parent)
+
+        if tag in cur_tags:
+            if dryrun:
+                print('Would run {0} in {1}'.format(command, parent_dir))
+            else:
+                os.chdir(parent_dir)
+                call(command, shell=True)
+
+
+if __name__ == '__main__':
+    pymr()
diff --git a/python/pymr/pymr.pyc b/python/pymr/pymr.pyc
diff --git a/python/setup.py b/python/setup.py
@@ -0,0 +1,37 @@
+from setuptools import setup, find_packages
+
+classifiers = [
+    'Environment :: Console',
+    'Operating System :: OS Independent',
+    'License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)',
+    'Intended Audience :: Developers',
+    'Programming Language :: Python',
+    'Programming Language :: Python :: 2',
+    'Programming Language :: Python :: 2.7'
+]
+
+setuptools_kwargs = {
+    'install_requires': [
+        'click>=4,<5',
+        'pathlib>=1,<2'
+    ],
+    'entry_points': {
+        'console_scripts': [
+            'pymr = pymr.pymr:pymr',
+        ]
+    }
+}
+
+
+setup(
+    name='pymr',
+    description='A tool for executing ANY command in a set of tagged directories.',
+    author='Kyle W Purdon',
+    author_email='kylepurdon@gmail.com',
+    url='https://github.com/kpurdon/pymr',
+    download_url='https://github.com/kpurdon/pymr',
+    version='2.0.1',
+    packages=find_packages(),
+    classifiers=classifiers,
+    **setuptools_kwargs
+)
diff --git a/ruby/Gemfile b/ruby/Gemfile
@@ -0,0 +1,3 @@
+source 'https://rubygems.org'
+
+gem 'thor', '~>0.19.1'
diff --git a/ruby/README.md b/ruby/README.md
@@ -0,0 +1,29 @@
+rumr: [Ru]by [m]utli-[r]epository Tool
+-----
+
+Rumr is a ruby command-line tool used to execute arbitrary commands on a set of tagged directories.
+
+## Features
+
+* Configuration is stored in the registered directory (allows for version-controlled configuration)
+* Any arbitrary command can be run, no ties to version-control specific commands.
+
+
+## Installation
+
+From rubygems: `gem install rumr`
+
+From source:
+* `gem build rumr.gemspec`
+* `gem install rumr-[version].gem`
+
+## Command Help
+
+For help run:
+* `rumr help`
+* `rumr help register`
+* `rumr help exec`
+
+## Ports
+
+rumr has been ported to python as [pymr](https://github.com/kpurdon/pymr) and go as [gomr](https://github.com/kpurdon/gomr).
diff --git a/ruby/bin/rumr b/ruby/bin/rumr
@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'rumr'
+
+Rumr.start(ARGV)
diff --git a/ruby/lib/rumr.rb b/ruby/lib/rumr.rb
diff --git a/ruby/rumr.gemspec b/ruby/rumr.gemspec

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+source 'https://rubygems.org'`
	`2`	`+`
	`3`	`+gem 'thor', '~>0.19.1'`