-
-
Notifications
You must be signed in to change notification settings - Fork 373
/
prototypes_adder.go
106 lines (90 loc) · 3.39 KB
/
prototypes_adder.go
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
// This file is part of arduino-cli.
//
// Copyright 2020 ARDUINO SA (http://www.arduino.cc/)
//
// This software is released under the GNU General Public License version 3,
// which covers the main part of arduino-cli.
// The terms of this license can be found at:
// https://www.gnu.org/licenses/gpl-3.0.en.html
//
// You can be released from the requirements of the above licenses by purchasing
// a commercial license. Buying such a license is mandatory if you want to
// modify or otherwise use the software for commercial activities involving the
// Arduino software without disclosing the source code of your own applications.
// To purchase a commercial license, send an email to license@arduino.cc.
package builder
import (
"fmt"
"github.com/arduino/arduino-cli/legacy/builder/constants"
"github.com/arduino/arduino-cli/legacy/builder/types"
"github.com/arduino/arduino-cli/legacy/builder/utils"
"strconv"
"strings"
)
type PrototypesAdder struct{}
func (s *PrototypesAdder) Run(ctx *types.Context) error {
debugOutput := ctx.DebugPreprocessor
source := ctx.Source
source = strings.Replace(source, "\r\n", "\n", -1)
source = strings.Replace(source, "\r", "\n", -1)
sourceRows := strings.Split(source, "\n")
firstFunctionLine := ctx.PrototypesLineWhereToInsert
if isFirstFunctionOutsideOfSource(firstFunctionLine, sourceRows) {
return nil
}
insertionLine := firstFunctionLine + ctx.LineOffset - 1
firstFunctionChar := len(strings.Join(sourceRows[:insertionLine], "\n")) + 1
prototypeSection := composePrototypeSection(firstFunctionLine, ctx.Prototypes)
ctx.PrototypesSection = prototypeSection
source = source[:firstFunctionChar] + prototypeSection + source[firstFunctionChar:]
if debugOutput {
fmt.Println("#PREPROCESSED SOURCE")
prototypesRows := strings.Split(prototypeSection, "\n")
prototypesRows = prototypesRows[:len(prototypesRows)-1]
for i := 0; i < len(sourceRows)+len(prototypesRows); i++ {
if i < insertionLine {
fmt.Printf(" |%s\n", sourceRows[i])
} else if i < insertionLine+len(prototypesRows) {
fmt.Printf("PRO|%s\n", prototypesRows[i-insertionLine])
} else {
fmt.Printf(" |%s\n", sourceRows[i-len(prototypesRows)])
}
}
fmt.Println("#END OF PREPROCESSED SOURCE")
}
ctx.Source = source
return nil
}
func composePrototypeSection(line int, prototypes []*types.Prototype) string {
if len(prototypes) == 0 {
return constants.EMPTY_STRING
}
str := joinPrototypes(prototypes)
str += "\n#line "
str += strconv.Itoa(line)
str += " " + utils.QuoteCppString(prototypes[0].File)
str += "\n"
return str
}
func joinPrototypes(prototypes []*types.Prototype) string {
prototypesSlice := []string{}
for _, proto := range prototypes {
if signatureContainsaDefaultArg(proto) {
continue
}
prototypesSlice = append(prototypesSlice, "#line "+strconv.Itoa(proto.Line)+" "+utils.QuoteCppString(proto.File))
prototypeParts := []string{}
if proto.Modifiers != "" {
prototypeParts = append(prototypeParts, proto.Modifiers)
}
prototypeParts = append(prototypeParts, proto.Prototype)
prototypesSlice = append(prototypesSlice, strings.Join(prototypeParts, " "))
}
return strings.Join(prototypesSlice, "\n")
}
func signatureContainsaDefaultArg(proto *types.Prototype) bool {
return strings.Contains(proto.Prototype, "=")
}
func isFirstFunctionOutsideOfSource(firstFunctionLine int, sourceRows []string) bool {
return firstFunctionLine > len(sourceRows)-1
}