Class: Warehaus::Getter

Inherits:
Object
  • Object
show all
Includes:
HTTParty
Defined in:
lib/warehaus.rb

Constant Summary collapse

ENTITY_PATH = 
'/3dw/GetEntity'
KMZ_PATH = 
'/warehouse/getpubliccontent'

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(url_or_id, dir = './', name = 'warehouse-model') ⇒ Getter

Returns a new instance of Getter.



18
19
20
21
22
23
24
25
26
27
 
# File 'lib/warehaus.rb', line 18

def initialize(url_or_id, dir='./', name='warehouse-model')
  @name = name
  @path = dir
  @model_path = @path + '/' + @name
  @options = {
    :query => {
      :id => parse_input(url_or_id)
    }
  }
end

Class Method Details

.from_hash(hash) ⇒ Object 



134
135
136
137
138
139
140
 
# File 'lib/warehaus.rb', line 134

def self.from_hash(hash)
  hash[:models].each do |k, v|

    getter = self.new(v, "#{hash[:dir]}", k.to_s)
    getter.unbox
  end
end

Instance Method Details

#cleanup ⇒ Object 



106
107
108
109
 
# File 'lib/warehaus.rb', line 106

def cleanup
  log("🛀  Cleaning up")
  FileUtils.rm_rf("#{@model_path}/.tmp")
end

#download_kmz ⇒ Object 



64
65
66
67
 
# File 'lib/warehaus.rb', line 64

def download_kmz
  log("📥  Downloading KMZ file")
  @kmz = self.class.get(KMZ_PATH, @kmz_options).parsed_response
end

#erase ⇒ Object 



122
123
124
 
# File 'lib/warehaus.rb', line 122

def erase
  FileUtils.rm_rf "#{@model_path}"
end

#fetch_entity ⇒ Object 



33
34
35
36
 
# File 'lib/warehaus.rb', line 33

def fetch_entity
  log("📠  Fetching JSON representation of model")
  @response = JSON.parse(self.class.get(ENTITY_PATH, @options), :symbolize_names => true)
end

#get_kmz_id ⇒ Object 



38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
 
# File 'lib/warehaus.rb', line 38

def get_kmz_id
  log("🔍  Checking for KMZ id in JSON")

  if !@response[:binaries] 
    return log("No KMZ file available for this object 💣")
  end
    
  if @response[:binaries][:ks]
    kmz_id = @response[:binaries][:ks][:id]
  else
    @response[:binaries].each{ |k, bin|
      kmz_id = bin[:id] if (bin[:originalFileName] =~ /kmz$/) != nil
    }
    if !kmz_id
      return log("No KMZ file available for this object 💣")
    end
  end

  @kmz_options = {
    :query =>{
      :contentId => kmz_id,
      :fn => "#{@name}.kmz"
    }
  }
end

#log(msg) ⇒ Object 



130
131
132
 
# File 'lib/warehaus.rb', line 130

def log(msg)
  puts("Warehaus: #{msg}") if $mode == "verbose"
end

#parse_input(input) ⇒ Object 



29
30
31
 
# File 'lib/warehaus.rb', line 29

def parse_input(input)
  (input =~ /^[\w|\-|\d]+$/) ? input : input.match(/id=([\w|\-|\d]+)/)[1]
end

#raise_error(msg) ⇒ Object 



126
127
128
 
# File 'lib/warehaus.rb', line 126

def raise_error(msg)
  raise "Warehaus Error: #{msg}"
end

#unbox ⇒ Object 



111
112
113
114
115
116
117
118
119
120
 
# File 'lib/warehaus.rb', line 111

def unbox
  log("📦  Beginning unbox!")
  fetch_entity
  get_kmz_id
  download_kmz
  write_kmz
  unzip_kmz
  cleanup
  "#{@path}/#{@name}/"
end

#unzip_kmz ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/warehaus.rb', line 82

def unzip_kmz
  log("📁  Creating model directory")
  FileUtils::mkdir_p "#{@model_path}/untitled"

  log("🎊  Cracking open the ZIP file")
  Zip::File.open("#{@model_path}/.tmp/#{@name}.zip") do |zip_file|
    
    valid_paths = zip_file.entries.select do |entry|
      (entry.name =~ /models\//) != nil
    end
    
    valid_paths.each do |entry|

      if entry.name =~ /\.dae$/
        dest = "#{@model_path}/#{@name}.dae"
      else
        dest = "#{@model_path}/#{entry.name.match(/models\/(.*)$/)[1]}"
      end

      entry.extract(dest)
    end
  end
end

#write_kmz ⇒ Object 



69
70
71
72
73
74
75
76
77
78
79
80
 
# File 'lib/warehaus.rb', line 69

def write_kmz
  FileUtils::mkdir_p "#{@model_path}/.tmp"
  log("📝  Writing KMZ file")
  begin
    File.open("#{@model_path}/.tmp/#{@name}.kmz", "w+") do |f|
      f.write @kmz
    end
    FileUtils::mv "#{@model_path}/.tmp/#{@name}.kmz", "#{@model_path}/.tmp/#{@name}.zip"
  rescue Exception => e
    return raise_error e.message
  end
end