#!/usr/bin/python

import os
import sys
import datetime
import shutil
import copy

class Version:
    def __init__(self, s):
        self.pre = False
        self.beta = None

        if s.startswith("'"):
            s = s[1:]
        if s.endswith("'"):
            s = s[0:-1]
        
        if s.endswith('pre'):
            s = s[0:-3]
            self.pre = True

        b = s.find("beta")
        if b != -1:
            self.beta = int(s[b+4:])
            s = s[0:b]

        p = s.split('.')
        self.major = int(p[0])
        self.minor = int(p[1])

    def bump(self):
        self.minor += 1
        self.pre = False
        self.beta = None

    def to_pre(self):
        self.pre = True
        self.beta = None

    def bump_and_to_pre(self):
        self.bump()
        self.pre = True
        self.beta = None

    def to_release(self):
        self.pre = False
        self.beta = None

    def bump_beta(self):
        if self.pre:
            self.pre = False
            self.beta = 1
        elif self.beta is not None:
            self.beta += 1
        elif self.beta is None:
            self.beta = 1

    def __str__(self):
        s = '%d.%02d' % (self.major, self.minor)
        if self.beta is not None:
            s += 'beta%d' % self.beta
        elif self.pre:
            s += 'pre'

        return s
        
def rewrite_wscript(method):
    f = open('wscript', 'rw')
    o = open('wscript.tmp', 'w')
    version = None
    while 1:
        l = f.readline()
        if l == '':
            break

        s = l.split()
        if len(s) == 3 and s[0] == "VERSION":
            version = Version(s[2])
            method(version)
            print "Writing %s" % version
            print >>o,"VERSION = '%s'" % version
        else:
            print >>o,l,
    f.close()
    o.close()

    os.rename('wscript.tmp', 'wscript')
    return version

def append_to_changelog(version):
    f = open('ChangeLog', 'r')
    c = f.read()
    f.close()

    f = open('ChangeLog', 'w')
    now = datetime.datetime.now()
    f.write('%d-%02d-%02d  Carl Hetherington  <cth@carlh.net>\n\n\t* Version %s released.\n\n' % (now.year, now.month, now.day, version))
    f.write(c)