a2km 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3df1210c6b58fc739fde387f9280174885f01e58
-  data.tar.gz: 9413d30be46d71bf2a5155169f4295543be6dc70
+  metadata.gz: 1bd0be9d8cdd24b99dfca16dad1fdaca90ae3eb2
+  data.tar.gz: 3e45923484920d0c0849d9a408d0ba7e644257d9
 SHA512:
-  metadata.gz: 2feac328e487b78a2dcb4b06001e5371ba71f4f412ef7c025d6a7f4be880002e65fb53f0b2d2e06815749f0361fc8eea057e3edf1b7628ef0e81a77da1cbb88c
-  data.tar.gz: fc39bc254f3056a4afc6ffa6e84114deaba7c96e06640cafd6af9d126ae48aebc07a3bd878dced14c22825cecc67204327a30d7d4517c7d4e8deb5ddbd3f9566
+  metadata.gz: 94d3adad6abb635bc980075c362bf875af5b8a78a9ed8f8d9fefa53b37e8eddcff9f1b0ee996a8dc756836a2e8b88d3b4966ac35bd7757e0ffb402e8a72d9a23
+  data.tar.gz: 5194abffa5630161a00d9f938d9ee29319f83bfe668d30cfb0689d3286b2edf6a34d08b7995df704f7277322a78f84125504dcd077f441eafae2adedfa30ba4b

data/.gitignore

@@ -0,0 +1,35 @@
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/test/tmp/
+/test/version_tmp/
+/tmp/
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalisation:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+# Gemfile.lock
+# .ruby-version
+# .ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc

data/LICENSE

@@ -0,0 +1,28 @@
+Copyright (c) 2015, Min RK
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without
+modification, are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+
+* Redistributions in binary form must reproduce the above copyright notice,
+  this list of conditions and the following disclaimer in the documentation
+  and/or other materials provided with the distribution.
+
+* Neither the name of a2km nor the names of its
+  contributors may be used to endorse or promote products derived from
+  this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+

data/README.md

@@ -0,0 +1,45 @@
+# Assistant to the Kernel Manager
+
+Utility for working with [Jupyter](https://jupyter.org) kernelspecs.
+
+
+## Install
+
+    gem install a2km
+
+
+## Examples
+
+    a2km clone python3 myenvpy3
+    a2km set python3-copy display_name "Super cool Python Kernel"
+    a2km add-env python3-copy SPARK_HOME=/path/to/spark
+    a2km add-argv python3-copy -- --debug
+    # <debug some stuff>
+    a2km rm-argv python3-copy -- debug
+
+## Kernelspecs for environments
+
+a2km has an `env-kernel` subcommand for creating kernelspecs for your conda or virtual environments.
+Just pass a2km the name of the env, and you should be set:
+
+    conda create -n myenv ipykernel
+    a2km env-kernel myenv
+
+
+## Commands
+
+    add-argv   Add argument(s) to a kernelspec launch command
+    add-env    Add environment variables to a kernelspec
+    clone      Clone a kernelspec
+    env-kernel Create a kernel from an env (conda or virtualenv)
+    help       Display global or [command] help documentation
+    locate     Print the path of a kernelspec
+    rename     Rename a kernelspec
+    rm         Remove a kernelspec
+    rm-argv    Remove arguments from a kernelspec launch command
+    rm-env     Remove environment variables from a kernelspec
+    set        Set a value in the kernelspec
+    show       Show info about a kernelspec
+
+
+![Assistant TO the Kernel Manager](http://i.imgur.com/F0WLaYR.jpg)

data/a2km.gemspec

@@ -0,0 +1,21 @@
+require File.dirname(__FILE__) + '/lib/a2km/version'
+require 'date'
+
+Gem::Specification.new do |s|
+  s.require_paths = ["lib"]
+  s.name        = 'a2km'
+  s.version     = A2KM::VERSION
+  s.date        = Date.today.to_s
+  s.summary     = 'Assistant to the Kernel Manager'
+  s.description = 'Working with Juptyer kernels'
+  s.authors     = ['Min RK']
+  s.email       = 'benjaminrk@gmail.com'
+  s.homepage    = 'http://github.com/minrk/a2km'
+  s.license     = 'BSD'
+  s.executables = ['a2km']
+  s.files       = `git ls-files`.split($/)
+  s.require_paths = %w(lib)
+  s.add_runtime_dependency 'commander'
+  s.add_runtime_dependency 'highline'
+  s.add_runtime_dependency 'liquid'
+end

data/bin/a2km

@@ -1,286 +1,5 @@
 #!/usr/bin/env ruby
 
-require 'json'
+require File.expand_path('../lib/a2km', __dir__)
 
-require 'rubygems'
-require 'commander'
-require 'highline'
-
-
-class AssitantToTheKernelManager
-  include Commander::Methods
-  
-  def initialize
-    @kernels = nil
-  end
-  
-  def kernels
-    if @kernels.nil?
-      js = JSON.parse `jupyter kernelspec list --json`
-      @kernels = js['kernelspecs']
-    end
-    @kernels
-  end
-  
-  def kernel_dirs
-    paths = JSON.parse `jupyter --paths --json`
-    paths['data'].map do |p|
-      p + '/' + 'kernels'
-    end
-  end
-  
-  def user_kernel_dir
-    kernel_dirs.first
-  end
-  
-  def get_kernel(name)
-    "Get a kernel by name. Die with message if not found."
-    if not kernels.has_key? name
-      STDERR.puts "No such kernel: #{name}"
-      STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
-      exit(-1)
-    end
-    kernels[name]
-  end
-  
-  def kernel_json_path(name)
-    "Return path to a kernel's kernel.json"
-    kernel = get_kernel(name)
-    File.join(kernel['resources_dir'], 'kernel.json')
-  end
-  
-  def get_kernel_json(name)
-    "Return kernel.json contents for a given kernelspec"
-    File.open(kernel_json_path(name)) do |f|
-      return JSON.parse f.read
-    end
-  end
-  
-  def save_kernel_json(name, data)
-    "save new kernel JSON"
-    File.open(kernel_json_path(name), 'w') do |f|
-      f.write JSON.pretty_generate(data)
-    end
-  end
-  
-  def run
-    program :name, 'Assistant to the KernelManager'
-    program :version, '0.0.1'
-    program :description, 'Work with Jupyter kernelspecs'
-    
-    command :rename do |c|
-      c.syntax = 'a2km rename <from> <to>'
-      c.summary = 'Rename a kernelspec'
-      c.description = 'Rename kernelspec FROM to TO'
-      c.action do |args, options|
-        from = args.shift
-        to = args.shift
-        kernel = get_kernel(from)
-        src = kernel['resources_dir']
-        
-        dst = File.join File.dirname(src), to
-        if File.exists? dst
-          STDERR.puts "Destination already exists: #{dst}"
-          exit(-1)
-        end
-        
-        puts "Moving #{src} → #{dst}"
-        FileUtils.mv src, dst
-      end
-    end
-    
-    command :show do |c|
-      c.syntax = 'a2km show <spec>'
-      c.description = 'Show info about a kernelspec'
-      c.action do |args, options|
-        name = args.first
-        kernel = get_kernel(name)
-        spec = kernel['spec']
-        puts "Kernel: #{name} (#{spec['display_name']})"
-        puts "  path: #{kernel['resources_dir']}"
-        puts "  argv: #{spec['argv'].join(' ')}"
-      end
-    end
-    
-    command :locate do |c|
-      c.syntax = 'a2km locate <spec>'
-      c.description = 'Print the path of a kernelspec'
-      c.action do |args, options|
-        puts get_kernel(args.first)['resources_dir']
-      end
-    end
-    
-    command :"add-argv" do |c|
-      c.syntax = 'a2km add-argv <spec> <arg1> [arg2] ...'
-      c.description = 'Add argument(s) to a kernelspec launch command'
-      c.action do |args, options|
-        name = args.shift
-        kernelspec = get_kernel_json(name)
-        kernelspec['argv'] += args
-        save_kernel_json(name, kernelspec)
-        puts "New argv: #{kernelspec['argv'].join(' ')}"
-      end
-    end
-    
-    command :"rm-argv" do |c|
-      c.syntax = 'a2km rm-argv <spec> <arg1> [arg2] ...'
-      c.summary = 'Remove arguments from a kernelspec launch command'
-      c.description = c.summary + ". To remove args starting with '-'," \
-        " use '--'."
-      c.examples = [
-        "a2km rm-argv myspec x",
-        "a2km rm-argv myspec -- --debug",
-      ]
-      c.action do |args, options|
-        name = args.shift
-        to_remove = args
-        kernelspec = get_kernel_json(name)
-        argv = kernelspec['argv']
-        argv.reject! { |arg| to_remove.include? arg }
-        save_kernel_json(name, kernelspec)
-        puts "New argv: #{argv.join(' ')}"
-      end
-    end
-
-    command :"add-env" do |c|
-      c.syntax = 'a2km add-env <spec> <key=value> [key=value] ...'
-      c.summary = 'Add environment variables to a kernelspec'
-      c.description = 'Add environment variables to a kernelspec.' \
-        ' If no value is given, the value from the current env is used.'
-      c.action do |args, options|
-        name = args.shift
-        spec = get_kernel_json(name)
-        if not spec.has_key? 'env'
-          spec['env'] = {}
-        end
-        env = spec['env']
-        args.each do |arg|
-          key_value = arg.split('=', 2)
-          key = key_value.first
-          if key_value.length == 2
-            value = key_value[1]
-          else
-            value = ENV[key]
-          end
-          env[key] = value
-        end
-        save_kernel_json(name, spec)
-        puts "New env: #{env}"
-      end
-    end
-    
-    command :"rm-env" do |c|
-      c.syntax = 'a2km rm-env <spec> <key> [key] ...'
-      c.description = 'Remove environment variables from a kernelspec'
-      c.action do |args, options|
-        name = args.shift
-        spec = get_kernel_json(name)
-        if not spec.has_key? 'env'
-          spec['env'] = {}
-        end
-        env = spec['env']
-        args.each do |arg|
-          env.delete(arg)
-        end
-        save_kernel_json(name, spec)
-        puts "New env: #{env}"
-      end
-    end
-
-    command :rm do |c|
-      c.syntax = 'a2km rm <spec>'
-      c.description = 'Remove a kernelspec'
-      c.option '-f', "Force removal (skip confirmation)"
-      c.action do |args, options|
-        name = args.shift
-        if not kernels.has_key? name
-          STDERR.puts "No such kernel: #{name}"
-          STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
-          exit(-1)
-        end
-        path = kernels[name]['resources_dir']
-        if options.f or not HighLine.agree("Permanently delete #{path}? (yes/no)")
-          STDERR.puts "Aborting."
-          exit(-1)
-        end
-        puts "Removing #{path}"
-        FileUtils.rm_r path
-      end
-    end
-    
-    command :set do |c|
-      c.syntax = 'a2km set <name> <key> <value>'
-      c.description = 'Set a value in the kernelspec'
-      c.examples = [
-        'a2km set python3 display_name "My Python 3"'
-      ]
-      c.action do |args, options|
-        name = args.shift
-        key = args.shift
-        value = args.shift
-        if value.nil?
-          STDERR.puts c.syntax
-          exit(-1)
-        end
-        puts "Setting #{name}.#{key} = '#{value}'"
-        spec = get_kernel_json(name)
-        spec[key] = value
-        save_kernel_json(name, spec)
-      end
-    end
-    
-    command :clone do |c|
-      c.syntax = 'a2km clone <from> <to> [display_name]'
-      c.summary = 'Clone a kernelspec'
-      c.description = 'Clones kernelspec FROM to TO'
-      c.option '--user', 'Force clone to be in the user directory' \
-        ' (default is to use the same directory as FROM)'
-      c.action do |args, options|
-        options.default :user => false
-        
-        if args.length < 2 or args.length > 3
-          STDERR.puts "Must specify FROM and TO"
-          exit(-1)
-        end
-        from_name = args.shift
-        to_name = args.shift
-        if args.length > 0
-          display_name = args.shift
-        else
-          display_name = to_name
-        end
-        
-        from = get_kernel(from_name)
-        from = kernels[from_name]
-        src = from['resources_dir']
-        
-        if options.user?
-          dst_dir = user_kernel_dir
-          makedirs(dst_dir)
-        else
-          dst_dir = File.dirname src
-        end
-        dst = File.join dst_dir, to_name
-        if File.exists? dst
-          STDERR.puts "Destination already exists: #{dst}"
-          exit(-1)
-        end
-        
-        puts "Cloning #{src} → #{dst}"
-        FileUtils.cp_r src, dst
-        kernel_json = File.join(dst, 'kernel.json')
-        kernelspec = File.open(kernel_json) do |f|
-          JSON.parse f.read
-        end
-        kernelspec['display_name'] = display_name
-        File.open(kernel_json, 'w') do |f|
-          f.write JSON.pretty_generate(kernelspec)
-        end
-      end
-    end
-
-    run!
-  end
-end
-
-AssitantToTheKernelManager.new.run
+A2KM::run

data/lib/a2km.rb

@@ -0,0 +1,4 @@
+require __dir__ + '/a2km/version'
+require __dir__ + '/a2km/api'
+require __dir__ + '/a2km/env'
+require __dir__ + '/a2km/command'

data/lib/a2km/api.rb

@@ -0,0 +1,59 @@
+# Copyright Min RK, License: BSD 3-clause
+
+require 'json'
+
+module A2KM
+
+  def initialize
+    @kernels = nil
+  end
+
+  def A2KM.kernels
+    if @kernels.nil?
+      js = JSON.parse `jupyter kernelspec list --json`
+      @kernels = js['kernelspecs']
+    end
+    @kernels
+  end
+
+  def A2KM.kernel_dirs
+    paths = JSON.parse `jupyter --paths --json`
+    paths['data'].map do |p|
+      p + '/' + 'kernels'
+    end
+  end
+
+  def A2KM.user_kernel_dir
+    kernel_dirs.first
+  end
+
+  def A2KM.get_kernel(name)
+    "Get a kernel by name. Die with message if not found."
+    if not kernels.has_key? name
+      STDERR.puts "No such kernel: #{name}"
+      STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
+      exit(-1)
+    end
+    kernels[name]
+  end
+
+  def A2KM.kernel_json_path(name)
+    "Return path to a kernel's kernel.json"
+    kernel = get_kernel(name)
+    File.join(kernel['resources_dir'], 'kernel.json')
+  end
+
+  def A2KM.get_kernel_json(name)
+    "Return kernel.json contents for a given kernelspec"
+    File.open(kernel_json_path(name)) do |f|
+      return JSON.parse f.read
+    end
+  end
+
+  def A2KM.save_kernel_json(name, data)
+    "save new kernel JSON"
+    File.open(kernel_json_path(name), 'w') do |f|
+      f.write JSON.pretty_generate(data)
+    end
+  end
+end

data/lib/a2km/command.rb

@@ -0,0 +1,263 @@
+# Copyright Min RK, License: BSD 3-clause
+
+require 'json'
+
+require 'rubygems'
+require 'commander'
+require 'highline'
+
+module A2KM
+  
+  # Start the a2km entrypoint
+  def A2KM.run
+    CLI.new.run
+  end
+  
+  # The A2KM CLI entrypoint
+  class CLI
+    include Commander::Methods
+    
+    def run
+      program :name, 'Assistant to the KernelManager'
+      program :version, VERSION
+      program :description, 'Work with Jupyter kernelspecs'
+
+      command :rename do |c|
+        c.syntax = 'a2km rename <from> <to>'
+        c.summary = 'Rename a kernelspec'
+        c.description = 'Rename kernelspec FROM to TO'
+        c.action do |args, options|
+          from = args.shift
+          to = args.shift
+          kernel = A2KM.get_kernel(from)
+          src = kernel['resources_dir']
+
+          dst = File.join File.dirname(src), to
+          if File.exists? dst
+            STDERR.puts "Destination already exists: #{dst}"
+            exit(-1)
+          end
+
+          puts "Moving #{src} → #{dst}"
+          FileUtils.mv src, dst
+        end
+      end
+
+      command :show do |c|
+        c.syntax = 'a2km show <spec>'
+        c.description = 'Show info about a kernelspec'
+        c.action do |args, options|
+          name = args.first
+          kernel = A2KM.get_kernel(name)
+          spec = kernel['spec']
+          puts "Kernel: #{name} (#{spec['display_name']})"
+          puts "  path: #{kernel['resources_dir']}"
+          puts "  argv: #{spec['argv'].join(' ')}"
+        end
+      end
+
+      command :locate do |c|
+        c.syntax = 'a2km locate <spec>'
+        c.description = 'Print the path of a kernelspec'
+        c.action do |args, options|
+          puts A2KM.get_kernel(args.first)['resources_dir']
+        end
+      end
+
+      command :"add-argv" do |c|
+        c.syntax = 'a2km add-argv <spec> <arg1> [arg2] ...'
+        c.description = 'Add argument(s) to a kernelspec launch command'
+        c.action do |args, options|
+          name = args.shift
+          kernelspec = A2KM.get_kernel_json(name)
+          kernelspec['argv'] += args
+          A2KM.save_kernel_json(name, kernelspec)
+          puts "New argv: #{kernelspec['argv'].join(' ')}"
+        end
+      end
+
+      command :"rm-argv" do |c|
+        c.syntax = 'a2km rm-argv <spec> <arg1> [arg2] ...'
+        c.summary = 'Remove arguments from a kernelspec launch command'
+        c.description = c.summary + ". To remove args starting with '-'," \
+          " use '--'."
+        c.examples = [
+          "a2km rm-argv myspec x",
+          "a2km rm-argv myspec -- --debug",
+        ]
+        c.action do |args, options|
+          name = args.shift
+          to_remove = args
+          kernelspec = A2KM.get_kernel_json(name)
+          argv = kernelspec['argv']
+          argv.reject! { |arg| to_remove.include? arg }
+          A2KM.save_kernel_json(name, kernelspec)
+          puts "New argv: #{argv.join(' ')}"
+        end
+      end
+
+      command :"add-env" do |c|
+        c.syntax = 'a2km add-env <spec> <key=value> [key=value] ...'
+        c.summary = 'Add environment variables to a kernelspec'
+        c.description = 'Add environment variables to a kernelspec.' \
+          ' If no value is given, the value from the current env is used.'
+        c.action do |args, options|
+          name = args.shift
+          spec = A2KM.get_kernel_json(name)
+          if not spec.has_key? 'env'
+            spec['env'] = {}
+          end
+          env = spec['env']
+          args.each do |arg|
+            key_value = arg.split('=', 2)
+            key = key_value.first
+            if key_value.length == 2
+              value = key_value[1]
+            else
+              value = ENV[key]
+            end
+            env[key] = value
+          end
+          A2KM.save_kernel_json(name, spec)
+          puts "New env: #{env}"
+        end
+      end
+
+      command :"rm-env" do |c|
+        c.syntax = 'a2km rm-env <spec> <key> [key] ...'
+        c.description = 'Remove environment variables from a kernelspec'
+        c.action do |args, options|
+          name = args.shift
+          spec = A2KM.get_kernel_json(name)
+          if not spec.has_key? 'env'
+            spec['env'] = {}
+          end
+          env = spec['env']
+          args.each do |arg|
+            env.delete(arg)
+          end
+          A2KM.save_kernel_json(name, spec)
+          puts "New env: #{env}"
+        end
+      end
+
+      command :rm do |c|
+        c.syntax = 'a2km rm <spec>'
+        c.description = 'Remove a kernelspec'
+        c.option '-f', "Force removal (skip confirmation)"
+        c.action do |args, options|
+          name = args.shift
+          if not A2KM.kernels.has_key? name
+            STDERR.puts "No such kernel: #{name}"
+            STDERR.puts "Found kernels: #{kernels.keys.sort.join(' ')}"
+            exit(-1)
+          end
+          path = A2KM.kernels[name]['resources_dir']
+          if options.f or not HighLine.agree("Permanently delete #{path}? (yes/no)")
+            STDERR.puts "Aborting."
+            exit(-1)
+          end
+          puts "Removing #{path}"
+          FileUtils.rm_r path
+        end
+      end
+
+      command :"env-kernel" do |c|
+        c.syntax = 'a2km env-kernel [--venv|--conda] <name> [-e env-name] [--prefix PREFIX] [--user]'
+        c.description = 'Create a kernel from an env (conda or virtualenv)'
+        
+        c.option '--conda', "use conda env (default)"
+        c.option '--venv', "use virtualenv"
+        c.option '-e ENV', "specify the environment name to use (if different from <name>)"
+        c.option '--prefix PREFIX', 'specify the prefix for the executable to be installed'
+        c.action do |args, options|
+          name = args.shift
+          options.default :user => false
+          options.default :e => name
+          options.default :prefix => '~'
+          env = options.e
+
+          kind = 'conda'
+          if options.venv
+            kind = 'venv'
+          end
+          puts "Making kernel '#{name}' for #{kind}:#{env}"
+          spec = A2KM::make_env_kernel(name, env, kind: kind, prefix: options.prefix)
+        end
+      end
+
+      command :set do |c|
+        c.syntax = 'a2km set <name> <key> <value>'
+        c.description = 'Set a value in the kernelspec'
+        c.examples = [
+          'a2km set python3 display_name "My Python 3"'
+        ]
+        c.action do |args, options|
+          name = args.shift
+          key = args.shift
+          value = args.shift
+          if value.nil?
+            STDERR.puts c.syntax
+            exit(-1)
+          end
+          puts "Setting #{name}.#{key} = '#{value}'"
+          spec = A2KM.get_kernel_json(name)
+          spec[key] = value
+          A2KM.save_kernel_json(name, spec)
+        end
+      end
+
+      command :clone do |c|
+        c.syntax = 'a2km clone <from> <to> [display_name]'
+        c.summary = 'Clone a kernelspec'
+        c.description = 'Clones kernelspec FROM to TO'
+        c.option '--user', 'Force clone to be in the user directory' \
+          ' (default is to use the same directory as FROM)'
+        c.action do |args, options|
+          options.default :user => false
+
+          if args.length < 2 or args.length > 3
+            STDERR.puts "Must specify FROM and TO"
+            exit(-1)
+          end
+          from_name = args.shift
+          to_name = args.shift
+          if args.length > 0
+            display_name = args.shift
+          else
+            display_name = to_name
+          end
+
+          from = A2KM.get_kernel(from_name)
+          from = A2KM.kernels[from_name]
+          src = from['resources_dir']
+
+          if options.user?
+            dst_dir = A2KM.user_kernel_dir
+            makedirs(dst_dir)
+          else
+            dst_dir = File.dirname src
+          end
+          dst = File.join dst_dir, to_name
+          if File.exists? dst
+            STDERR.puts "Destination already exists: #{dst}"
+            exit(-1)
+          end
+
+          puts "Cloning #{src} → #{dst}"
+          FileUtils.cp_r src, dst
+          kernel_json = File.join(dst, 'kernel.json')
+          kernelspec = File.open(kernel_json) do |f|
+            JSON.parse f.read
+          end
+          kernelspec['display_name'] = display_name
+          File.open(kernel_json, 'w') do |f|
+            f.write JSON.pretty_generate(kernelspec)
+          end
+        end
+      end
+
+      run!
+    end
+  end
+end

data/lib/a2km/env.rb

@@ -0,0 +1,104 @@
+# Copyright Min RK, License: BSD 3-clause
+
+require 'Open3'
+require 'liquid'
+
+class ENVError < StandardError
+end
+
+module A2KM
+  
+  module ENV_UTILS
+
+    ACTIVATE_TPL_S = "{{activate}} {{name}}"
+    ACTIVATE_TPL = Liquid::Template.parse ACTIVATE_TPL_S
+    ENV_CMD_TPL = Liquid::Template.parse(<<-END
+#!/usr/bin/env bash
+set -e
+#{ACTIVATE_TPL_S}
+exec python -m ipykernel $@
+END
+    )
+
+    CONDA_ACTIVATE = 'source activate'
+    VIRTUALENV_ACTIVATE = 'workon'
+    
+    module_function
+    
+    def activate_cmd(kind)
+      case kind
+      when 'conda'
+        return CONDA_ACTIVATE
+      when 'venv'
+        return VIRTUALENV_ACTIVATE
+      else
+        throw ArgumentError, "kind must be 'conda' or 'venv', not '#{kind}'"
+      end
+    end
+  
+    def in_env(env, cmd, kind: 'conda')
+      Open3::popen2('bash') do |stdin, stdout, wait_thr|
+        stdin.puts 'set -e'
+        stdin.puts ACTIVATE_TPL.render('name' => env, 'activate' => activate_cmd(kind))
+        stdin.puts cmd
+        stdin.close
+        status = wait_thr.value
+        if status != 0
+          raise ENVError.new("Failed to run #{cmd} in #{kind} env: #{env}")
+        end
+        return stdout.read
+      end
+    end
+  
+    def kernel_script(name, kind: 'conda')
+      # return kernel script as a string
+      ENV_CMD_TPL.render('name' => name, 'activate' => activate_cmd(kind))
+    end
+  
+    def ipykernel_version(env, kind: 'conda')
+      in_env(env, "python -c 'import ipykernel; print(ipykernel.__version__)'", kind: kind)
+    end
+
+    def make_kernel_exe(kernel_name, env_name, kind: 'conda', prefix: '/usr/local')
+      cmd = kernel_script(env_name, kind: kind)
+      bin = File.join(prefix, 'bin')
+      if not File.directory? bin
+        FileUtils.mkdir bin
+      end
+      kernel_exe = File.join(bin, "jupyter-kernel-#{kernel_name}")
+      puts "Making executable '#{kernel_exe}'"
+      
+      File.open(kernel_exe, 'w') do |f|
+        f.write(cmd)
+      end
+      File.chmod(0o755, kernel_exe)
+      return kernel_exe
+    end
+  
+    def get_prefix(name, kind)
+      in_env(name, 'python -c "import sys; print(sys.prefix)"', kind: kind)
+    end
+  end
+
+  def self.make_env_kernel(kernel_name, env_name, kind: 'conda', prefix: '/usr/local', user: true)
+    prefix = File.expand_path(prefix)
+    v = ENV_UTILS.ipykernel_version(env_name, kind: kind)
+    
+    puts "Found ipykernel-#{v}"
+    
+    exe = ENV_UTILS.make_kernel_exe(kernel_name, env_name, kind: kind, prefix: prefix)
+    if user
+      user_arg = '--user'
+    else
+      user_arg = ''
+    end
+    
+    ENV_UTILS.in_env(env_name, "python -m ipykernel.kernelspec --name #{kernel_name} #{user_arg}", kind: kind)
+    spec = A2KM.get_kernel_json(kernel_name)
+    spec['argv'] = [exe, '-f', '{connection_file}']
+    spec['display_name'] = "#{spec['display_name']} (#{kind}:#{env_name})"
+    A2KM.save_kernel_json(kernel_name, spec)
+    
+  end
+
+end

data/lib/a2km/version.rb

@@ -0,0 +1,5 @@
+# Copyright Min RK, License: BSD 3-clause
+
+module A2KM
+  VERSION = '0.0.2'
+end

metadata

@@ -1,15 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: a2km
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Min RK
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-08-18 00:00:00.000000000 Z
-dependencies: []
+date: 2015-09-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: commander
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: highline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: liquid
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Working with Juptyer kernels
 email: benjaminrk@gmail.com
 executables:
@@ -17,7 +59,16 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
+- LICENSE
+- README.md
+- a2km.gemspec
 - bin/a2km
+- lib/a2km.rb
+- lib/a2km/api.rb
+- lib/a2km/command.rb
+- lib/a2km/env.rb
+- lib/a2km/version.rb
 homepage: http://github.com/minrk/a2km
 licenses:
 - BSD
@@ -36,11 +87,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
-requirements:
-- commander
-- highline
+requirements: []
 rubyforge_project: 
-rubygems_version: 2.4.5
+rubygems_version: 2.4.5.1
 signing_key: 
 specification_version: 4
 summary: Assistant to the Kernel Manager