mongogems 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 76e7dbd3e0694f3ad409e13c3530ed907a4d828281bec0f3f964e63cf3cc706c
-  data.tar.gz: c59ee55d845c14dac46e17d6dbcf0ba10b2242d9623839c94b212c92b9f7499a
+  metadata.gz: d6b110efba52c4efdfb689e7af26b0e97870e13ac53c03624649c4f4b8f3c7a2
+  data.tar.gz: eec7fa33083d44430f3f9b2d4934ba1bea15ba714284fb07b58c3852dbb24a59
 SHA512:
-  metadata.gz: 941bf504d23cb010f8c21293c161788cb45d38e91e85aaaa3bd4dc2c9f5c70c9be0c240823c363889645d8f7d416af92dc380823706a6560a445bb2dc2458918
-  data.tar.gz: b9616a4cfd7458dc0bb1deb57b29c4dae8becff446fbef902fab8f00f60e4ce8eec616f70279c4393a9bc1c635b8a9e9fc862cb27a9ff825ca7416922824cd0e
+  metadata.gz: 6d93488d894bc85eb43b3713b98b00a8d13cc448c60649ecad754efab53c61ccbc99534d1b77da74de3e1a36b3c79e70ef24b0bef15c36585724cbef798df180
+  data.tar.gz: 2a26b5a049cba6a60519d8045dae7edba880308a236eec7405fee626544008f466c17b152cfc4be51f6d913c27739f922e3de97b10fc78db85e714f74dad7455

data/bin/mg

@@ -0,0 +1,69 @@
+#!/usr/bin/env ruby
+require 'optparse'
+require_relative '../lib/config'
+require 'shellwords'
+
+options = {}
+
+case ARGV.shift
+when 'config'
+  options[:cmd] = 'config'
+  case ARGV.shift
+  when 'set'
+    options[:subcmd] = 'set'
+    key = ARGV.shift
+    value = ARGV.join ' '
+    options[:key] = key
+    options[:val] = value
+
+    write_config_entry options[:key], options[:val]
+  when 'get'
+    options[:subcmd] = 'get'
+    key = ARGV.shift
+    options[:key] = key
+
+    val = read_config_entry options[:key]
+    puts "Config for #{options[:key]} is #{val}"
+  when 'help'
+    puts '
+Usage: mg config [subcommand] ...
+Commands:
+ get KEY
+ set KEY VALUE
+'
+    exit
+  else
+    puts "Unrecognized #{options[:cmd]} command"
+    exit -1
+  end
+when 'help', '--help', '-h', '-?'
+  puts '
+Usage: mg [command] [subcommand] ...
+Commands:
+ config
+ help
+'
+  exit
+when 'shell', 's'
+  options[:cmd] = 'shell'
+  m_uri = read_config_entry 'uri'
+  if m_uri.nil?
+    puts 'URI not found in config.
+Must set URI with: mg config set uri YOUR_URI_HERE
+Example: mg config set uri \'mongodb://username:password@localhost:27017\'
+'
+    exit
+  end
+  remaining_cmds = ARGV.map{|arg| Shellwords.escape arg}.join ' '
+  cmdstr = "mongo #{m_uri} #{remaining_cmds}"
+  exec(cmdstr)
+# when 'echo'
+#   remaining_cmds = ARGV.map{|arg| Shellwords.escape arg}.join ' '
+#   puts remaining_cmds
+#   exec("echo #{remaining_cmds}")
+else
+  puts 'Unrecognized command'
+  exit -1
+end
+
+

data/lib/config.rb

@@ -0,0 +1,31 @@
+require 'yaml'
+
+$config_filename = File.join(Dir.home, '.mongogems')
+$has_config = lambda { File.exists? $config_filename }
+
+# Write a key-value pair to the config file
+def write_config_entry(key, value)
+  if $has_config.call then
+    curr_config = YAML.load(File.read($config_filename))
+  else
+    curr_config = {}
+  end
+
+  curr_config[key] = value
+
+  File.open($config_filename, 'w') { |f| f.write(curr_config.to_yaml) }
+end
+
+# Reads a key-value pair from the config file
+def read_config_entry(key)
+  if $has_config.call then
+    curr_config = YAML.load(File.read($config_filename))
+  else
+    return nil
+  end
+
+  curr_config[key]
+end
+
+# write_config_entry 'mongouri', 'mongodb://localhost'
+# p (read_config_entry 'mongouri')

data/lib/indexes.rb

@@ -4,7 +4,6 @@ require 'json'
 def get_indexes_script(uri, db_name=nil, coll_name=nil, outfile)
   client = Mongo::Client.new(uri)
 
-
   db_list = []
   if not db_name.nil?
     db_list.push db_name
@@ -15,18 +14,11 @@ def get_indexes_script(uri, db_name=nil, coll_name=nil, outfile)
     end
   end
 
-  # client = client.use 'test'
-
-  # cur = client.command listCollections: 1
-  # rset = cur.to_a
-  # rset = rset.first['cursor']['firstBatch']
-  # rset
-
   scripts = []
 
   db_list.each do |iter_db|
     client = client.use iter_db
-    scripts.push "use #{iter_db};"
+    scripts.push "db = db.getSiblingDB('#{iter_db}')';"
     if coll_name.nil?
       colls = client.collections
     else
@@ -70,6 +62,3 @@ def gen_idx_script(idIndex, collname)
   opt_spec = JSON.dump idx_options
   txt = "db.#{collname}.createIndex(#{idx_spec}, #{opt_spec});"
 end
-
-# mdb_uri = 'mongodb://localhost'
-# get_indexes_script mdb_uri, nil, nil, 'idx_out.txt'

data/lib/savedPipeline.rb

@@ -1,8 +1,7 @@
 require 'json'
-# require 'hjson'
 
-# filename = '/Users/nitin/Downloads/Zaloni-DQ-A/607d7460cb21963a143d1d0f.json'
 
+# Exports the pipeline to a .aggregate() Mongo shell script
 def pipeline_to_mongosh_script(infile, outfile)
   txt = File.read(infile)
   data = JSON.parse(txt)
@@ -42,6 +41,7 @@ db.getSiblingDB('#{dbname}').#{collname}.aggregate(
 end
 
 
+# Exports the pipeline to a createView Mongo shell script
 def pipeline_to_view_mongosh_script(infile, viewname, outfile)
   txt = File.read(infile)
   data = JSON.parse(txt)
@@ -78,5 +78,3 @@ db.getSiblingDB('#{dbname}').createView('#{viewname}', '#{collname}',
 
   File.write(outfile, mongosh_script_out)
 end
-
-# pipeline_to_mongosh_script filename, '/dev/null'

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mongogems
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Katkam Nitin Reddy
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-05-11 00:00:00.000000000 Z
+date: 2021-05-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mongo
@@ -24,18 +24,49 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 2.14.0
+- !ruby/object:Gem::Dependency
+  name: os
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.1.1
+- !ruby/object:Gem::Dependency
+  name: shellwords
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.1.0
 description:
 email:
 executables:
 - compass2mongosh
 - mongolog2file
 - mongoindex2file
+- mg
 extensions: []
 extra_rdoc_files: []
 files:
 - bin/compass2mongosh
+- bin/mg
 - bin/mongoindex2file
 - bin/mongolog2file
+- lib/config.rb
 - lib/indexes.rb
 - lib/logs.rb
 - lib/savedPipeline.rb