dist_agent_lang 1.0.18

use crate::lexer::tokens::Literal;
use crate::lexer::tokens::Operator;
use std::collections::HashMap;

/// Source location (line, column) for error reporting and runtime diagnostics.
#[derive(Debug, Clone, Copy, Default)]
pub struct Span {
    pub line: usize,
    pub column: usize,
}

#[derive(Debug, Clone)]
pub struct Program {
    pub statements: Vec<Statement>,
    /// Parallel to `statements`: span for each top-level statement (from parser).
    pub statement_spans: Vec<Option<Span>>,
}

#[derive(Debug, Clone)]
pub enum Statement {
    Expression(Expression),
    Let(LetStatement),
    Return(ReturnStatement),
    Block(BlockStatement),
    Function(FunctionStatement),
    Service(ServiceStatement),
    Spawn(SpawnStatement),
    Agent(AgentStatement),
    Message(MessageStatement),
    Event(EventStatement),
    If(IfStatement),
    While(WhileStatement),
    Try(TryStatement),
    ForIn(ForInStatement),
    Break(BreakStatement),
    Continue(ContinueStatement),
    Loop(LoopStatement),
    Match(MatchStatement),
    /// Top-level only: `import <path>;` or `import <path> as <alias>;`
    Import(ImportStatement),
}

/// `import <path>;` or `import <path> as <alias>;` — path is stdlib::name, relative path, or package name.
#[derive(Debug, Clone)]
pub struct ImportStatement {
    /// Import path: e.g. "stdlib::chain", "./mymod.dal", "mypkg"
    pub path: String,
    /// Optional alias: name to bind in current scope (e.g. `as mymod`)
    pub alias: Option<String>,
}

/// `while ( condition ) { body }`
#[derive(Debug, Clone)]
pub struct WhileStatement {
    pub condition: Expression,
    pub body: BlockStatement,
}

/// `for <variable> in <iterable> { body }`
#[derive(Debug, Clone)]
pub struct ForInStatement {
    pub variable: String,
    pub iterable: Expression,
    pub body: BlockStatement,
}

/// `break;` - exit the innermost loop
#[derive(Debug, Clone)]
pub struct BreakStatement {
    pub value: Option<Expression>, // Optional value to return from loop
}

/// `continue;` - skip to next iteration of innermost loop
#[derive(Debug, Clone)]
pub struct ContinueStatement;

/// `loop { body }` - infinite loop (can be exited with break)
#[derive(Debug, Clone)]
pub struct LoopStatement {
    pub body: BlockStatement,
}

/// `match expr { case1 => body1, case2 => body2, default => body }`
#[derive(Debug, Clone)]
pub struct MatchStatement {
    pub expression: Expression,
    pub cases: Vec<MatchCase>,
    pub default_case: Option<BlockStatement>,
}

/// A single case in a match statement: `pattern => { body }`
#[derive(Debug, Clone)]
pub struct MatchCase {
    pub pattern: MatchPattern,
    pub body: BlockStatement,
}

/// Pattern for matching in match statements
#[derive(Debug, Clone)]
pub enum MatchPattern {
    Literal(Literal),   // Match literal value (42, "hello", true, null)
    Identifier(String), // Match and bind to variable
    Wildcard,           // _ matches anything
    Range(Box<Expression>, Box<Expression>), // start..end (inclusive range)
}

#[derive(Debug, Clone)]
pub struct LetStatement {
    pub name: String,
    pub value: Expression,
    /// 1-based line number for warnings (e.g. unused variable); None if unknown.
    pub line: Option<usize>,
}

#[derive(Debug, Clone)]
pub struct ReturnStatement {
    pub value: Option<Expression>,
}

#[derive(Debug, Clone)]
pub struct BlockStatement {
    pub statements: Vec<Statement>,
}

#[derive(Debug, Clone)]
pub struct FunctionStatement {
    pub name: String,
    pub parameters: Vec<Parameter>,
    pub return_type: Option<String>,
    pub body: BlockStatement,
    pub attributes: Vec<Attribute>,
    pub is_async: bool,
    /// M5: true when declared with `export fn ...` (visible when module is imported).
    pub exported: bool,
}

impl FunctionStatement {
    pub fn new(
        name: String,
        parameters: Vec<Parameter>,
        return_type: Option<String>,
        body: BlockStatement,
    ) -> Self {
        Self {
            name,
            parameters,
            return_type,
            body,
            attributes: Vec::new(),
            is_async: false,
            exported: false,
        }
    }
}

#[derive(Debug, Clone)]
pub struct Parameter {
    pub name: String,
    pub param_type: Option<String>,
}

#[derive(Debug, Clone)]
pub struct FunctionCall {
    pub name: String,
    pub arguments: Vec<Expression>,
}

#[derive(Debug, Clone)]
pub struct SpawnStatement {
    pub agent_name: String,
    pub agent_type: Option<String>, // "ai", "system", "worker", etc.
    pub config: Option<HashMap<String, Expression>>, // Agent configuration
    pub body: BlockStatement,
}

#[derive(Debug, Clone)]
pub struct AgentStatement {
    pub name: String,
    pub agent_type: AgentType,
    pub config: HashMap<String, Expression>,
    pub capabilities: Vec<String>,
    pub body: BlockStatement,
}

#[derive(Debug, Clone)]
pub enum AgentType {
    AI,
    System,
    Worker,
    IDE,
    Custom(String),
}

#[derive(Debug, Clone)]
pub struct MessageStatement {
    pub recipient: String,
    pub data: HashMap<String, Expression>,
}

#[derive(Debug, Clone)]
pub struct EventStatement {
    pub event_name: String,
    pub data: HashMap<String, Expression>,
}

#[derive(Debug, Clone)]
pub struct IfStatement {
    pub condition: Expression,
    pub consequence: BlockStatement,
    pub alternative: Option<BlockStatement>,
}

#[derive(Debug, Clone)]
pub struct TryStatement {
    pub try_block: BlockStatement,
    pub catch_blocks: Vec<CatchBlock>,
    pub finally_block: Option<BlockStatement>,
}

#[derive(Debug, Clone)]
pub struct CatchBlock {
    pub error_type: Option<String>,
    pub error_variable: Option<String>,
    pub body: BlockStatement,
}

// NEW: Service-related structures

#[derive(Debug, Clone)]
pub struct ServiceField {
    pub name: String,
    pub field_type: String,
    pub initial_value: Option<Expression>,
    pub visibility: FieldVisibility,
}

#[derive(Debug, Clone)]
pub enum FieldVisibility {
    Public,
    Private,
    Internal,
}

#[derive(Debug, Clone)]
pub struct EventDeclaration {
    pub name: String,
    pub parameters: Vec<Parameter>,
}

// NEW: Compilation Target System
#[derive(Debug, Clone)]
pub struct CompilationTargetInfo {
    pub target: crate::lexer::tokens::CompilationTarget,
    pub constraints: crate::lexer::tokens::TargetConstraint,
    pub validation_errors: Vec<String>,
}

// Extend ServiceStatement with compilation target and M5 export
#[derive(Debug, Clone)]
pub struct ServiceStatement {
    pub name: String,
    pub attributes: Vec<Attribute>,
    pub fields: Vec<ServiceField>,
    pub methods: Vec<FunctionStatement>,
    pub events: Vec<EventDeclaration>,
    pub compilation_target: Option<CompilationTargetInfo>,
    /// M5: true when declared with `export service ...` (visible when module is imported).
    pub exported: bool,
}

#[derive(Debug, Clone)]
pub struct Attribute {
    pub name: String,
    pub parameters: Vec<Expression>,
    pub target: AttributeTarget,
}

#[derive(Debug, Clone)]
pub enum AttributeTarget {
    Function,
    Block,
    Variable,
    Module,
}

#[derive(Debug, Clone)]
pub enum Expression {
    Literal(Literal),
    Identifier(String),
    BinaryOp(Box<Expression>, Operator, Box<Expression>),
    UnaryOp(Operator, Box<Expression>),
    Assignment(String, Box<Expression>),
    FunctionCall(FunctionCall),
    FieldAccess(Box<Expression>, String), // NEW: self.field syntax
    FieldAssignment(Box<Expression>, String, Box<Expression>), // NEW: self.field = value
    Await(Box<Expression>),
    Spawn(Box<Expression>), // spawn <expr> e.g. spawn worker_process(i)
    Throw(Box<Expression>),
    ObjectLiteral(HashMap<String, Expression>), // NEW: object literal syntax
    ArrayLiteral(Vec<Expression>),              // NEW: array literal syntax [expr1, expr2, ...]
    /// Index access: `expr[index]` — array or map key access
    IndexAccess(Box<Expression>, Box<Expression>),
    /// Arrow function: (param => { body }) — single param, block body
    ArrowFunction {
        param: String,
        body: BlockStatement,
    },
    /// Range expression: start..end for for-loops
    Range(Box<Expression>, Box<Expression>),
    /// Method call: receiver.method_name(args) — dispatch by receiver type (service or value)
    MethodCall {
        receiver: Box<Expression>,
        method_name: String,
        arguments: Vec<Expression>,
    },
}

impl Program {
    pub fn new() -> Self {
        Self {
            statements: Vec::new(),
            statement_spans: Vec::new(),
        }
    }

    pub fn add_statement(&mut self, statement: Statement) {
        self.add_statement_with_span(statement, None);
    }

    pub fn add_statement_with_span(&mut self, statement: Statement, span: Option<Span>) {
        self.statements.push(statement);
        self.statement_spans.push(span);
    }
}

impl Default for Program {
    fn default() -> Self {
        Self::new()
    }
}

impl BlockStatement {
    pub fn new() -> Self {
        Self {
            statements: Vec::new(),
        }
    }

    pub fn add_statement(&mut self, statement: Statement) {
        self.statements.push(statement);
    }
}

impl Default for BlockStatement {
    fn default() -> Self {
        Self::new()
    }
}