trie-storage 0.0.2

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 85e93644fb5f2752bae30d4abf768045de14729e
+  data.tar.gz: d4f3ecbc1abd9c311381ee2cd8628102ec0fb045
+SHA512:
+  metadata.gz: 403a931c9d58715bd9be69739a7c554c0f51c7cdf3a4f07bfeede3277497160d9355e5fb20c83bd92e45c6819f9f5c66a4a72d2c10286b5a19563fbe8ef0d3b5
+  data.tar.gz: c0dc8c73a86f9d6c8e24c246617d5ea51a3474d8d175c99e8043b29417ec7643f2d6dd32478f54f88ca87f5bcdc53e32fbd27d0b1d0af0a99b79542a5f94677a

data/lib/storage.rb

@@ -0,0 +1,117 @@
+require 'zip'
+class Storage
+  attr_accessor :trie
+  def initialize
+    @trie=Node.new
+  end
+
+  def add(str)
+    words = str.split(",")
+    words.each do |word|
+      @trie.add_word(word)
+    end
+  end
+
+  def print
+    puts @trie.get_subtrie_words([],"")
+  end
+
+  def contains?(str)
+    @trie.contains?(str)
+  end
+
+  def find(str)
+    if str.length<3
+      raise ArgumentError
+    end
+    st = str.dup
+    @trie.find([],str,st)
+  end
+
+  def load_from_file(filename)
+    lines=IO.readlines(filename)
+    lines.map {|line| add(line)}
+  end
+
+  def save_to_file(filename)
+    File.open(filename, "w+") do |f|
+      f.puts(@trie.get_subtrie_words([],"").join(","))
+    end
+  end
+
+  def load_from_zip(filename)
+    Zip::File.foreach(filename) do |entry|
+      istream = entry.get_input_stream
+      istream.read.split("\n").each do |line|
+        add(line)
+      end
+    end
+  end
+
+  def save_to_zip(filename)
+    Zip::File.open(filename, Zip::File::CREATE) {
+        |zipfile|
+      zipfile.get_output_stream("storage.txt") { |f| f.puts  @trie.get_subtrie_words([],"").join(",") }
+    }
+  end
+
+end
+
+class Node
+  attr_accessor :subtrie,:final
+  def initialize
+    @final = false
+    @subtrie = {}
+  end
+
+  def contains?(str)
+    let=str[0]
+    if str.empty?
+      @final
+    elsif @subtrie[let]
+      @subtrie[let].contains?(str[1..-1])
+    else
+      return false
+    end
+  end
+
+  def add_word(word)
+    let=word[0]
+    if !word.empty?
+      if !@subtrie[let]
+        @subtrie[let]=Node.new
+      end
+      @subtrie[let].add_word(word[1..-1])
+    else
+      @final=true
+    end
+  end
+
+  def get_subtrie_words(res,str)
+    @subtrie.each do |k,node|
+      if node.final
+         res.push(str+k)
+      end
+      if !node.subtrie.empty?
+        node.get_subtrie_words(res,str+k)
+      end
+    end
+    res
+  end
+
+  def find(res,str,st)
+    let=str[0]
+    if @subtrie[let]
+      if str.length==1
+        if @subtrie[let].final
+          res.push(st)
+        end
+        @subtrie[let].get_subtrie_words(res,st)
+      else
+        @subtrie[let].find(res,str[1..-1],st)
+      end
+    else
+      res
+    end
+  end
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: trie-storage
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+platform: ruby
+authors:
+- Danila Potapov
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-10-26 00:00:00.000000000 Z
+dependencies: []
+description: A simple trie storage gem
+email: potapovd36@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/storage.rb
+homepage: http://rubygems.org/gems/trie-storage
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.7
+signing_key: 
+specification_version: 4
+summary: Trie storage
+test_files: []