BadModuleImporter.cs

Go to the documentation of this file.

using BadScript2.Runtime.Error;
using BadScript2.Runtime.Objects;
using BadScript2.Runtime.Settings;
 
namespace BadScript2.Runtime.Module;
 
public class BadModuleImporter
{
    private readonly List<BadImportHandler> m_Handlers = new List<BadImportHandler>();
 
    private readonly BadModuleStore m_Store;
 
    public BadModuleImporter(BadModuleStore store)
    {
        m_Store = store;
    }
 
    public BadModuleImporter Clear()
    {
        m_Handlers.Clear();
        return this;
    }
 
    public BadModuleImporter Clone(bool onlyTransient = true)
    {
        BadModuleImporter importer = new BadModuleImporter(m_Store);
 
        if (onlyTransient)
        {
            importer.m_Handlers.AddRange(m_Handlers.Where(x => x.IsTransient()));
        }
        else
        {
            importer.m_Handlers.AddRange(m_Handlers);
        }
 
        return importer;
    }
 
    public BadModuleImporter AddHandler(BadImportHandler handler)
    {
        m_Handlers.Add(handler);
 
        return this;
    }
 
    public IEnumerable<BadObject> Get(string path)
    {
        for (int i = m_Handlers.Count - 1; i >= 0; i--)
        {
            BadImportHandler handler = m_Handlers[i];
            string hash = handler.GetHash(path);
 
            if (BadModuleSettings.Instance.UseModuleCaching && m_Store.IsCached(hash))
            {
                yield return m_Store.Get(hash);
 
                yield break;
            }
 
            if (handler.Has(path))
            {
                IEnumerable<BadObject> result = handler.Get(path);
                BadObject r = BadObject.Null;
 
                foreach (BadObject o in result)
                {
                    r = o;
 
                    yield return r;
                }
 
                r = r.Dereference(null);
 
                if (!BadModuleSettings.Instance.AllowImportHandlerNullReturn && r == BadObject.Null)
                {
                    continue;
                }
 
                if (BadModuleSettings.Instance.UseModuleCaching && r != BadObject.Null)
                {
                    m_Store.Cache(hash, r);
                }
 
                yield return r;
 
                yield break;
            }
        }
 
        if (BadModuleSettings.Instance.ThrowOnModuleUnresolved)
        {
            throw new BadRuntimeException("Module " + path + " could not be resolved.");
        }
 
        yield return BadObject.Null;
    }
}