111 lines
		
	
	
	
		
			2.9 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
			
		
		
	
	
			111 lines
		
	
	
	
		
			2.9 KiB
		
	
	
	
		
			Elixir
		
	
	
	
	
	
| # Pleroma: A lightweight social networking server
 | |
| # Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
 | |
| # SPDX-License-Identifier: AGPL-3.0-only
 | |
| 
 | |
| defmodule Pleroma.Frontend do
 | |
|   alias Pleroma.Config
 | |
| 
 | |
|   require Logger
 | |
| 
 | |
|   def install(name, opts \\ []) do
 | |
|     frontend_info = %{
 | |
|       "ref" => opts[:ref],
 | |
|       "build_url" => opts[:build_url],
 | |
|       "build_dir" => opts[:build_dir]
 | |
|     }
 | |
| 
 | |
|     frontend_info =
 | |
|       [:frontends, :available, name]
 | |
|       |> Config.get(%{})
 | |
|       |> Map.merge(frontend_info, fn _key, config, cmd ->
 | |
|         # This only overrides things that are actually set
 | |
|         cmd || config
 | |
|       end)
 | |
| 
 | |
|     ref = frontend_info["ref"]
 | |
| 
 | |
|     unless ref do
 | |
|       raise "No ref given or configured"
 | |
|     end
 | |
| 
 | |
|     dest = Path.join([dir(), name, ref])
 | |
| 
 | |
|     label = "#{name} (#{ref})"
 | |
|     tmp_dir = Path.join(dir(), "tmp")
 | |
|     IO.puts("Downloading #{label}...")
 | |
| 
 | |
|     with {_, :ok} <-
 | |
|            {:download_or_unzip, download_or_unzip(frontend_info, tmp_dir, opts[:file])},
 | |
|          IO.puts("Installing #{label} to #{dest}"),
 | |
|          :ok <- install_frontend(frontend_info, tmp_dir, dest) do
 | |
|       File.rm_rf!(tmp_dir)
 | |
|       IO.puts("Frontend #{label} installed to #{dest}")
 | |
|     else
 | |
|       {:download_or_unzip, _} ->
 | |
|         IO.puts("Could not download or unzip the frontend")
 | |
|         {:error, "Could not download or unzip the frontend"}
 | |
| 
 | |
|       _e ->
 | |
|         IO.puts("Could not install the frontend")
 | |
|         {:error, "Could not install the frontend"}
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def dir(opts \\ []) do
 | |
|     if is_nil(opts[:static_dir]) do
 | |
|       Pleroma.Config.get!([:instance, :static_dir])
 | |
|     else
 | |
|       opts[:static_dir]
 | |
|     end
 | |
|     |> Path.join("frontends")
 | |
|   end
 | |
| 
 | |
|   defp download_or_unzip(frontend_info, temp_dir, nil),
 | |
|     do: download_build(frontend_info, temp_dir)
 | |
| 
 | |
|   defp download_or_unzip(_frontend_info, temp_dir, file) do
 | |
|     with {:ok, zip} <- File.read(Path.expand(file)) do
 | |
|       unzip(zip, temp_dir)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   def unzip(zip, dest) do
 | |
|     with {:ok, unzipped} <- :zip.unzip(zip, [:memory]) do
 | |
|       File.rm_rf!(dest)
 | |
|       File.mkdir_p!(dest)
 | |
| 
 | |
|       Enum.each(unzipped, fn {filename, data} ->
 | |
|         path = filename
 | |
| 
 | |
|         new_file_path = Path.join(dest, path)
 | |
| 
 | |
|         new_file_path
 | |
|         |> Path.dirname()
 | |
|         |> File.mkdir_p!()
 | |
| 
 | |
|         File.write!(new_file_path, data)
 | |
|       end)
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   defp download_build(frontend_info, dest) do
 | |
|     Logger.info("Downloading pre-built bundle for #{frontend_info["name"]}")
 | |
|     url = String.replace(frontend_info["build_url"], "${ref}", frontend_info["ref"])
 | |
| 
 | |
|     with {:ok, %{status: 200, body: zip_body}} <-
 | |
|            Pleroma.HTTP.get(url, [], recv_timeout: 120_000) do
 | |
|       unzip(zip_body, dest)
 | |
|     else
 | |
|       {:error, e} -> {:error, e}
 | |
|       e -> {:error, e}
 | |
|     end
 | |
|   end
 | |
| 
 | |
|   defp install_frontend(frontend_info, source, dest) do
 | |
|     from = frontend_info["build_dir"] || "dist"
 | |
|     File.rm_rf!(dest)
 | |
|     File.mkdir_p!(dest)
 | |
|     File.cp_r!(Path.join([source, from]), dest)
 | |
|     :ok
 | |
|   end
 | |
| end
 | 
