password_genie 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ffac553f7f831402019caa9247f7c41d42c8fe94
-  data.tar.gz: 31f5881099c8ef55d1304216407ed476673f365a
+  metadata.gz: 410cb42e5da2ca1c2477fa5d5da8c55ca360ecca
+  data.tar.gz: 116a44efaaadb730961aaf74455c12b815db1b05
 SHA512:
-  metadata.gz: 2dd3e0802621da04431927662d97fd835161228d54a605e42a1102fedaf5a33c1625cada326bcf90510c6a9c336034d6048ef47a2e6aac918fabf5b694863647
-  data.tar.gz: e31dd89b143e58e72b1dc2a98a7ac387ba85cc317b4f158d1bdf229b891b2423db18a49e89cb7069e0b3d1dd455a417fd636c56fe2c91a96d4402e0a37bf583a
+  metadata.gz: 7de69552b0b12f7a4ba966fcf0f7a27e0546485732217a46bacf60c3ce8324deabd8cc1ae916af2a0cd82b54e0e93b69cc411cedd9270910834b295aae9a1253
+  data.tar.gz: 3167f2fc90b401b42b1e606f6c83ff86a9d620eb9112052d99b0ec181d8bd6215e8d7b8338ed70d002fd415cd7c9ced119e75522763d517f6ec4bf1bdaff5d45

data/lib/password_genie.rb

@@ -1,10 +1,15 @@
 #!/usr/bin/env ruby
 #password_genie.rb
 
+require 'sqlite3'
+
 class WordBank
 	attr_reader :word_bank, :number, :special_chars
 	
 	def word_bank=(words)
+		if words == ""
+			raise "can't be empty."
+		end
 		word_bank= words.split("").to_a
 		word_bank.delete(" ")
 		unless word_bank.is_a?(Array)
@@ -30,7 +35,7 @@ class WordBank
 	def initialize(words,number=8)
 		self.word_bank = words
 		self.number = number
-		self.special_chars = special_chars
+		self.special_chars = special_chars.to_a
 		@password_ary = []
 		@numbers= (0..9).to_a
 	end
@@ -46,13 +51,17 @@ class WordBank
 	end
 	
 	def create
+		puts "wordbank: #{@word_bank}, special #{@special_chars}"
 		i = number
-		while i != 0
-			if i == 2 && word_bank.include?(@special_chars) && @password_ary.include?(@special_chars) == false
+		puts "create"
+		until i == 0
+			puts " i is #{i}"
+			if i == 2 && (@password_ary.include?(@special_chars) == false && @word_bank.include?(@special_chars))
 					@password_ary << @special_chars[rand(0..@special_chars.size)]
 					puts "1 == 2"
-			elsif i == 1 && word_bank.include?(@numbers) && @password_ary.include?(@numbers) == false
-					@password_ary << @numbers[rand(1..@numbers.size)]
+			end
+			if i == 1 && (@password_ary.include?(@numbers) == false && @word_bank.include?(@numbers))
+					@password_ary << @numbers[rand(0..@numbers.size)]
 					puts "i == 1"
 			end
 			letter = rand(0..word_bank.size)
@@ -65,25 +74,64 @@ class WordBank
 		@password
 	end
 	
-	def save_info(site,username)
-		myfile=File.open('site_info.txt','a+') do |f|
-			puts "[#{site}] username: #{username} | password: #{@password}"
-			f.puts("[#{site}] username: #{username} | password: #{@password}")
+	def save_info(site,username,password=@password)
+		begin
+			db = SQLite3::Database.open('genie.db')
+			puts db.get_first_value "select SQLite_VERSION()"
+			db.results_as_hash = true
+			site_in = site; username_in = username; pw_in = password
+			db.transaction
+			db.execute "create table if not exists site_info(Id INTEGER PRIMARY KEY, Site TEXT, Username TEXT, Password TEXT)"
+			db_in = db.prepare "insert into site_info(Site, Username, Password) values(:site_in, :username_in, :pw_in)"
+			db_in.execute site_in, username_in, pw_in
+			db.commit
+		rescue SQLite3::Exception => e
+			puts "something went wrong: #{e}"
+			db.rollback
+		ensure
+			db_in.close if db_in
+			db.close if db
 		end
 	end
 
 	def find_info(info)
-		return "please create a password directory first" unless File.exist?('site_info.txt')
-			IO.foreach('site_info.txt') do |line|
-				return line if line.include?(info)
-				end
-			return "no match"
+		begin
+			db = SQLite3::Database.open('genie.db')
+			puts db.get_first_value "select SQLite_VERSION()"
+			return "please create a directory first" unless File.exist?('genie.db')
+			puts "hello, again."
+			puts info
+			print_out = db.execute2 "SELECT * FROM site_info WHERE Site= :info_in OR Username = :info", info
+			puts "foo"
+			return "no match" unless print_out != nil
+			puts "bar"
+			print_out.each do |line|
+				puts "[%5s] %8s | %s" % [line[1], line[2], line[3]]
+			end
+			puts "boo"
+		rescue SQLite3::Exception => e
+			puts "foo2"
+			puts e
+		ensure
+			db.close if db
+		end
 	end
 	
-	def push_info(site,username,password)
-		my_file = File.open('site_info.txt', 'a+') do |f|
-			f.puts("[#{site}] username: #{username} | password: #{password}")
-			end
+	def add_or_replace_info(site, username, password)
+		begin
+			db = SQLite3::Database.open('genie.db')
+			return "please set up database by restarting and choosing [1]" unless File.file?('genie.db')
+			puts db.get_first_value "select SQLite_VERSION()"
+			db.transaction
+			db.execute2 "UPDATE site_info SET Password = :password WHERE Site = :site AND Username = :username", password, site, username 
+			db.commit
+			puts "you made #{db.changes} changes."
+		rescue SQLite3::Exception => e
+			puts e
+			db.rollback
+		ensure
+			db.close if db
+		end
 	end
 end
 
@@ -91,7 +139,7 @@ go = true
 
 while go
 	puts "enter [1] to CREATE a new password\nenter [2] to LOOKUP a password"
-	puts "enter [3] to ADD an already existing password\nenter [4] to EXIT"
+	puts "enter [3] to ADD/CHANGE an already existing password\nenter [4] to EXIT"
 	
 	choice = gets.chomp.to_i
 	
@@ -100,6 +148,12 @@ while go
 		puts "How long do you want your password to be?"
 		
 		number = gets.chomp.to_i
+		
+		unless number.is_a?(Integer)
+			puts "number MUST be an integer."
+			next
+		end
+		
 		if number == 1
 			word_number = "character"
 		else
@@ -144,14 +198,19 @@ while go
 		puts x.find_info(word)
 		next
 	elsif choice == 3
+		puts "**MUST supply an EXISTING site and username**"
 		print "enter site: "
 		site = gets.chomp
 		print "enter username: "
 		username = gets.chomp
+		if site == '' || username == ''
+			puts "BOTH site and username must be filled in. Back to the beginning."
+			next
+		end
 		print "enter password: "
 		password = gets.chomp
 		x = WordBank.new(site)
-		x.push_info(site,username,password)
+		x.add_or_replace_info(site,username,password)
 		next
 	elsif choice == 4
 		go = false

metadata

@@ -1,18 +1,18 @@
 --- !ruby/object:Gem::Specification
 name: password_genie
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - fookh
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-12-01 00:00:00.000000000 Z
+date: 2019-02-23 00:00:00.000000000 Z
 dependencies: []
-description: a simple tool to build a password based on user defined parameters 2.
-  archive passwords, usernames, and corresponding site data 3. search data and 4.
-  push an existing password into txt file
+description: a simple tool to build a password repository based on user defined parameters
+  2. archive passwords, usernames, and corresponding site data 3. search data and
+  4. push an existing password into sqlite
 email: foo@yetispeak.com
 executables: []
 extensions: []
@@ -42,5 +42,5 @@ rubyforge_project:
 rubygems_version: 2.5.2.1
 signing_key: 
 specification_version: 4
-summary: introduces a puts function for CURRENT passwords
+summary: migrated over to sqlite
 test_files: []