Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

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

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*- 

# 

# Copyright (C) 2015 Canonical Ltd 

# 

# This program is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License version 3 as 

# published by the Free Software Foundation. 

# 

# This program is distributed in the hope that it will be useful, 

# but WITHOUT ANY WARRANTY; without even the implied warranty of 

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the 

# GNU General Public License for more details. 

# 

# You should have received a copy of the GNU General Public License 

# along with this program.  If not, see <http://www.gnu.org/licenses/>. 

 

"""The go plugin can be used for go projects using `go get`. 

 

This plugin uses the common plugin keywords, for more information check the 

'plugins' topic. 

 

This plugin uses the common plugin keywords as well as those for "sources". 

For more information check the 'plugins' topic for the former and the 

'sources' topic for the latter. 

 

Additionally, this plugin uses the following plugin-specific keywords: 

 

    - go-packages: 

      (list of strings) 

      Go packages to fetch, these must be a "main" package. Dependencies 

      are pulled in automatically by `go get`. 

      Packages that are not "main" will not cause an error, but would 

      not be useful either. 

""" 

 

import os 

import shutil 

 

import snapcraft 

 

 

class GoPlugin(snapcraft.BasePlugin): 

 

    @classmethod 

    def schema(cls): 

        schema = super().schema() 

        schema['properties']['go-packages'] = { 

            'type': 'array', 

            'minitems': 1, 

            'uniqueItems': True, 

            'items': { 

                'type': 'string', 

            }, 

            'default': [], 

        } 

 

        if 'required' in schema: 

            del schema['required'] 

 

        return schema 

 

    def __init__(self, name, options): 

        super().__init__(name, options) 

        self.build_packages.append('golang-go') 

        self._gopath = os.path.join(self.partdir, 'go') 

        self._gopath_src = os.path.join(self._gopath, 'src') 

        self._gopath_bin = os.path.join(self._gopath, 'bin') 

 

    def env(self, root): 

        # usr/lib/go/bin on newer Ubuntus, usr/bin on trusty 

        env = [ 

            'GOPATH={}/go'.format(root), 

            'CGO_LDFLAGS=$CGO_LDFLAGS"' + ' '.join([ 

                '-L{0}/lib', 

                '-L{0}/usr/lib', 

                '-L{0}/lib/{1}', 

                '-L{0}/usr/lib/{1}', 

                '$LDFLAGS' 

            ]).format(root, snapcraft.common.get_arch_triplet()) + '"', 

        ] 

        return env 

 

    def pull(self): 

        # use -d to only download (build will happen later) 

        # use -t to also get the test-deps 

        super().pull() 

        os.makedirs(self._gopath_src, exist_ok=True) 

        if self.options.source is not None: 

            self._local_pull() 

        self._remote_pull() 

 

    def _local_pull(self): 

        go_package = os.path.basename(os.path.abspath(self.options.source)) 

        local_path = os.path.join(self._gopath_src, go_package) 

96        if os.path.islink(local_path): 

            os.unlink(local_path) 

        os.symlink(self.sourcedir, local_path) 

        self._run(['go', 'get', '-t', '-d', './{}/...'.format(go_package)]) 

 

    def _remote_pull(self): 

        for go_package in self.options.go_packages: 

            self._run(['go', 'get', '-t', '-d', go_package]) 

 

    def build(self): 

        super().build() 

        if self.options.source is not None: 

            self._local_build() 

        self._remote_build() 

 

        install_bin_path = os.path.join(self.installdir, 'bin') 

        os.makedirs(install_bin_path, exist_ok=True) 

        os.makedirs(self._gopath_bin, exist_ok=True) 

        for binary in os.listdir(os.path.join(self._gopath_bin)): 

            binary_path = os.path.join(self._gopath_bin, binary) 

            shutil.copy2(binary_path, install_bin_path) 

 

    def _local_build(self): 

        go_package = os.path.basename(os.path.abspath(self.options.source)) 

        self._run(['go', 'install', './{}/...'.format(go_package)]) 

 

    def _remote_build(self): 

        for go_package in self.options.go_packages: 

            self._run(['go', 'install', go_package]) 

 

    def _run(self, cmd, **kwargs): 

        cmd = ['env', 'GOPATH={}'.format(self._gopath)] + cmd 

        return self.run(cmd, cwd=self._gopath_src, **kwargs)