arke

# Copyright 2023 Arkemis S.r.l.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

defmodule Arke.Core.File do
  @moduledoc """
  Defines a file that can be used to store data
  """

  use Arke.System
  alias Arke.Utils.Gcp
  alias Arke.Boundary.ArkeManager

  arke id: :arke_file, label: "Arke file" do
    parameter(:name, :string, required: true)
    parameter(:path, :string, required: true)
    parameter(:provider, :string, values: ["local", "gcloud", "aws"], default: "gcloud")
    parameter(:size, :float, required: false)
    parameter(:extension, :string, required: false)
    parameter(:binary, :bynary, required: true, only_runtime: true)
  end

  def before_load(
        %{path: path, content_type: content_type, filename: filename} = _,
        :create
      ) do
    {:ok, file_stat} = File.stat(path)
    extension = Path.extname(filename)
    {:ok, binary} = File.read(path)
    path = "arke_file/#{DateTime.to_string(DateTime.utc_now())}"

    unit_data = %{
      binary: binary,
      extension: extension,
      size: file_stat.size,
      provider: "gcloud",
      path: path,
      name: filename
    }

    {:ok, unit_data}
  end

  def before_load(opts, _persistence_fn), do: {:ok, opts}

  def on_struct_encode(arke, unit, data, opts) do
    load_files = Keyword.get(opts, :load_files, false)

    case load_files do
      false -> {:ok, data}
      true -> {:ok, Map.put(data, :signed_url, get_signed_url(unit))}
    end
  end

  def before_create(_, %{data: %{name: name, path: path, binary: binary}} = unit) do
    case Gcp.upload_file("#{path}/#{name}", binary) do
      {:ok, _object} -> {:ok, unit}
      {:error, error} -> {:error, error}
    end
  end

  def before_delete(_, %{data: %{name: name, path: path}} = unit) do
    case Gcp.delete_file("#{path}/#{name}") do
      {:ok, _e} -> {:ok, unit}
      {:error, error} -> {:error, error}
    end
  end

  def before_update(_, %{binary: binary} = unit) when is_nil(binary), do: {:ok, unit}

  def before_update(_, %{data: %{name: name, path: path, binary: binary}} = unit) do
    case Gcp.upload_file("#{path}/#{name}", binary) do
      {:ok, _object} -> {:ok, unit}
      {:error, error} -> {:error, error}
    end
  end

  def get_signed_url(%{data: data} = unit) do
    {:ok, signed_url} = Gcp.get_bucket_file_signed_url("#{data.path}/#{data.name}")
    signed_url
  end

  # def test() do
  #   {:ok, token} = Goth.Token.fetch([])
  #   conn = GoogleApi.Storage.V1.Connection.new(token.token)

  #   # Call the Storage V1 API (for example) to list buckets
  #   {:ok, response} = GoogleApi.Storage.V1.Api.Buckets.storage_buckets_list(conn, "arkemis-lab")

  #   # Print the response
  #   Enum.each(response.items, &IO.puts(&1.id))
  # end

  # @spec upload_file :: none
  # def upload_file() do
  #   # Authenticate.
  #   {:ok, token} = Goth.Token.fetch([])
  #   conn = GoogleApi.Storage.V1.Connection.new(token.token)

  #   # Make the API request.
  #   {:ok, object} =
  #     GoogleApi.Storage.V1.Api.Objects.storage_objects_insert_iodata(
  #       conn,
  #       "arke_demo",
  #       "multipart",
  #       %{name: "file.txt"},
  #       "Hello " <> "Dorian aa" <> "!"
  #     )

  #   # Print the object.
  #   IO.puts("Uploaded #{object.name} to #{object.selfLink}")
  # end

  # def get() do
  #   {:ok, token} = Goth.Token.fetch([])
  #   conn = GoogleApi.Storage.V1.Connection.new(token.token)

  #   # Make the API request.
  #   {:ok, object} =
  #     GoogleApi.Storage.V1.Api.Objects.storage_objects_get(
  #       conn,
  #       "arke_demo",
  #       "file.txt"
  #     )
  # end

  # def get_url() do
  #   service_account = "arke-storage@arkemis-lab.iam.gserviceaccount.com"
  #   bucket = "arke_demo"
  #   object = "file.txt"

  #   get_signed_url(service_account, bucket, object)
  # end

  # def compose_signed_url(gcp_service_account, bucket, object) do
  #   %Tesla.Client{pre: [{Tesla.Middleware.Headers, :call, [auth_headers]}]} = get_connection()
  #   headers = [{"Content-Type", "application/json"}] ++ auth_headers

  #   url =
  #     "https://iamcredentials.googleapis.com/v1/projects/-/serviceAccounts/#{gcp_service_account}:signBlob"

  #   expires = DateTime.utc_now() |> DateTime.to_unix() |> Kernel.+(1 * 3600)
  #   resource = "/#{bucket}/#{object}"
  #   signature = ["GET", "", "", expires, resource] |> Enum.join("\n") |> Base.encode64()
  #   body = %{"payload" => signature} |> Poison.encode!()
  #   IO.inspect(HTTPoison.post(url, body, headers))
  #   {:ok, %{status_code: 200, body: result}} = HTTPoison.post(url, body, headers)

  #   %{"signedBlob" => signed_blob} = Poison.decode!(result)

  #   qs =
  #     %{
  #       "GoogleAccessId" => gcp_service_account,
  #       "Expires" => expires,
  #       "Signature" => signed_blob
  #     }
  #     |> URI.encode_query()

  #   Enum.join(["https://storage.googleapis.com#{resource}", "?", qs])
  # end

  # defp get_connection() do
  #   {:ok, token} = Goth.Token.fetch([])
  #   conn = GoogleApi.Storage.V1.Connection.new(token.token)
  # end
end