1

Option Explicit

     2

     3

Dim FSO

     4

Set FSO = CreateObject("Scripting.FileSystemObject")

     5

     6

Sub FindReplaceInFile(filename, to_find, replacement)

     7

	Dim file, data

     8

	Set file = FSO.OpenTextFile(filename, 1, 0, 0)

     9

	data = file.ReadAll

    10

	file.Close

    11

	data = Replace(data, to_find, replacement)

    12

	Set file = FSO.CreateTextFile(filename, -1, 0)

    13

	file.Write data

    14

	file.Close

    15

End Sub

    16

    17

Sub UpdateFile(modified, revision, version, cur_date, filename)

    18

	FSO.CopyFile filename & ".in", filename

    19

	FindReplaceInFile filename, "@@MODIFIED@@", modified

    20

	FindReplaceInFile filename, "@@REVISION@@", revision

    21

	FindReplaceInFile filename, "@@VERSION@@", version

    22

	FindReplaceInFile filename, "@@DATE@@", cur_date

    23

End Sub

    24

    25

Sub UpdateFiles(version)

    26

	Dim modified, revision, cur_date

    27

	cur_date = DatePart("D", Date) & "." & DatePart("M", Date) & "." & DatePart("YYYY", Date)

    28

    29

	If InStr(version, Chr(9)) Then

    30

		revision = Mid(version, InStr(version, Chr(9)) + 1)

    31

		revision = Mid(revision, 1, InStr(revision, Chr(9)) - 1)

    32

		modified = Mid(version, InStrRev(version, Chr(9)) + 1)

    33

		version  = Mid(version, 1, InStr(version, Chr(9)) - 1)

    34

	Else

    35

		revision = 0

    36

		modified = 1

    37

	End If

    38

    39

	UpdateFile modified, revision, version, cur_date, "../src/rev.cpp"

    40

	UpdateFile modified, revision, version, cur_date, "../src/ottdres.rc"

    41

End Sub

    42

    43

Function ReadRegistryKey(shive, subkey, valuename, architecture)

    44

	Dim hiveKey, objCtx, objLocator, objServices, objReg, Inparams, Outparams

    45

    46

	' First, get the Registry Provider for the requested architecture

    47

	Set objCtx = CreateObject("WbemScripting.SWbemNamedValueSet")

    48

	objCtx.Add "__ProviderArchitecture", architecture ' Must be 64 of 32

    49

	Set objLocator = CreateObject("Wbemscripting.SWbemLocator")

    50

	Set objServices = objLocator.ConnectServer("","root\default","","",,,,objCtx)

    51

	Set objReg = objServices.Get("StdRegProv")

    52

    53

	' Check the hive and give it the right value

    54

	Select Case shive

    55

		Case "HKCR", "HKEY_CLASSES_ROOT"

    56

			hiveKey = &h80000000

    57

		Case "HKCU", "HKEY_CURRENT_USER"

    58

			hiveKey = &H80000001

    59

		Case "HKLM", "HKEY_LOCAL_MACHINE"

    60

			hiveKey = &h80000002

    61

		Case "HKU", "HKEY_USERS"

    62

			hiveKey = &h80000003

    63

		Case "HKCC", "HKEY_CURRENT_CONFIG"

    64

			hiveKey = &h80000005

    65

		Case "HKDD", "HKEY_DYN_DATA" ' Only valid for Windows 95/98

    66

			hiveKey = &h80000006

    67

		Case Else

    68

			MsgBox "Hive not valid (ReadRegistryKey)"

    69

	End Select

    70

    71

	Set Inparams = objReg.Methods_("GetStringValue").Inparameters

    72

	Inparams.Hdefkey = hiveKey

    73

	Inparams.Ssubkeyname = subkey

    74

	Inparams.Svaluename = valuename

    75

	Set Outparams = objReg.ExecMethod_("GetStringValue", Inparams,,objCtx)

    76

    77

	ReadRegistryKey = Outparams.SValue

    78

End Function

    79

    80

Function DetermineSVNVersion()

    81

	Dim WshShell, version, branch, modified, revision, url, oExec, line, hash

    82

	Set WshShell = CreateObject("WScript.Shell")

    83

	On Error Resume Next

    84

    87

	' Try TortoiseSVN

    88

	' Get the directory where TortoiseSVN (should) reside(s)

    89

	Dim sTortoise

    90

	' First, try with 32-bit architecture

    91

	sTortoise = ReadRegistryKey("HKLM", "SOFTWARE\TortoiseSVN", "Directory", 32)

    93

		' No 32-bit version of TortoiseSVN installed, try 64-bit version (doesn't hurt on 32-bit machines, it returns nothing or is ignored)

    94

		sTortoise = ReadRegistryKey("HKLM", "SOFTWARE\TortoiseSVN", "Directory", 64)

    95

	End If

    96

    97

	' If TortoiseSVN is installed, try to get the revision number

   107

	End If

   108

   109

	' Looks like there is no TortoiseSVN installed either. Then we don't know it.

   111

		' Reset error and version

   112

		Err.Clear

   113

		version = "norev000"

   114

		modified = 0

   115

   116

		' Set the environment to english

   117

		WshShell.Environment("PROCESS")("LANG") = "en"

   118

   119

		' Do we have subversion installed? Check immediatelly whether we've got a modified WC.

   120

		Set oExec = WshShell.Exec("svnversion ../src")

   121

		If Err.Number = 0 Then

   122

			' Wait till the application is finished ...

   123

			Do While oExec.Status = 0

   124

			Loop

   125

   126

			line = OExec.StdOut.ReadLine()

   127

			If line <> "exported" Then

   128

				If InStr(line, "M") Then

   129

					modified = 2

   130

				End If

   131

   132

				' And use svn info to get the correct revision and branch information.

   133

				Set oExec = WshShell.Exec("svn info ../src")

   134

				If Err.Number = 0 Then

   135

					Do

   136

						line = OExec.StdOut.ReadLine()

   137

						If InStr(line, "URL") Then

   138

							url = line

   139

						End If

   140

						If InStr(line, "Last Changed Rev") Then

   141

							revision = Mid(line, 19)

   142

							version = "r" & revision

   143

						End If

   144

					Loop While Not OExec.StdOut.atEndOfStream

   145

				End If ' Err.Number = 0

   146

			End If ' line <> "exported"

   147

		End If ' Err.Number = 0

   148

	End If ' InStr(version, "$")

   149

   150

	If version <> "norev000" Then

   151

		If InStr(url, "branches") Then

   152

			url = Mid(url, InStr(url, "branches/") + 9)

   153

			branch = Mid(url, 1, InStr(2, url, "/") - 1)

   154

		End If

   155

	Else ' version <> "norev000"

   156

		' svn detection failed, reset error and try git

   157

		Err.Clear

   158

		Set oExec = WshShell.Exec("git rev-parse --verify HEAD")

   159

		If Err.Number = 0 Then

   160

			' Wait till the application is finished ...

   161

			Do While oExec.Status = 0

   162

			Loop

   163

   164

			If oExec.ExitCode = 0 Then

   165

				hash = oExec.StdOut.ReadLine()

   166

				version = "g" & Mid(hash, 1, 8)

   167

				Set oExec = WshShell.Exec("git diff-index --exit-code --quiet HEAD ../src")

   168

				If Err.Number = 0 Then

   169

					' Wait till the application is finished ...

   170

					Do While oExec.Status = 0

   171

					Loop

   172

   173

					If oExec.ExitCode = 1 Then

   174

						modified = 2

   175

					End If ' oExec.ExitCode = 1

   176

   177

					Set oExec = WshShell.Exec("git symbolic-ref HEAD")

   178

					If Err.Number = 0 Then

   179

						line = oExec.StdOut.ReadLine()

   180

						line = Mid(line, InStrRev(line, "/") + 1)

   181

						If line <> "master" Then

   182

							branch = line

   183

						End If ' line <> "master"

   184

					End If ' Err.Number = 0

   185

   186

					Set oExec = WshShell.Exec("git log --pretty=format:%s --grep=" & Chr(34) & "^(svn r[0-9]*)" & Chr(34) & " -1 ../src")

   187

					if Err.Number = 0 Then

   188

						revision = Mid(oExec.StdOut.ReadLine(), 7)

   189

						revision = Mid(revision, 1, InStr(revision, ")") - 1)

   190

					End If ' Err.Number = 0

   191

				End If ' Err.Number = 0

   192

			End If ' oExec.ExitCode = 0

   193

		End If ' Err.Number = 0

   194

   195

		If version = "norev000" Then

   196

			' git detection failed, reset error and try mercurial (hg)

   197

			Err.Clear

   198

			Set oExec = WshShell.Exec("hg parents")

   199

			If Err.Number = 0 Then

   200

				' Wait till the application is finished ...

   201

				Do While oExec.Status = 0

   202

				Loop

   203

   204

				If oExec.ExitCode = 0 Then

   205

					line = OExec.StdOut.ReadLine()

   206

					hash = Mid(line, InStrRev(line, ":") + 1)

   207

					version = "h" & Mid(hash, 1, 8)

   208

					Set oExec = WshShell.Exec("hg status ../src")

   209

					If Err.Number = 0 Then

   210

						Do

   211

							line = OExec.StdOut.ReadLine()

   212

							If Len(line) > 0 And Mid(line, 1, 1) <> "?" Then

   213

								modified = 2

   214

								Exit Do

   215

							End If ' Len(line) > 0 And Mid(line, 1, 1) <> "?"

   216

						Loop While Not OExec.StdOut.atEndOfStream

   217

   218

						Set oExec = WshShell.Exec("hg branch")

   219

						If Err.Number = 0 Then

   220

							line = OExec.StdOut.ReadLine()

   221

							If line <> "default" Then

   222

								branch = line

   223

							End If ' line <> "default"

   224

						End If ' Err.Number = 0

   225

   226

						Set oExec = WshShell.Exec("hg log -r " & hash & ":0 -k " & Chr(34) & "svn" & Chr(34) & " -l 1 --template " & Chr(34) & "{desc}\n" & Chr(34) & " ../src")

   227

						If Err.Number = 0 Then

   228

							revision = Mid(OExec.StdOut.ReadLine(), 7)

   229

							revision = Mid(revision, 1, InStr(revision, ")") - 1)

   230

						End If ' Err.Number = 0

   231

					End If ' Err.Number = 0

   232

				End If ' oExec.ExitCode = 0

   233

			End If ' Err.Number = 0

   234

		End If ' version = "norev000"

   235

	End If ' version <> "norev000"

   236

   237

	If modified = 2 Then

   238

		version = version & "M"

   239

	End If

   240

   241

	If branch <> "" Then

   242

		version = version & "-" & branch

   243

	End If

   244

   245

	If version <> "norev000" Then

   246

		DetermineSVNVersion = version & Chr(9) & revision & Chr(9) & modified

   247

	Else

   248

		DetermineSVNVersion = version

   249

	End If

   250

End Function

   251

   252

Function IsCachedVersion(ByVal version)

   253

	Dim cache_file, cached_version

   254

	cached_version = ""

   255

	Set cache_file = FSO.OpenTextFile("../config.cache.version", 1, True, 0)

   256

	If Not cache_file.atEndOfStream Then

   257

		cached_version = cache_file.ReadLine()

   258

	End If

   259

	cache_file.Close

   260

   261

	If InStr(version, Chr(9)) Then

   262

		version = Mid(version, 1, Instr(version, Chr(9)) - 1)

   263

	End If

   264

   265

	If version <> cached_version Then

   266

		Set cache_file = fso.CreateTextFile("../config.cache.version", True)

   267

		cache_file.WriteLine(version)

   268

		cache_file.Close

   269

		IsCachedVersion = False

   270

	Else

   271

		IsCachedVersion = True

   272

	End If

   273

End Function

   274

   275

Dim version

   276

version = DetermineSVNVersion

   277

If Not (IsCachedVersion(version) And FSO.FileExists("../src/rev.cpp") And FSO.FileExists("../src/ottdres.rc")) Then

   278

	UpdateFiles version

   279

End If