aboutsummaryrefslogtreecommitdiffstats
path: root/packages/utils/src/abi_encoder/utils/signatureParser.ts
blob: c4796e24ee01a3727f43534f1463c3a6f06fc53e (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import * as _ from 'lodash';

import { DataItem } from 'ethereum-protocol';

/*
export function generateDataItemFromSignature(signature: string): DataItem {
    const dataItems = generateDataItemsFromSignature(signature);
    if (dataItems.length === 1) {
        return dataItems[0];
    }
    // signature represents a tuple
    return {
        name: '',
        type: 'tuple',
        components: dataItems
    };
}*/

export function generateDataItemsFromSignature(signature: string): DataItem[] {
    let trimmedSignature = signature;
    if (signature.startsWith('(')) {
        if (!signature.endsWith(')')) {
            throw new Error(`Failed to generate data item. Must end with ')'`);
        }
        trimmedSignature = signature.substr(1, signature.length - 2);
    }
    trimmedSignature += ',';
    let currTokenIsArray = false;
    let currTokenArrayModifier = '';
    let isParsingArrayModifier = false;
    let currToken = '';
    let parenCount = 0;
    let currTokenName = '';
    const dataItems: DataItem[] = [];
    for (const char of trimmedSignature) {
        // Tokenize the type string while keeping track of parentheses.
        switch (char) {
            case '(':
                parenCount += 1;
                currToken += char;
                break;
            case ')':
                parenCount -= 1;
                currToken += char;
                break;
            case '[':
                if (parenCount === 0) {
                    isParsingArrayModifier = true;
                    currTokenIsArray = true;
                    currTokenArrayModifier += '[';
                } else {
                    currToken += char;
                }
                break;
            case ']':
                if (parenCount === 0) {
                    isParsingArrayModifier = false;
                    currTokenArrayModifier += ']';
                } else {
                    currToken += char;
                }
                break;
            case ' ':
                if (parenCount === 0) {
                    currTokenName = currToken;
                    currToken = '';
                } else {
                    currToken += char;
                }
                break;
            case ',':
                if (parenCount === 0) {
                    //throw new Error(`Generating Data Items`);
                    const components = currToken.startsWith('(') ? generateDataItemsFromSignature(currToken) : [];
                    const isTuple = !_.isEmpty(components);
                    const isArray = currTokenIsArray;
                    let dataItem: DataItem = { name: currTokenName, type: '' };
                    if (isTuple) {
                        dataItem.type = 'tuple';
                        dataItem.components = components;
                    } else {
                        dataItem.type = currToken;
                    }
                    if (isArray) {
                        dataItem.type += currTokenArrayModifier;
                    }

                    dataItems.push(dataItem);

                    currTokenName = '';
                    currToken = '';
                    currTokenIsArray = false;
                    currTokenArrayModifier = '';
                    break;
                } else {
                    currToken += char;
                    break;
                }
            default:
                if (isParsingArrayModifier) {
                    currTokenArrayModifier += char;
                } else {
                    currToken += char;
                }
                break;
        }
    }
    return dataItems;
}