defmodule Mix.Tasks.Compile.StemexSnowball do
  use Mix.Task.Compiler

  @shortdoc "Compiles stemmers c source, meant to be used only for algorithms development"
  def run(_) do
    File.touch!("lib/stemex.ex") # force recompilation of stemex
    File.write!("c_src/gen/wrapper.c",EEx.eval_file("c_src/wrapper.c.eex", stemmers: Mix.Project.get.stemmers))
    if System.find_executable("snowball") do
      for stemmer<-Mix.Project.get.stemmers do
        opts = ["algorithms/#{stemmer}.sbl","-o","c_src/gen/#{stemmer}","-utf8","-eprefix","#{stemmer}_","-vprefix","#{stemmer}_var","-r","../runtime"]
        System.cmd "snowball", opts, into: IO.stream(:stdio, :line)
      end
      |> Enum.any?(fn {_, exit_status} -> exit_status != 0 end)
      |> case do
        false -> :ok
        true -> :error
      end
    else
      Mix.Shell.IO.error("""
        Missing snowball compiler on your path
        install it for instance doing : 
          wget http://snowball.tartarus.org/dist/snowball_code.tgz
          tar xvzf snowball_code.tgz
          cd snowball_code
          make
          cp snowball ~/bin/snowball
      """)
      :error
    end
  end
end

defmodule Mix.Tasks.Compile.StemexNif do
  use Mix.Task.Compiler

  # TODO : Implement clean/0

  @shortdoc "Compiles stemmer nif .so library"
  def run(_) do
    File.mkdir "priv"
    System.cmd "cc", cc_args()++["-o","priv/Elixir.Stemex_nif.so"]++Path.wildcard("c_src/*/*.c"), into: IO.stream(:stdio, :line)
    Mix.Project.build_structure
    :ok
  end

  defp cc_args do
    [i_erts]=Path.wildcard("#{:code.root_dir}/erts*/include")
    i_ei=:code.lib_dir(:erl_interface,:include)
    l_ei=:code.lib_dir(:erl_interface,:lib)
    args = ["-L#{l_ei}","-lei","-I#{i_ei}","-I#{i_erts}","-Wall","-shared","-fPIC"] #,"-v"]
    args ++ if {:unix,:darwin}==:os.type, do: ["-undefined","dynamic_lookup","-dynamiclib"], else: []
  end
end

defmodule Stemex.Mixfile do
  use Mix.Project

  def stemmers do
    Enum.map(File.ls!("algorithms"),&Path.rootname/1)
  end

  def project do
    [app: :stemex,
     version: "0.2.1",
     elixir: "~> 1.0",
     compilers: compilers(Mix.env),
     package: [
       files: ["lib", "c_src", "algorithms", "mix.exs", "README.md", "LICENSE"],
       maintainers: ["Arnaud Wetzel"],
       licenses: ["MIT"],
       links: %{
         "GitHub" => "https://github.com/kbrw/stemex"
       }
     ],
     description: """
       Stemex is a NIF wrapper above snowball containing stemmers for : 
       danish, dutch, english, finnish, french, german, hungarian, italian, kraaij_pohlmann, 
       lovins, norwegian, portuguese, romanian, russian, spanish, swedish, turkish.
     """,
     deps: []]
  end

  def application do
    [
      extra_applications: [
        :erl_interface,
      ],
    ]
  end

  defp compilers(:dev), do: [:stemex_snowball] ++ compilers(:all) 
  defp compilers(_), do: [:stemex_nif] ++ Mix.compilers
end