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

    85

	' Try TortoiseSVN

    86

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

    87

	Dim sTortoise

    88

	' First, try with 32-bit architecture

    89

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

    90

	If sTortoise = Nothing Then

    91

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

    92

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

    93

	End If

    94

    95

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

    96

	If sTortoise <> Nothing Then

    97

		Dim file

    98

		' Write some "magic" to a temporary file so we can acquire the svn revision/state

    99

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

   100

		file.WriteLine "r$WCREV$"

   101

		file.WriteLine "$WCURL$"

   102

		file.WriteLine "$WCMODS?2:0$"

   103

		file.WriteLine "$WCREV$"

   104

		file.Close

   105

		Set oExec = WshShell.Exec(sTortoise & "\bin\SubWCRev.exe ../src tsvn_tmp tsvn_tmp")

   106

		' Wait till the application is finished ...

   107

		Do

   108

			OExec.StdOut.ReadLine()

   109

		Loop While Not OExec.StdOut.atEndOfStream

   110

   111

		Set file = FSO.OpenTextFile("tsvn_tmp", 1, 0, 0)

   112

		version = file.ReadLine

   113

		url = file.ReadLine

   114

		modified = file.ReadLine

   115

		revision = file.ReadLine

   116

		file.Close

   117

   118

		Set file = FSO.GetFile("tsvn_tmp")

   119

		file.Delete

   120

	End If

   121

   122

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

   123

	If InStr(version, "$") Then

   124

		' Reset error and version

   125

		Err.Clear

   126

		version = "norev000"

   127

		modified = 0

   128

   129

		' Set the environment to english

   130

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

   131

   132

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

   133

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

   134

		If Err.Number = 0 Then

   135

			' Wait till the application is finished ...

   136

			Do While oExec.Status = 0

   137

			Loop

   138

   139

			line = OExec.StdOut.ReadLine()

   140

			If line <> "exported" Then

   141

				If InStr(line, "M") Then

   142

					modified = 2

   143

				End If

   144

   145

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

   146

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

   147

				If Err.Number = 0 Then

   148

					Do

   149

						line = OExec.StdOut.ReadLine()

   150

						If InStr(line, "URL") Then

   151

							url = line

   152

						End If

   153

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

   154

							revision = Mid(line, 19)

   155

							version = "r" & revision

   156

						End If

   157

					Loop While Not OExec.StdOut.atEndOfStream

   158

				End If ' Err.Number = 0

   159

			End If ' line <> "exported"

   160

		End If ' Err.Number = 0

   161

	End If ' InStr(version, "$")

   162

   163

	If version <> "norev000" Then

   164

		If InStr(url, "branches") Then

   165

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

   166

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

   167

		End If

   168

	Else ' version <> "norev000"

   169

		' svn detection failed, reset error and try git

   170

		Err.Clear

   171

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

   172

		If Err.Number = 0 Then

   173

			' Wait till the application is finished ...

   174

			Do While oExec.Status = 0

   175

			Loop

   176

   177

			If oExec.ExitCode = 0 Then

   178

				hash = oExec.StdOut.ReadLine()

   179

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

   180

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

   181

				If Err.Number = 0 Then

   182

					' Wait till the application is finished ...

   183

					Do While oExec.Status = 0

   184

					Loop

   185

   186

					If oExec.ExitCode = 1 Then

   187

						modified = 2

   188

					End If ' oExec.ExitCode = 1

   189

   190

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

   191

					If Err.Number = 0 Then

   192

						line = oExec.StdOut.ReadLine()

   193

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

   194

						If line <> "master" Then

   195

							branch = line

   196

						End If ' line <> "master"

   197

					End If ' Err.Number = 0

   198

   199

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

   200

					if Err.Number = 0 Then

   201

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

   202

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

   203

					End If ' Err.Number = 0

   204

				End If ' Err.Number = 0

   205

			End If ' oExec.ExitCode = 0

   206

		End If ' Err.Number = 0

   207

   208

		If version = "norev000" Then

   209

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

   210

			Err.Clear

   211

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

   212

			If Err.Number = 0 Then

   213

				' Wait till the application is finished ...

   214

				Do While oExec.Status = 0

   215

				Loop

   216

   217

				If oExec.ExitCode = 0 Then

   218

					line = OExec.StdOut.ReadLine()

   219

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

   220

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

   221

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

   222

					If Err.Number = 0 Then

   223

						Do

   224

							line = OExec.StdOut.ReadLine()

   225

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

   226

								modified = 2

   227

								Exit Do

   228

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

   229

						Loop While Not OExec.StdOut.atEndOfStream

   230

   231

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

   232

						If Err.Number = 0 Then

   233

							line = OExec.StdOut.ReadLine()

   234

							If line <> "default" Then

   235

								branch = line

   236

							End If ' line <> "default"

   237

						End If ' Err.Number = 0

   238

   239

						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")

   240

						If Err.Number = 0 Then

   241

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

   242

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

   243

						End If ' Err.Number = 0

   244

					End If ' Err.Number = 0

   245

				End If ' oExec.ExitCode = 0

   246

			End If ' Err.Number = 0

   247

		End If ' version = "norev000"

   248

	End If ' version <> "norev000"

   249

   250

	If modified = 2 Then

   251

		version = version & "M"

   252

	End If

   253

   254

	If branch <> "" Then

   255

		version = version & "-" & branch

   256

	End If

   257

   258

	If version <> "norev000" Then

   259

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

   260

	Else

   261

		DetermineSVNVersion = version

   262

	End If

   263

End Function

   264

   265

Function IsCachedVersion(ByVal version)

   266

	Dim cache_file, cached_version

   267

	cached_version = ""

   268

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

   269

	If Not cache_file.atEndOfStream Then

   270

		cached_version = cache_file.ReadLine()

   271

	End If

   272

	cache_file.Close

   273

   274

	If InStr(version, Chr(9)) Then

   275

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

   276

	End If

   277

   278

	If version <> cached_version Then

   279

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

   280

		cache_file.WriteLine(version)

   281

		cache_file.Close

   282

		IsCachedVersion = False

   283

	Else

   284

		IsCachedVersion = True

   285

	End If

   286

End Function

   287

   288

Dim version

   289

version = DetermineSVNVersion

   290

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

   291

	UpdateFiles version

   292

End If