Class: DiscourseTheme::Uploader

Inherits:
Object
  • Object
show all
Defined in:
lib/discourse_theme/uploader.rb

Instance Method Summary collapse

Constructor Details

#initialize(dir:, api_key:, site:) ⇒ Uploader

Returns a new instance of Uploader.



3
4
5
6
7
8
 
# File 'lib/discourse_theme/uploader.rb', line 3

def initialize(dir:, api_key:, site:)
  @dir = dir
  @api_key = api_key
  @site = site
  @theme_id = nil
end

Instance Method Details

#compress_dir(gzip, dir) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'lib/discourse_theme/uploader.rb', line 10

def compress_dir(gzip, dir)
  sgz = Zlib::GzipWriter.new(File.open(gzip, 'wb'))
  tar = Archive::Tar::Minitar::Output.new(sgz)

  Dir.chdir(dir + "/../") do
    Find.find(File.basename(dir)) do |x|
      Find.prune if File.basename(x)[0] == ?.
      next if File.directory?(x)

      Minitar.pack_file(x, tar)
    end
  end
ensure
  tar.close
  sgz.close
end

#diagnose_errors(json) ⇒ Object 



27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/discourse_theme/uploader.rb', line 27

def diagnose_errors(json)
  count = 0
  json["theme"]["theme_fields"].each do |row|
    if (error = row["error"]) && error.length > 0
      if count == 0
        puts
      end
      count += 1
      puts
      puts "Error in #{row["target"]} #{row["name"]}: #{row["error"]}"
      puts
    end
  end
  count
end

#upload_full_theme ⇒ Object 



79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
 
# File 'lib/discourse_theme/uploader.rb', line 79

def upload_full_theme
  filename = "#{Pathname.new(Dir.tmpdir).realpath}/bundle_#{SecureRandom.hex}.tar.gz"
  compress_dir(filename, @dir)

  # new full upload endpoint
  uri = URI.parse(@site + "/admin/themes/import.json?api_key=#{@api_key}")
  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = URI::HTTPS === uri
  File.open(filename) do |tgz|

    request = Net::HTTP::Post::Multipart.new(
      uri.request_uri,
      "bundle" => UploadIO.new(tgz, "application/tar+gzip", "bundle.tar.gz"),
    )
    response = http.request(request)
    if response.code.to_i == 201
      json = JSON.parse(response.body)
      @theme_id = json["theme"]["id"]
      if diagnose_errors(json) == 0
        puts "(done)"
      end
    else
      puts "Error importing theme status: #{response.code}"

      puts response.body
    end
  end

ensure
  FileUtils.rm_f filename
end

#upload_theme_field(target:, name:, type_id:, value:) ⇒ Object 



43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/discourse_theme/uploader.rb', line 43

def upload_theme_field(target: , name: , type_id: , value:)

  raise "expecting theme_id to be set!" unless @theme_id

  args = {
    theme: {
      theme_fields: [{
        name: name,
        target: target,
        type_id: type_id,
        value: value
      }]
    }
  }

  uri = URI.parse(@site + "/admin/themes/#{@theme_id}?api_key=#{@api_key}")

  http = Net::HTTP.new(uri.host, uri.port)
  http.use_ssl = URI::HTTPS === uri

  request = Net::HTTP::Put.new(uri.request_uri, 'Content-Type' => 'application/json')
  request.body = args.to_json

  http.start do |h|
    response = h.request(request)
    if response.code.to_i == 200
      json = JSON.parse(response.body)
      if diagnose_errors(json) == 0
        puts "(done)"
      end
    else
      puts "Error importing field status: #{response.code}"
    end
  end
end