git.squeep.com Git - akkoma/blob - lib/mix/tasks/pleroma/emoji.ex

   1 # Pleroma: A lightweight social networking server
   2 # Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/>
   3 # SPDX-License-Identifier: AGPL-3.0-only
   4
   5 defmodule Mix.Tasks.Pleroma.Emoji do
   6   use Mix.Task
   7
   8   @shortdoc "Manages Pleroma instance"
   9   @moduledoc """
  10   """
  11
  12   @default_manifest "https://git.pleroma.social/vaartis/emoji-index/raw/master/index.json"
  13
  14   defp fetch_manifest(from) do
  15     Tesla.get!(from).body |> Poison.decode!()
  16   end
  17
  18   defp parse_global_opts(args) do
  19     OptionParser.parse(
  20       args,
  21       strict: [
  22         manifest: :string
  23       ],
  24       aliases: [
  25         m: :manifest
  26       ]
  27     )
  28   end
  29
  30   def run(["ls-packs" | args]) do
  31     Application.ensure_all_started(:hackney)
  32
  33     {options, [], []} = parse_global_opts(args)
  34
  35     manifest =
  36       fetch_manifest(if options[:manifest], do: options[:manifest], else: @default_manifest)
  37
  38     Enum.each(manifest, fn {name, info} ->
  39       to_print = [
  40         {"Name", name},
  41         {"Homepage", info["homepage"]},
  42         {"Description", info["description"]},
  43         {"License", info["license"]},
  44         {"Source", info["src"]}
  45       ]
  46
  47       for {param, value} <- to_print do
  48         IO.puts(IO.ANSI.format([:bright, param, :normal, ": ", value]))
  49       end
  50     end)
  51   end
  52
  53   def run(["get-packs" | args]) do
  54     Application.ensure_all_started(:hackney)
  55
  56     {options, pack_names, []} = parse_global_opts(args)
  57
  58     manifest_url = if options[:manifest], do: options[:manifest], else: @default_manifest
  59
  60     manifest = fetch_manifest(manifest_url)
  61
  62     for pack_name <- pack_names do
  63       if Map.has_key?(manifest, pack_name) do
  64         pack = manifest[pack_name]
  65         src_url = pack["src"]
  66
  67         IO.puts(
  68           IO.ANSI.format([
  69             "Downloading ",
  70             :bright,
  71             pack_name,
  72             :normal,
  73             " from ",
  74             :underline,
  75             src_url
  76           ])
  77         )
  78
  79         binary_archive = Tesla.get!(src_url).body
  80         archive_md5 = :crypto.hash(:md5, binary_archive) |> Base.encode16()
  81
  82         md5_status_text = ["MD5 of ", :bright, pack_name, :normal, " source file is ", :bright]
  83         if archive_md5 == String.upcase(pack["src_md5"]) do
  84           IO.puts(IO.ANSI.format(md5_status_text ++ [:green, "OK"]))
  85         else
  86           IO.puts(IO.ANSI.format(md5_status_text ++ [:red, "BAD"]))
  87
  88           raise "Bad MD5 for #{pack_name}"
  89         end
  90
  91         # The url specified in files should be in the same directory
  92         files_url = Path.join(Path.dirname(manifest_url), pack["files"])
  93
  94         IO.puts(
  95           IO.ANSI.format([
  96             "Fetching the file list for ",
  97             :bright,
  98             pack_name,
  99             :normal,
 100             " from ",
 101             :underline,
 102             files_url
 103           ])
 104         )
 105
 106         files = Tesla.get!(files_url).body |> Poison.decode!()
 107
 108         IO.puts(IO.ANSI.format(["Unpacking ", :bright, pack_name]))
 109
 110         static_path = Path.join(:code.priv_dir(:pleroma), "static")
 111
 112         pack_path =
 113           Path.join([
 114             static_path,
 115             Pleroma.Config.get!([:instance, :static_dir]),
 116             "emoji",
 117             pack_name
 118           ])
 119
 120         files_to_unzip =
 121           Enum.map(
 122             files,
 123             fn {_, f} -> to_charlist(f) end
 124           )
 125
 126         {:ok, _} =
 127           :zip.unzip(binary_archive,
 128             cwd: pack_path,
 129             file_list: files_to_unzip
 130           )
 131
 132         IO.puts(IO.ANSI.format(["Writing emoji.txt for ", :bright, pack_name]))
 133
 134         emoji_txt_str =
 135           Enum.map(
 136             files,
 137             fn {shortcode, path} ->
 138               "#{shortcode}, /instance/static/emoji/#{pack_name}/#{path}"
 139             end
 140           )
 141           |> Enum.join("\n")
 142
 143         File.write!(Path.join(pack_path, "emoji.txt"), emoji_txt_str)
 144       else
 145         IO.puts(IO.ANSI.format([:bright, :red, "No pack named \"#{pack_name}\" found"]))
 146       end
 147     end
 148   end
 149 end