#!/usr/bin/python
# ==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
# ------------------------------------------------------------------------------

USAGE = '''
This script needs to have write access to the directory /labels (off of root,
not your home directory) on Linux or c:\labels on Windows in order to work
properly.

Usage:
  convert_label_to_autoreload.py <label or file_with_list_of_labels>
'''

import os
import os.path
import re
import sys
import p4lock
import p4unlock

def check_auto(label, auto):
  for line in os.popen('p4 label -o %s' % label).readlines():
    if re.search("^Version:", line):
      auto = 1
    if re.search(" autoreload", line):
      auto = 1
  return(auto)

def droplabellines(labelname):
  input = open('%s.labelfiles.tmp' % (labelname), 'r')
  output = open('%s.labelfiles' % (labelname), 'w')

  for line in input.readlines():
    line = line.rstrip()
    def rep(mo): return mo.group(1)
    line = re.sub(r'(#\d+) - .*', rep, line)
    output.write(line + '\n')

  input.close()
  output.close()
  os.remove('%s.labelfiles.tmp' % (labelname))

def convertlabel(label):
    auto = 0
    auto = check_auto(label, auto)
    if auto:
      return()
    
    # The next two lines are replacing any illegal file name characters with
    # and exclamation point and the hex value of the replaced character.
    def rep(mo): return '_%s' % hex(ord(mo.group(0)))[-2:]
    labelfile = re.sub(r'[\@=`~+<>/:\|\\ ,\(\)\[\]\$%^&?;{}\'\"]', rep, label)
    os.system('p4 label -o "%s" | sed s/noautoreload/autoreload/ > %s.labelspec' % (label, labelfile))
    os.system('p4 files "@%s" > %s.labelfiles.tmp' % (label, labelfile))
    droplabellines(labelfile)
    if (os.path.isfile('%s.labelspec' % (labelfile)) and os.path.isfile('%s.labelfiles' % (labelfile))):
      os.system('p4 label -f -d "%s"' % (label))
      os.system('p4 label -i < %s.labelspec' % (labelfile))
      p4unlock.unlock(label)
      os.system('p4 -x %s.labelfiles labelsync -l "%s"' % ( labelfile, label))
      p4lock.lock(label)
      os.remove('%s.labelspec' % (labelfile))
      os.remove('%s.labelfiles' % (labelfile))

###############################################################################
# main
if __name__ == '__main__':
  if len (sys.argv) < 2:
    print USAGE
    sys.exit (1)

  try:
    labellistfile = open( sys.argv[1], "r" )
    for line in labellistfile.readlines():
      line = line.strip()
      if line:
        convertlabel(sys.argv[1].rstrip())
    labellistfile.close()
  except:
    print "No file " + sys.argv[1] + " available, assuming it is the actual label name"
    convertlabel(sys.argv[1].rstrip())