@awayfl/avm2/lib/abc/lazy/createMethodForTrait.ts

Virtual machine for executing AS3 code

import { MethodTraitInfo } from './MethodTraitInfo';
import { Scope } from '../../run/Scope';
import { ScriptInfo } from './ScriptInfo';
import { METHOD } from './METHOD';
import { createGlobalNative } from './createGlobalNative';
import { flashlog, release } from '@awayfl/swf-loader';
import { interpret } from '../../int';
import { getNative } from '../../nat/getNative';
import { getMethodOrAccessorNative } from '../../nat/getMethodOrAccessorNative';
import { assert } from '@awayjs/graphics';

export function createMethodForTrait(
	methodTraitInfo: MethodTraitInfo,
	scope: Scope,
	forceNativeMethods: boolean = false
) {
	if (methodTraitInfo.method) {
		return methodTraitInfo.method;
	}
	const methodInfo = methodTraitInfo.methodInfo;
	let method;
	if (methodInfo.flags & METHOD.Native) {
		const metadata = methodInfo.getNativeMetadata();
		if (metadata || methodTraitInfo.holder instanceof ScriptInfo) {
			if (metadata) {
				method = getNative(metadata.values[0]);
			} else {
				const mn = methodTraitInfo.multiname;
				method = getNative(mn.uri + '.' + mn.name);
			}
			method = createGlobalNative(method, scope.object.sec);
		} else {
			method = getMethodOrAccessorNative(methodTraitInfo);
		}
		if (method && !release) {
			method.toString = function () {
				return 'Native ' + methodTraitInfo.toString();
			};
			method.isInterpreted = false;
		}
	} else {
		if (forceNativeMethods)
			method = getMethodOrAccessorNative(methodTraitInfo, false);
		if (!method) {
			method = interpret(methodInfo, scope, null);

			if (!release) {
				method.toString = function () {
					return 'Interpreted ' + methodTraitInfo.toString();
				};
				method.isInterpreted = true;
			}
		}
	}
	if (!release && flashlog && methodInfo.trait) {
		method = (function (wrapped, methodInfo) {
			const traceMsg = methodInfo.toFlashlogString();
			const result: any = function () {
				flashlog.writeAS3Trace(traceMsg);
				return wrapped.apply(this, arguments);
			};
			result.toString = wrapped.toString;
			result.isInterpreted = wrapped.isInterpreted;
			return result;
		})(method, methodInfo);
	}

	assert(method, 'Not found method:' + methodTraitInfo.toString());

	methodTraitInfo.method = method;
	method.methodInfo = methodInfo;
	if (!release) {
		try {
			Object.defineProperty(method, 'name', { value: methodInfo.name });
		} catch (e) {
			// Ignore errors in browsers that don't allow overriding Function#length;
		}
	}
	method.methodInfo = methodInfo;
	return method;
}