deskflow/CI/build_version.py

import subprocess

class VersionPart:

   def __init__(self, part = ''):
      self.prefix = ''
      self.suffix = ''
      self.number = 0

      if part:
         self.__parsePreffix(part)
         self.__parseNumber(part)
         self.__parseSuffix(part)

   def __parsePreffix(self, part):
      if not part[0].isdigit():
         for i in part:
            if not i.isdigit():
               self.prefix += i
            else:
               break

   def __parseNumber(self, part):
      start = len(self.prefix)
      end = part.find('-')
      if end > 0:
         self.number = int(part[start:end])
      else:
         self.number = int(part[start:])

   def __parseSuffix(self, part):
      items = part.split('-')
      if len(items) == 2:
         self.suffix = '-' + items[1]

   def __str__(self):
      return self.prefix + str(self.number) + self.suffix
class Version:

   def __init__(self, version):
      versionParts = version.split('.')

      if len(versionParts) == 3:
         self.major = VersionPart(versionParts[0])
         self.minor = VersionPart(versionParts[1])
         self.build = VersionPart(versionParts[2])
         self.patch = VersionPart()
         self.patch.number = self.build.number
      elif len(versionParts) == 4:
         self.major = VersionPart(versionParts[0])
         self.minor = VersionPart(versionParts[1])
         self.patch = VersionPart(versionParts[2])
         self.build = VersionPart(versionParts[3])
      else:
         print('ERROR: Wrong version number')

   def isSamePatch(self, version):
      return (self.major.number == version.major.number and
              self.minor.number == version.minor.number and
              self.patch.number == version.patch.number)

   def __str__(self):
      result  = str(self.major) + '.'
      result += str(self.minor) + '.'
      result += str(self.patch) + '.'
      result += str(self.build) 
      return  result

class VersionFile:
   def __init__(self, file):
      self.file = file

   def setOption(self, name, value):
      fp = open(self.file, 'rt+')
      content = ''
      for line in fp:
         print(line)
         if line.find(name) != -1:
            line = 'set ('+ name + ' ' + value + ')'
         content += line
      fp.write(content)
      fp.close()

def findVersion(versions, cmakeVersion):
   gitVersion = Version(versions[0])
   for version in versions:
      ver = Version(version)
      if ver.isSamePatch(cmakeVersion):
         gitVersion = ver
         break
   print('INFO: Version '+ str(gitVersion) + ' has been read from git')
   return gitVersion

def getVersionFromGit(cmakeVersion):
   try:
      taggedRevision = subprocess.check_output(('git rev-list --tags --max-count=100').split(),  universal_newlines=True)
      cmd = ('git describe --tags ' + taggedRevision).split()
      versions = subprocess.check_output(cmd, universal_newlines=True).split()
      return findVersion(versions, cmakeVersion)
   except subprocess.CalledProcessError:
      print('ERROR: Unable to get version from git')
      exit(1)

def updateVersionFile(number):
   fp = open('cmake/Version.cmake')
   content = fp.read()
   fp.close()

   fp = open('cmake/Version.cmake', 'wt')
   fp.write(content.replace('set (SYNERGY_VERSION_BUILD 1)', 'set (SYNERGY_VERSION_BUILD ' + str(number) + ')'))
   fp.close()

def getOptionValue(source, option):
   start = source.find(option)
   if (start != -1):
      start += len(option) + 1
      end = source.find(')', start)
      return source[start : end]
   else:
      print("ERROR: Can't find option <" + option + ">");

def getVersionFromFile():
   fp = open('cmake/Version.cmake')
   content = fp.read()
   fp.close()

   major = getOptionValue(content, 'SYNERGY_VERSION_MAJOR')
   minor = getOptionValue(content, 'SYNERGY_VERSION_MINOR')
   patch = getOptionValue(content, 'SYNERGY_VERSION_PATCH')

   return Version(major + '.' + minor + '.' + patch)

if __name__ == '__main__':
   cmakeVersion = getVersionFromFile()
   gitVersion = getVersionFromGit(cmakeVersion)
   updateVersionFile(gitVersion.build.number)
   print('INFO: Build number is: <' + str(gitVersion) + '>')