using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

using PascalABCCompiler;
using PascalABCCompiler.SyntaxTree;

namespace SyntaxVisitors
{
    public class CollectClassMethodsVisitor : CollectUpperNodesVisitor
    {
        private ident _className;

        public ISet<ident> CollectedMethods { get; private set; }

        public CollectClassMethodsVisitor(ident className)
        {
            CollectedMethods = new HashSet<ident>();
            _className = className;
        }

        public override void visit(class_definition cd)
        {
            type_declaration td = UpperNode(1) as type_declaration;

            if ((object)td == null)
            {
                // Something went wrong
                return;
            }

            if (td.type_name.name != _className.name)
            {
                // Not the class we search for
                // Yoda
                return;
            }

            var methods = cd.body.class_def_blocks.SelectMany(cm => cm.members.Select(decl1 => 
                {
                    if (decl1 is procedure_header)
                        return (decl1 as procedure_header).name.meth_name;
                    else if (decl1 is procedure_definition)
                        return (decl1 as procedure_definition).proc_header.name.meth_name;
                    return null;
                }).Where(name => (object)name != null));

            CollectedMethods.UnionWith(methods);
        }

    }
}