Contents

# encoding: utf-8

require 'csv'
class DbWords
  attr_reader :words

  def initialize(argv=nil)
    @argv = argv
    @words = Array.new
    @file = File.join(Dir.home, DATABASE_DIR, "#{DATABASE}.csv")
    create_db unless File.exist?(@file)
  end

  #collecion of words
  def collection
    col = CSV.read(@file)
    col
  end

  #adding word in db
  def add
    split
    if double?
      puts "You have '#{@words[0]}' in database"
      exit
    end

    CSV.open(@file, "a") do |csv|
      csv << [@words[0], @words[1]]
    end
    puts "Word is added"
  end

  #Destroy a word from db file
  def destroy
    list_w = collection
    @argv.map(&:to_i).each do |i|
      list_w.delete_at(i-1)
    end

    CSV.open(@file, "w") do |csv|
      list_w.each do |row|
        csv << row
      end
    end

    puts @argv.join(" ") + " deleted"
    list
  end

  def update(index, updated_row)
    list_w = collection

    CSV.open(@file, "w") do |csv|
      list_w.each_with_index do |row, idx|
        if index == idx
          csv << updated_row
        else
          csv << row
        end
      end
    end

    GC.start
  end

  #Print the list of words
  def list
    list_of_words do |word, idx|
      puts "#{idx + 1} #{word[0]} | #{word[1]} | #{word[2]}"
    end
  end

  private
  # input is ["to", "run:", "бежать"]
  # output is ["to run", "бежать"]
  def split
    w = @argv.join(" ").split(":")
    if w.size < 2
      pp 'You mast set ":" on word. Like this: "jword -a to run: бежать"'
    end
    @words = w.map(&:strip)
  end

  #return true if have double in db
  def double?
    w = @words.first
    list = collection
    list.map(&:first).include?( w )
  end

  def list_of_words
    collection.each_with_index do |word, idx|
      yield(word, idx)
    end
  end

  def create_db
    Dir.mkdir(File.join(Dir.home, DATABASE_DIR), 0700)
    File.new(@file, File::RDWR|File::CREAT, 0644)
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
jumping_words-0.2.2	lib/jwords/db_words.rb
jumping_words-0.2.1	lib/jwords/db_words.rb