%{

#include <iostream>
#include <string>
#include <cstdlib>
#include <string>
#include <vector>
#include "ast.hpp"
#define YYDEBUG 1  
  using namespace std;

  int yylex();

  void yyerror(string s) {
    cerr << s << endl;
    exit(EXIT_FAILURE);
  }

%}

%union {
  double d;
  string *s;
  ExprAST *e;
  vector<ExprAST*> *v;
  vector<string> *v1;
}

%token def_token
%token <s> id_token
%token <d> num_token

%type <e> E
%type <v> EArray Arguments
%type <v1> Args

%left '+' '-'
%left '*' '/'


%%
Program: Program Statement ';'
| Statement ';'
;
Statement: def_token id_token '(' Args ')' E {
  //TODO:
  
 }
| E {
  //TODO: Uradi nesto
  
  delete $1;
 }
;
Args: Args id_token {
  $$ = $1;
  $$->push_back(*$2);
  delete $2;
 }
| {
  $$ = new vector<string>();
 }
;
E: E '+' E {
  $$ = new AddExprAST($1, $3);
 }
| E '-' E {
  $$ = new SubExprAST($1, $3);
 }
| E '*' E {
  $$ = new MulExprAST($1, $3);
 }
| E '/' E {
  $$ = new DivExprAST($1, $3);
 }
| '(' E ')' {
  $$ = $2;
 }
| num_token {
  $$ = new NumberExprAST($1);
 }
| id_token {
  $$ = new VariableExprAST(*$1);
  delete $1;
 }
| id_token '(' Arguments ')' {
  $$ = new CallExprAST(*$1, *$3);
  delete $1;
  delete $3;
 }
;
Arguments: EArray {
  $$ = $1;
 }
| {
  $$ = new vector<ExprAST*>();
 }
;
EArray: EArray ',' E {
  $$ = $1;
  $$->push_back($3);
 }
| E {
  $$ = new vector<ExprAST*>();
  $$->push_back($1);
 }
;
%%
int main() {
  //yydebug = 1;
  yyparse();
  return 0;
}