{-# LANGUAGE QuasiQuotes #-}
{-# OPTIONS -fno-warn-name-shadowing #-}
module Atomo.Kernel.Particle (load) where

import Atomo.Environment
import Atomo.Haskell


load :: VM ()
load = do
    [$p|(p: Particle) show|] =::: [$e|
        p type match: {
            @keyword -> {
                operator? = @(all?: @(in?: "~!@#$%^&*-_=+./\\\|<>?:"))

                keywordfy = { str |
                    if: (operator? call: [str])
                        then: { str }
                        else: { str .. ":" }
                }
                    
                vs := p values map: { v |
                    v match: {
                        @none -> "_"
                        @(ok: v) -> v show
                    }
                }

                initial := vs head match: {
                    "_" -> ""
                    v -> v .. " "
                }

                rest := (p names zip: vs tail) (map: { pair |
                    keywordfy call: [pair from] .. " " .. pair to
                }) (join: " ")

                if: p values (all?: @(== @none))
                    then: { "@" .. p names (map: { n | keywordfy call: [n] }) join }
                    else: {
                        "@(" .. initial .. rest .. ")"
                    }
            } call

            @single ->
                ("@" .. p name)
        }
    |]

    [$p|(p: Particle) call: (targets: List)|] =:::
        [$e|(p complete: targets) send|]

    [$p|(p: Particle) name|] =: do
        Particle (PMSingle n) <- here "p" >>= findParticle
        return (string n)

    [$p|(p: Particle) names|] =: do
        Particle (PMKeyword ns _) <- here "p" >>= findParticle
        list (map string ns)

    [$p|(p: Particle) values|] =: do
        (Particle (PMKeyword _ mvs)) <- here "p" >>= findParticle
        list $
            map
                (maybe (particle "none") (keyParticleN ["ok"] . (:[])))
                mvs

    [$p|(p: Particle) type|] =: do
        Particle p <- here "p" >>= findParticle
        case p of
            PMKeyword {} -> return (particle "keyword")
            PMSingle {} -> return (particle "single")

    [$p|(p: Particle) complete: (targets: List)|] =: do
        Particle p <- here "p" >>= findParticle
        vs <- getList [$e|targets|]

        case p of
            PMKeyword ns mvs -> do
                let blanks = length (filter (== Nothing) mvs)

                if blanks > length vs
                    then throwError (ParticleArity blanks (length vs))
                    else return . Message . keyword ns $ completeKP mvs vs
            PMSingle n -> do
                if length vs == 0
                    then throwError (ParticleArity 1 0)
                    else return . Message . single n $ head vs