chiark / gitweb /
all: make newer pycodestyle happy
[fdroidserver.git] / fdroidserver / import.py
1 #!/usr/bin/env python3
2 #
3 # import.py - part of the FDroid server tools
4 # Copyright (C) 2010-13, Ciaran Gultnieks, ciaran@ciarang.com
5 # Copyright (C) 2013-2014 Daniel Martí <mvdan@mvdan.cc>
6 #
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU Affero General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
11 #
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU Affero General Public License for more details.
16 #
17 # You should have received a copy of the GNU Affero General Public License
18 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
19
20 import binascii
21 import sys
22 import os
23 import shutil
24 import urllib.request
25 from argparse import ArgumentParser
26 from configparser import ConfigParser
27 import logging
28
29 from . import common
30 from . import metadata
31
32
33 # Get the repo type and address from the given web page. The page is scanned
34 # in a rather naive manner for 'git clone xxxx', 'hg clone xxxx', etc, and
35 # when one of these is found it's assumed that's the information we want.
36 # Returns repotype, address, or None, reason
37 def getrepofrompage(url):
38
39     req = urllib.request.urlopen(url)
40     if req.getcode() != 200:
41         return (None, 'Unable to get ' + url + ' - return code ' + str(req.getcode()))
42     page = req.read()
43
44     # Works for BitBucket
45     index = page.find('hg clone')
46     if index != -1:
47         repotype = 'hg'
48         repo = page[index + 9:]
49         index = repo.find('<')
50         if index == -1:
51             return (None, "Error while getting repo address")
52         repo = repo[:index]
53         repo = repo.split('"')[0]
54         return (repotype, repo)
55
56     # Works for BitBucket
57     index = page.find('git clone')
58     if index != -1:
59         repotype = 'git'
60         repo = page[index + 10:]
61         index = repo.find('<')
62         if index == -1:
63             return (None, "Error while getting repo address")
64         repo = repo[:index]
65         repo = repo.split('"')[0]
66         return (repotype, repo)
67
68     return (None, "No information found." + page)
69
70
71 config = None
72 options = None
73
74
75 def get_metadata_from_url(app, url):
76
77     tmp_dir = 'tmp'
78     if not os.path.isdir(tmp_dir):
79         logging.info("Creating temporary directory")
80         os.makedirs(tmp_dir)
81
82     # Figure out what kind of project it is...
83     projecttype = None
84     app.WebSite = url  # by default, we might override it
85     if url.startswith('git://'):
86         projecttype = 'git'
87         repo = url
88         repotype = 'git'
89         app.SourceCode = ""
90         app.WebSite = ""
91     elif url.startswith('https://github.com'):
92         projecttype = 'github'
93         repo = url
94         repotype = 'git'
95         app.SourceCode = url
96         app.IssueTracker = url + '/issues'
97         app.WebSite = ""
98     elif url.startswith('https://gitlab.com/'):
99         projecttype = 'gitlab'
100         # git can be fussy with gitlab URLs unless they end in .git
101         if url.endswith('.git'):
102             url = url[:-4]
103         repo = url + '.git'
104         repotype = 'git'
105         app.WebSite = url
106         app.SourceCode = url + '/tree/HEAD'
107         app.IssueTracker = url + '/issues'
108     elif url.startswith('https://bitbucket.org/'):
109         if url.endswith('/'):
110             url = url[:-1]
111         projecttype = 'bitbucket'
112         app.SourceCode = url + '/src'
113         app.IssueTracker = url + '/issues'
114         # Figure out the repo type and adddress...
115         repotype, repo = getrepofrompage(app.SourceCode)
116         if not repotype:
117             logging.error("Unable to determine vcs type. " + repo)
118             sys.exit(1)
119     if not projecttype:
120         logging.error("Unable to determine the project type.")
121         logging.error("The URL you supplied was not in one of the supported formats. Please consult")
122         logging.error("the manual for a list of supported formats, and supply one of those.")
123         sys.exit(1)
124
125     # Ensure we have a sensible-looking repo address at this point. If not, we
126     # might have got a page format we weren't expecting. (Note that we
127     # specifically don't want git@...)
128     if ((repotype != 'bzr' and (not repo.startswith('http://') and
129         not repo.startswith('https://') and
130         not repo.startswith('git://'))) or
131             ' ' in repo):
132         logging.error("Repo address '{0}' does not seem to be valid".format(repo))
133         sys.exit(1)
134
135     # Get a copy of the source so we can extract some info...
136     logging.info('Getting source from ' + repotype + ' repo at ' + repo)
137     build_dir = os.path.join(tmp_dir, 'importer')
138     if os.path.exists(build_dir):
139         shutil.rmtree(build_dir)
140     vcs = common.getvcs(repotype, repo, build_dir)
141     vcs.gotorevision(options.rev)
142     root_dir = get_subdir(build_dir)
143
144     app.RepoType = repotype
145     app.Repo = repo
146
147     return root_dir, build_dir
148
149
150 config = None
151 options = None
152
153
154 def get_subdir(build_dir):
155     if options.subdir:
156         return os.path.join(build_dir, options.subdir)
157
158     return build_dir
159
160
161 def main():
162
163     global config, options
164
165     # Parse command line...
166     parser = ArgumentParser()
167     common.setup_global_opts(parser)
168     parser.add_argument("-u", "--url", default=None,
169                         help="Project URL to import from.")
170     parser.add_argument("-s", "--subdir", default=None,
171                         help="Path to main android project subdirectory, if not in root.")
172     parser.add_argument("--rev", default=None,
173                         help="Allows a different revision (or git branch) to be specified for the initial import")
174     metadata.add_metadata_arguments(parser)
175     options = parser.parse_args()
176     metadata.warnings_action = options.W
177
178     config = common.read_config(options)
179
180     apps = metadata.read_metadata()
181     app = metadata.App()
182     app.UpdateCheckMode = "Tags"
183
184     root_dir = None
185     build_dir = None
186
187     local_metadata_files = common.get_local_metadata_files()
188     if local_metadata_files != []:
189         logging.error("This repo already has local metadata: %s" % local_metadata_files[0])
190         sys.exit(1)
191
192     if options.url is None and os.path.isdir('.git'):
193         app.AutoName = os.path.basename(os.getcwd())
194         app.RepoType = 'git'
195
196         build = {}
197         root_dir = get_subdir(os.getcwd())
198         if os.path.exists('build.gradle'):
199             build.gradle = ['yes']
200
201         import git
202         repo = git.repo.Repo(root_dir)  # git repo
203         for remote in git.Remote.iter_items(repo):
204             if remote.name == 'origin':
205                 url = repo.remotes.origin.url
206                 if url.startswith('https://git'):  # github, gitlab
207                     app.SourceCode = url.rstrip('.git')
208                 app.Repo = url
209                 break
210         # repo.head.commit.binsha is a bytearray stored in a str
211         build.commit = binascii.hexlify(bytearray(repo.head.commit.binsha))
212         write_local_file = True
213     elif options.url:
214         root_dir, build_dir = get_metadata_from_url(app, options.url)
215         build = metadata.Build()
216         build.commit = '?'
217         build.disable = 'Generated by import.py - check/set version fields and commit id'
218         write_local_file = False
219     else:
220         logging.error("Specify project url.")
221         sys.exit(1)
222
223     # Extract some information...
224     paths = common.manifest_paths(root_dir, [])
225     if paths:
226
227         version, vercode, package = common.parse_androidmanifests(paths, app)
228         if not package:
229             logging.error("Couldn't find package ID")
230             sys.exit(1)
231         if not version:
232             logging.warn("Couldn't find latest version name")
233         if not vercode:
234             logging.warn("Couldn't find latest version code")
235     else:
236         spec = os.path.join(root_dir, 'buildozer.spec')
237         if os.path.exists(spec):
238             defaults = {'orientation': 'landscape', 'icon': '',
239                         'permissions': '', 'android.api': "18"}
240             bconfig = ConfigParser(defaults, allow_no_value=True)
241             bconfig.read(spec)
242             package = bconfig.get('app', 'package.domain') + '.' + bconfig.get('app', 'package.name')
243             version = bconfig.get('app', 'version')
244             vercode = None
245         else:
246             logging.error("No android or kivy project could be found. Specify --subdir?")
247             sys.exit(1)
248
249     # Make sure it's actually new...
250     if package in apps:
251         logging.error("Package " + package + " already exists")
252         sys.exit(1)
253
254     # Create a build line...
255     build.version = version or '?'
256     build.vercode = vercode or '?'
257     if options.subdir:
258         build.subdir = options.subdir
259     if os.path.exists(os.path.join(root_dir, 'jni')):
260         build.buildjni = ['yes']
261
262     metadata.post_metadata_parse(app)
263
264     app.builds.append(build)
265
266     if write_local_file:
267         metadata.write_metadata('.fdroid.yml', app)
268     else:
269         # Keep the repo directory to save bandwidth...
270         if not os.path.exists('build'):
271             os.mkdir('build')
272         if build_dir is not None:
273             shutil.move(build_dir, os.path.join('build', package))
274         with open('build/.fdroidvcs-' + package, 'w') as f:
275             f.write(app.RepoType + ' ' + app.Repo)
276
277         metadatapath = os.path.join('metadata', package + '.txt')
278         metadata.write_metadata(metadatapath, app)
279         logging.info("Wrote " + metadatapath)
280
281
282 if __name__ == "__main__":
283     main()