/*----------------------------------------------------------

This Source Code Form is subject to the terms of the

Mozilla Public License, v.2.0. If a copy of the MPL

was not distributed with this file, You can obtain one

at http://mozilla.org/MPL/2.0/.

----------------------------------------------------------*/

using System;

using System.Linq;

using ScriptEngine.Environment;

using ScriptEngine.Machine;

using ScriptEngine.Machine.Contexts;

using ScriptEngine.Compiler;

namespace ScriptEngine

{

public class ScriptingEngine : IDisposable

{

private readonly MachineInstance _machine;

private readonly ScriptSourceFactory _scriptFactory;

private AttachedScriptsFactory _attachedScriptsFactory;

private IDebugController _debugController;

public ScriptingEngine()

{

_machine = MachineInstance.Current;

TypeManager.Initialize(new StandartTypeManager());

GlobalsManager.Reset();

ContextDiscoverer.DiscoverClasses(System.Reflection.Assembly.GetExecutingAssembly());

_scriptFactory = new ScriptSourceFactory();

}

public CodeGenerationFlags ProduceExtraCode { get; set; }

public void AttachAssembly(System.Reflection.Assembly asm)

{

ContextDiscoverer.DiscoverClasses(asm);

}

public void AttachAssembly(System.Reflection.Assembly asm, RuntimeEnvironment globalEnvironment)

{

ContextDiscoverer.DiscoverClasses(asm);

ContextDiscoverer.DiscoverGlobalContexts(globalEnvironment, asm);

}

public RuntimeEnvironment Environment { get; set; }

public void Initialize()

{

SetDefaultEnvironmentIfNeeded();

UpdateContexts();

_attachedScriptsFactory = new AttachedScriptsFactory(this);

AttachedScriptsFactory.SetInstance(_attachedScriptsFactory);

}

public void UpdateContexts()

{

Environment.LoadMemory(_machine);

}

private void SetDefaultEnvironmentIfNeeded()

{

if (Environment == null)

Environment = new RuntimeEnvironment();

}

public ICodeSourceFactory Loader

{

get

{

return _scriptFactory;

}

}

public IDirectiveResolver DirectiveResolver { get; set; }

public CompilerService GetCompilerService()

{

var cs = new CompilerService(Environment.SymbolsContext);

cs.ProduceExtraCode = ProduceExtraCode;

cs.DirectiveResolver = DirectiveResolver;

return cs;

}

public LoadedModuleHandle LoadModuleImage(ScriptModuleHandle moduleImage)

{

var handle = new LoadedModuleHandle();

handle.Module = new LoadedModule(moduleImage.Module);

return handle;

}

internal IRuntimeContextInstance NewObject(LoadedModule module, ExternalContextData externalContext = null)

{

var scriptContext = new Machine.Contexts.UserScriptContextInstance(module);

scriptContext.AddProperty("ЭтотОбъект", "ThisObject", scriptContext);

if (externalContext != null)

{

foreach (var item in externalContext)

{

scriptContext.AddProperty(item.Key, item.Value);

}

}

scriptContext.InitOwnData();

InitializeSDO(scriptContext);

return scriptContext;

}

public IRuntimeContextInstance NewObject(LoadedModuleHandle module)

{

return NewObject(module.Module);

}

public IRuntimeContextInstance NewObject(LoadedModuleHandle module, ExternalContextData externalContext)

{

return NewObject(module.Module, externalContext);

}

public void InitializeSDO(ScriptDrivenObject sdo)

{

sdo.Initialize();

}

public void ExecuteModule(LoadedModuleHandle module)

{

var scriptContext = new Machine.Contexts.UserScriptContextInstance(module.Module);

InitializeSDO(scriptContext);

}

public MachineInstance Machine

{

get { return _machine; }

}

public AttachedScriptsFactory AttachedScriptsFactory

{

get

{

return _attachedScriptsFactory;

}

}

public IDebugController DebugController

{

get

{

return _debugController;

}

set

{

_debugController = value;

ProduceExtraCode = CodeGenerationFlags.DebugCode;

_machine.SetDebugMode(_debugController);

}

}

public void SetCodeStatisticsCollector(ICodeStatCollector collector)

{

ProduceExtraCode = CodeGenerationFlags.CodeStatistics;

_machine.SetCodeStatisticsCollector(collector);

}

#region IDisposable Members

public void Dispose()

{

AttachedScriptsFactory.Dispose();

}

#endregion

public void CompileEnvironmentModules(RuntimeEnvironment env)

{

var scripts = env.GetUserAddedScripts().Where(x => x.Type == UserAddedScriptType.Module && env.GetGlobalProperty(x.Symbol) == null);

foreach (var script in scripts)

{

var loaded = LoadModuleImage(script.Module);

var instance = (IValue)NewObject(loaded);

env.SetGlobalProperty(script.Symbol, instance);

}

}

}

}