adieush 1.0 → 1.1

data/lib/adieush.rb

@@ -1,5 +1,127 @@
+require 'zip/zip'
 class Adieush
-  def hello
-    p 'hello world'
+  attr_accessor :storage_array, :search_words
+
+  def initialize
+    @storage_array = {}
+    @search_words = []
+  end
+
+  def load_string(init_string)
+    input_array = init_string.split(',')
+    input_array.each do |store|
+      add(store)
+    end
+  end
+
+  def load_from_file(filename)
+    file_str = ''
+    File.open(filename, 'r') {|f|
+      file_str = f.read
+    }
+    load_string(file_str)
+  end
+
+  def save_to_file(filename)
+    input_array = output_words(@storage_array)
+    input_string = ''
+    input_array.each do |word|
+      input_string += "#{word},"
+    end
+    File.open(filename, 'w') {|f|
+      f.print input_string
+    }
+  end
+
+  def save_to_zip(zip_name)
+    input_array = output_words(@storage_array)
+    input_string = ''
+    input_array.each do |word|
+      input_string += "#{word},"
+    end
+    Zip::ZipFile.open(zip_name, Zip::ZipFile::CREATE) {
+        |zipfile|
+      zipfile.get_output_stream("storage") { |f| f.print input_string }
+    }
+  end
+
+  def load_from_zip(zip_name)
+    Zip::ZipFile.open(zip_name) do |zipfile|
+      zipfile.each do |entry|
+        file_string = entry.get_input_stream.read
+        load_string(file_string)
+      end
+    end
+  end
+
+  def add(string)
+    if contains(string)
+      p  "'#{string}' is already in the vocabulary"
+      return
+    end
+    prev_node = @storage_array
+    i = 0
+    string_size = string.size
+    string.each_char do |c|
+      current_node = prev_node
+      if  !current_node.has_key?(c)
+        current_node[c] = {}
+      end
+      prev_node = current_node[c]
+      i+=1
+      if string_size == i
+        current_node[c]['value'] = string
+        p "New word '#{string}' added"
+      end
+    end
+  end
+
+  def contains(string)
+    prev_node = @storage_array
+    i = 0
+    string_size = string.size
+    string.each_char do |c|
+      current_node = prev_node
+      if  !current_node.has_key?(c)
+        return false
+      end
+      prev_node = current_node[c]
+      i+=1
+      if string_size == i
+        if current_node[c]['value'] == string
+          return true
+        end
+      end
+    end
+  end
+
+  def find(string)
+    string_size = string.size
+    if string_size > 2
+      prev_node = @storage_array
+      i = 0
+      string.each_char do |c|
+        current_node = prev_node
+        if  !current_node.has_key?(c)
+          return false
+        end
+        prev_node = current_node[c]
+        i+=1
+        if string_size == i
+          return output_words(current_node[c])
+        end
+      end
+    end
+  end
+
+  def output_words(node)
+    node.keys.each do |child|
+      if child == 'value'
+        @search_words << node[child]
+      else
+        output_words(node[child])
+      end
+    end
+    return @search_words
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: adieush
 version: !ruby/object:Gem::Version
-  version: '1.0'
+  version: '1.1'
   prerelease: 
 platform: ruby
 authors: