manifest_versions.py revision a8af9a7a2462b00e72deff99327bdb452a715277 
1# Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
2# Use of this source code is governed by a BSD-style license that can be
3# found in the LICENSE file.
4"""Tools for searching/manipulating the manifests repository."""
5
6from __future__ import print_function
7
8__author__ = 'llozano@google.com (Luis Lozano)'
9
10import os
11import re
12import shutil
13import tempfile
14import time
15
16import command_executer
17import logger
18
19
20def IsCrosVersion(version):
21  match = re.search(r'(\d+\.\d+\.\d+\.\d+)', version)
22  return match is not None
23
24
25def IsRFormatCrosVersion(version):
26  match = re.search(r'(R\d+-\d+\.\d+\.\d+)', version)
27  return match is not None
28
29
30def RFormatCrosVersion(version):
31  assert IsCrosVersion(version)
32  tmp_major, tmp_minor = version.split('.', 1)
33  rformat = 'R' + tmp_major + '-' + tmp_minor
34  assert IsRFormatCrosVersion(rformat)
35  return rformat
36
37
38class ManifestVersions(object):
39  """This class handles interactions with the manifests repo."""
40
41  def __init__(self, internal=True):
42    self.internal = internal
43    self.clone_location = tempfile.mkdtemp()
44    self.ce = command_executer.GetCommandExecuter()
45    if internal:
46      versions_git = ('https://chrome-internal.googlesource.com/'
47                      'chromeos/manifest-versions.git')
48    else:
49      versions_git = (
50          'https://chromium.googlesource.com/chromiumos/manifest-versions.git')
51    commands = ['cd {0}'.format(self.clone_location),
52                'git clone {0}'.format(versions_git)]
53    ret = self.ce.RunCommands(commands)
54    if ret:
55      logger.GetLogger().LogFatal('Failed to clone manifest-versions.')
56
57  def __del__(self):
58    if self.clone_location:
59      shutil.rmtree(self.clone_location)
60
61  def TimeToVersion(self, my_time):
62    """Convert timestamp to version number."""
63    cur_time = time.mktime(time.gmtime())
64    des_time = float(my_time)
65    if cur_time - des_time > 7000000:
66      logger.GetLogger().LogFatal('The time you specify is too early.')
67    commands = ['cd {0}'.format(self.clone_location), 'cd manifest-versions',
68                'git checkout -f $(git rev-list' +
69                ' --max-count=1 --before={0} origin/master)'.format(my_time)]
70    ret = self.ce.RunCommands(commands)
71    if ret:
72      logger.GetLogger().LogFatal('Failed to checkout manifest at '
73                                  'specified time')
74    path = os.path.realpath('{0}/manifest-versions/LKGM/lkgm.xml'.format(
75        self.clone_location))
76    pp = path.split('/')
77    small = os.path.basename(path).split('.xml')[0]
78    version = pp[-2] + '.' + small
79    commands = ['cd {0}'.format(self.clone_location), 'cd manifest-versions',
80                'git checkout master']
81    self.ce.RunCommands(commands)
82    return version
83
84  def GetManifest(self, version, to_file):
85    """Get the manifest file from a given chromeos-internal version."""
86    assert not IsRFormatCrosVersion(version)
87    version = version.split('.', 1)[1]
88    os.chdir(self.clone_location)
89    files = [os.path.join(r, f)
90             for r, _, fs in os.walk('.') for f in fs if version in f]
91    if files:
92      command = 'cp {0} {1}'.format(files[0], to_file)
93      ret = self.ce.RunCommand(command)
94      if ret:
95        raise Exception('Cannot copy manifest to {0}'.format(to_file))
96    else:
97      raise Exception('Version {0} is not available.'.format(version))
98