Function parseCreateEdge

masfactory-visualizer/src/parser/edgeParser.ts:83–140

(args: TSNode[], code: string, ctx: EdgeParseContext)

};

function parseCreateEdge(args: TSNode[], code: string, ctx: EdgeParseContext): ParsedEdgeInfo {
    let sender = '';
    let receiver = '';
    let keys: string[] = [];
    let keysDetails: { [key: string]: string } | undefined;

    // create_edge(sender, receiver, keys={...})
    let positionalIndex = 0;
    const senderNode = getExpandedKeywordValue(args, code, 'sender', ctx.literalValues);
    if (senderNode) {
        sender = resolveNodeReference(getNodeText(senderNode, code), ctx.variableToNodeName);
    }
    const receiverNode = getExpandedKeywordValue(args, code, 'receiver', ctx.literalValues);
    if (receiverNode) {
        receiver = resolveNodeReference(getNodeText(receiverNode, code), ctx.variableToNodeName);
    }
    const keysNode = getExpandedKeywordValue(args, code, 'keys', ctx.literalValues);
    if (keysNode) {
        const keysDict = parseDictArgument(keysNode, code);
        if (keysDict) {
            keys = Object.keys(keysDict);
            keysDetails = keysDict;
        }
    }

    for (const arg of args) {
        if (arg.type === 'keyword_argument') {
            const argName = getNodeText(arg.childForFieldName('name'), code);
            const argValue = arg.childForFieldName('value');
            if (argName === 'sender' && argValue && !sender) {
                sender = resolveNodeReference(getNodeText(argValue, code), ctx.variableToNodeName);
            } else if (argName === 'receiver' && argValue && !receiver) {
                receiver = resolveNodeReference(getNodeText(argValue, code), ctx.variableToNodeName);
            } else if (argName === 'keys' && argValue) {
                const keysDict = parseDictArgument(argValue, code);
                if (keysDict) {
                    keys = Object.keys(keysDict);
                    keysDetails = keysDict;
                }
            }
        } else {
            if (positionalIndex === 0 && !sender) {
                sender = resolveNodeReference(getNodeText(arg, code), ctx.variableToNodeName);
            } else if (positionalIndex === 1 && !receiver) {
                receiver = resolveNodeReference(getNodeText(arg, code), ctx.variableToNodeName);
            } else if (positionalIndex === 2 && arg.type === 'dictionary') {
                const keysDict = parseDictArgument(arg, code);
                if (keysDict) {
                    keys = Object.keys(keysDict);
                    keysDetails = keysDict;
                }
            }
            positionalIndex++;
        }
    }

    return { sender, receiver, keys, keysDetails };
}