using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using OCamlC.AST;
using OCamlC.Lex;
using OCamlC.Parse;

namespace OCamlC.Tests
{
    public class TokenListBuilder
    {
        private readonly List<Terminal> _tokens = new List<Terminal>();
        private bool _verbose = false;

        public TokenListBuilder Int(long value)
        {
            _tokens.Add(new IntLiteral(value, value.ToString()));
            return this;
        }

        /// <summary>
        /// Adds an infix operator of the correct precedence,
        /// or a keyword if this particular operator is a kw.
        /// </summary>
        public TokenListBuilder Infix(string op)
        {
            TokenType kwType;
            if (StaticData.Keywords.TryGetValue(op, out kwType))
            {
                _tokens.Add(new Keyword(kwType, op));
            }
            else
            {
                _tokens.Add(new InfixSymbol(StaticData.InfixSymbolClass(op), op));
            }
            return this;
        }

        /// <summary>
        /// Adds a prefix operator of the correct precedence,
        /// or a keyword if this particular operator is a kw.
        /// </summary>
        public TokenListBuilder Prefix(string op)
        {
            TokenType kwType;
            if (StaticData.Keywords.TryGetValue(op, out kwType))
            {
                _tokens.Add(new Keyword(kwType, op));
            }
            else
            {
                _tokens.Add(new PrefixSymbol(op));
            }
            return this;
        }

        internal TokenListBuilder Kw(string kw)
        {
            _tokens.Add(new Keyword(StaticData.Keywords[kw], kw));
            return this;
        }
        internal TokenListBuilder Kw(TokenType kw)
        {
            _tokens.Add(new Keyword(kw, StaticData.KeywordTypes[kw]));
            return this;
        }

        internal TokenListBuilder Id(string p)
        {
            if (StaticData.Keywords.ContainsKey(p))
                throw new ArgumentException(String.Format("{0} is a keyword, not an identifier"), p);
            _tokens.Add(new Identifier(p));
            return this;
        }

        /// <summary>
        /// End of statement keyword, ";;"
        /// </summary>
        /// <returns></returns>
        public TokenListBuilder EOS()
        {
            _tokens.Add(new Keyword(TokenType.KW_SEMISEMI, ";;"));
            return this;
        }

        /// <summary>
        /// Set the resulting scanner to give verbose output
        /// </summary>
        public TokenListBuilder Verbose()
        {
            _verbose = true;
            return this;
        }

        public ScanBase ToScanner()
        {
            ScannerInterfacer scanner = new ScannerInterfacer(_tokens);
            scanner.Verbose = _verbose;
            return scanner;
        }
    }
}