Igniter installer

lib/mix/tasks/ExAtomVM.install.ex

@@ -0,0 +1,222 @@
+defmodule Mix.Tasks.Exatomvm.Install do
+  use Igniter.Mix.Task
+
+  @example "mix igniter.new my_project --install exatomvm@github:atomvm/exatomvm && cd my_project"
+
+  @shortdoc "Add and configure AtomVM for your project"
+  @moduledoc """
+  #{@shortdoc}
+
+  This task sets up your Elixir project to work with AtomVM, adding necessary dependencies
+  and configuration for targeting embedded devices like ESP32, Raspberry Pi Pico, and STM32.
+
+  ## Example
+
+  ```bash
+  #{@example}
+  ```
+
+  """
+
+  @impl Igniter.Mix.Task
+  def info(_argv, _composing_task) do
+    %Igniter.Mix.Task.Info{
+      # Groups allow for overlapping arguments for tasks by the same author
+      # See the generators guide for more.
+      group: :exatomvm,
+      # dependencies to add
+      adds_deps: [
+        {:pythonx, "~> 0.4.0", runtime: false},
+        {:req, "~> 0.5.0", runtime: false}
+      ],
+      # dependencies to add and call their associated installers, if they exist
+      installs: [],
+      # An example invocation
+      example: @example,
+      # A list of environments that this should be installed in.
+      only: nil,
+      # a list of positional arguments, i.e `[:file]`
+      positional: [],
+      # Other tasks your task composes using `Igniter.compose_task`, passing in the CLI argv
+      # This ensures your option schema includes options from nested tasks
+      composes: [],
+      # `OptionParser` schema
+      schema: [],
+      # Default values for the options in the `schema`
+      defaults: [],
+      # CLI aliases
+      aliases: [],
+      # A list of options in the schema that are required
+      required: []
+    }
+  end
+
+  @impl Igniter.Mix.Task
+  def igniter(igniter) do
+    selected_instructions =
+      Igniter.Util.IO.select(
+        """
+        About to configure project for AtomVM!
+
+        Which device would you like to see setup instructions for?
+        (Your project will be configured for all devices - this only affects the instructions shown):
+        """,
+        ["ESP32", "Pico", "STM32", "All"],
+        default: "All"
+      )
+
+    IO.puts("\nGreat! now we will make the necessary changes to your project...\n")
+    Process.sleep(2000)
+
+    options = [
+      start: Igniter.Project.Module.module_name_prefix(igniter),
+      esp32_flash_offset: Sourceror.parse_string!("0x250000"),
+      stm32_flash_offset: Sourceror.parse_string!("0x8080000"),
+      chip: "auto",
+      port: "auto"
+    ]
+
+    Igniter.update_elixir_file(igniter, "mix.exs", fn zipper ->
+      with {:ok, zipper} <- Igniter.Code.Function.move_to_def(zipper, :project, 0),
+           {:ok, zipper} <-
+             Igniter.Code.Keyword.put_in_keyword(
+               zipper,
+               [:atomvm],
+               options
+             ) do
+        {:ok, zipper}
+      end
+    end)
+    |> Igniter.mkdir("avm_deps")
+    |> Igniter.Project.Module.find_and_update_module!(
+      Igniter.Project.Module.module_name_prefix(igniter),
+      fn zipper ->
+        case Igniter.Code.Function.move_to_def(zipper, :start, 0) do
+          :error ->
+            # start function not available, so let's create one
+            zipper =
+              Igniter.Code.Common.add_code(
+                zipper,
+                """
+                def start do
+                  IO.inspect("Hello AtomVM!")
+                  :ok
+                end
+                """,
+                placement: :before
+              )
+
+            {:ok, zipper}
+
+          _ ->
+            {:ok, zipper}
+        end
+      end
+    )
+    |> Igniter.Project.Deps.set_dep_option(:exatomvm, :runtime, false)
+    |> Igniter.Project.Deps.set_dep_option(:igniter, :runtime, false)
+    |> output_instructions(selected_instructions)
+  end
+
+  defp common_intro do
+    """
+    🎉 Your AtomVM project is now ready!
+
+    Next, you need to install AtomVM itself on your device.
+
+    """
+  end
+
+  defp output_instructions(igniter, selected_instructions)
+       when selected_instructions == "ESP32" do
+    igniter
+    |> Igniter.add_notice("ESP32 Setup Instructions")
+    |> Igniter.add_notice("""
+    #{common_intro()}
+
+    ## Installing AtomVM on ESP32
+
+    Choose one of these methods:
+
+    1. **Using Mix task (recommended):**
+       mix atomvm.esp32.install
+
+    2. **Manual installation:**
+       Follow the guide at: https://doc.atomvm.org/main/getting-started-guide.html#flashing-a-binary-image-to-esp32
+
+    3. **Web flasher (Chrome browser only):**
+       (Choose the Elixir-enabled build of AtomVM.)
+       Visit: https://petermm.github.io/atomvm_flasher
+
+    """)
+    |> Igniter.add_notice("""
+    ## Flashing Your Project
+
+    Once AtomVM is installed on your device, flash your project with:
+
+    mix atomvm.esp32.flash
+
+    """)
+  end
+
+  defp output_instructions(igniter, selected_instructions)
+       when selected_instructions == "Pico" do
+    igniter
+    |> Igniter.add_notice("Raspberry Pi Pico Setup Instructions")
+    |> Igniter.add_notice("""
+    #{common_intro()}
+
+    ## Installing AtomVM on Raspberry Pi Pico
+
+    Follow the installation guide at:
+    https://doc.atomvm.org/main/getting-started-guide.html#flashing-a-binary-image-to-pico
+
+    """)
+    |> Igniter.add_notice("""
+    ## Flashing Your Project
+
+    Once AtomVM is installed on your device, flash your project with:
+
+    mix atomvm.pico.flash
+
+    For more details, see: https://github.com/atomvm/exatomvm?tab=readme-ov-file#the-atomvmpicoflash-task
+    """)
+  end
+
+  defp output_instructions(igniter, selected_instructions)
+       when selected_instructions == "STM32" do
+    igniter
+    |> Igniter.add_notice("STM32 Setup Instructions")
+    |> Igniter.add_notice("""
+    #{common_intro()}
+
+    ## Building AtomVM for STM32
+
+    STM32 requires building AtomVM for your specific board:
+    https://doc.atomvm.org/main/build-instructions.html#building-for-stm32
+
+    ## Installing st-link
+
+    You'll need st-link installed for flashing:
+    - Installation guide: https://github.com/stlink-org/stlink?tab=readme-ov-file#installation
+    - Flashing guide: https://doc.atomvm.org/main/getting-started-guide.html#flashing-a-binary-image-to-stm32
+    """)
+    |> Igniter.add_notice("""
+    ## Flashing Your Project
+
+    Once AtomVM is built and installed on your device, flash your project with:
+
+    mix atomvm.stm32.flash
+
+    For more details, see: https://github.com/atomvm/exatomvm?tab=readme-ov-file#the-atomvmstm32flash-task
+    """)
+  end
+
+  defp output_instructions(igniter, selected_instructions)
+       when selected_instructions == "All" do
+    igniter
+    |> output_instructions("ESP32")
+    |> output_instructions("Pico")
+    |> output_instructions("STM32")
+  end
+end

mix.exs

@@ -35,7 +35,8 @@ defmodule ExAtomVM.MixProject do
       {:uf2tool, "1.1.0", runtime: false},
       {:ex_doc, "~> 0.20", only: :dev, runtime: false},
       {:pythonx, "~> 0.4.0", runtime: false, optional: true},
-      {:req, "~> 0.5.0", runtime: false, optional: true}
+      {:req, "~> 0.5.0", runtime: false, optional: true},
+      {:igniter, "~> 0.6", runtime: false, optional: true}
     ]
   end
 end