31 class NoRepo(Exception): pass

    51 

    52 # CVS conversion code inspired by hg-cvs-import and git-cvsimport

    53 class convert_cvs:

    54     def __init__(self, path):

    55         self.path = path

    56         cvs = os.path.join(path, "CVS")

    57         if not os.path.exists(cvs):

    58             raise NoRepo("couldn't open CVS repo %s" % path)

    59 

    60         self.changeset = {}

    61         self.tags = {}

    62         self.lastbranch = {}

    63         self.parent = {}

    64         self.socket = None

    65         self.cvsroot = file(os.path.join(cvs, "Root")).read()[:-1]

    66         self.cvsrepo = file(os.path.join(cvs, "Repository")).read()[:-1]

    67         self.encoding = locale.getpreferredencoding()

    68         self._parse()

    69         self._connect()

    70 

    71     def _parse(self):

    72         if self.changeset:

    73             return

    74 

    75         d = os.getcwd()

    76         try:

    77             os.chdir(self.path)

    78             id = None

    79             state = 0

    80             for l in os.popen("cvsps -A"):

    81                 if state == 0: # header

    82                     if l.startswith("PatchSet"):

    83                         id = l[9:-2]

    84                     elif l.startswith("Date"):

    85                         date = util.parsedate(l[6:-1], ["%Y/%m/%d %H:%M:%S"])

    86                         date = util.datestr(date)

    87                     elif l.startswith("Branch"):

    88                         branch = l[8:-1]

    89                         self.parent[id] = self.lastbranch.get(branch,'bad')

    90                         self.lastbranch[branch] = id

    91                     elif l.startswith("Ancestor branch"):

    92                         ancestor = l[17:-1]

    93                         self.parent[id] = self.lastbranch[ancestor]

    94                     elif l.startswith("Author"):

    95                         author = self.recode(l[8:-1])

    96                     elif l.startswith("Tag: "):

    97                         t = l[5:-1]

    98                         if t != "(none) ":

    99                             self.tags[t] = id

   100                     elif l.startswith("Log:"):

   101                         state = 1

   102                         log = ""

   103                 elif state == 1: # log

   104                     if l == "Members: \n":

   105                         files = {}

   106                         log = self.recode(log[:-1])

   107                         if log.isspace():

   108                             log = "*** empty log message ***\n"

   109                         state = 2

   110                     else:

   111                         log += l

   112                 elif state == 2:

   113                     if l == "\n": #

   114                         state = 0

   115                         self.changeset[id] = (date, author, log, files)

   116                     else:

   117                         file,rev = l[1:-2].rsplit(':',1)

   118                         rev = rev.split("->")[1]

   119                         files[file] = rev

   120 

   121             self.heads = self.lastbranch.values()

   122         finally:

   123             os.chdir(d)

   124 

   125     def _connect(self):

   126         root = self.cvsroot

   127         local = False

   128         user, host = None, None

   129         cmd = ['cvs', 'server']

   130 

   131         status("connecting to %s\n" % root)

   132 

   133         # only non-pserver for now

   134         if root.startswith(":pserver"):

   135             abort("can't handle pserver mode yet: %s\n" % root)

   136 

   137         if root.startswith(":local:"):

   138             local = True

   139             root = root[7:]

   140         else:

   141             # :ext:user@host/home/user/path/to/cvsroot

   142             if root.startswith(":ext:"):

   143                 root = root[5:]

   144             m = re.match(r'(?:([^@:/]+)@)?([^:/]+):?(.*)', root)

   145             if not m:

   146                 local = True

   147             else:

   148                 local = False

   149                 user, host, root = m.group(1), m.group(2), m.group(3)

   150 

   151         if not local:

   152             rsh = os.environ.get("CVS_RSH" or "rsh")

   153             if user:

   154                 cmd = [rsh, '-l', user, host] + cmd

   155             else:

   156                 cmd = [rsh, host] + cmd

   157 

   158         self.writep, self.readp = os.popen2(cmd)

   159         self.realroot = root

   160 

   161         self.writep.write("Root %s\n" % root)

   162         self.writep.write("Valid-responses ok error Valid-requests Mode"

   163                           " M Mbinary E Checked-in Created Updated"

   164                           " Merged Removed\n")

   165         self.writep.write("valid-requests\n")

   166         self.writep.flush()

   167         r = self.readp.readline()

   168         if not r.startswith("Valid-requests"):

   169             abort("server sucks\n")

   170         if "UseUnchanged" in r:

   171             self.writep.write("UseUnchanged\n")

   172             self.writep.flush()

   173             r = self.readp.readline()

   174 

   175     def getheads(self):

   176         return self.heads

   177 

   178     def getfile(self, name, rev):

   179         if rev.endswith("(DEAD)"):

   180             raise IOError

   181 

   182         args = ("-N -P -kk -r %s --" % rev).split()

   183         args.append(os.path.join(self.cvsrepo, name))

   184         for x in args:

   185             self.writep.write("Argument %s\n" % x)

   186         self.writep.write("Directory .\n%s\nco\n" % self.realroot)

   187         self.writep.flush()

   188 

   189         data = ""

   190         while 1:

   191             line = self.readp.readline()

   192             if line.startswith("Created ") or line.startswith("Updated "):

   193                 self.readp.readline() # path

   194                 self.readp.readline() # entries

   195                 mode = self.readp.readline()[:-1]

   196                 count = int(self.readp.readline()[:-1])

   197                 data = self.readp.read(count)

   198             elif line.startswith(" "):

   199                 data += line[1:]

   200             elif line.startswith("M "):

   201                 pass

   202             elif line.startswith("Mbinary "):

   203                 count = int(self.readp.readline()[:-1])

   204                 data = self.readp.read(count)

   205             else:

   206                 if line == "ok\n":

   207                     return data

   208                 elif line.startswith("E "):

   209                     warn("cvs server: %s\n" % line[2:])

   210                 elif line.startswith("Remove"):

   211                     l = self.readp.readline()

   212                     l = self.readp.readline()

   213                     if l != "ok\n":

   214                         abort("unknown CVS response: %s\n" % l)

   215                 else:

   216                     abort("unknown CVS response: %s\n" % line)

   217 

   218     def getchanges(self, rev):

   219         files = self.changeset[rev][3]

   220         cl = [ (f, r, 0) for f,r in files.items() ]

   221         cl.sort()

   222         return cl

   223 

   224     def recode(self, text):

   225         return text.decode(self.encoding, "replace").encode("utf-8")

   226 

   227     def getcommit(self, rev):

   228         cs = self.changeset[rev]

   229         parents = [self.parent[rev]]

   230         if rev == "1":

   231             parents = []

   232         return (parents, cs[1], cs[0], cs[2])

   233 

   234     def gettags(self):

   235         return self.tags