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 WshShell, cur_date, modified, revision, oExec

    27

	Set WshShell = CreateObject("WScript.Shell")

    28

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

    29

	revision = 0

    30

	modified = 1

    31

	Select Case Mid(version, 1, 1)

    32

		Case "r" ' svn

    33

			revision = Mid(version, 2)

    34

			If InStr(revision, "M") Then

    35

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

    36

				modified = 2

    37

			Else

    38

				modified = 0

    39

			End If

    40

			If InStr(revision, "-") Then

    41

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

    42

			End If

    43

		Case "h" ' mercurial (hg)

    44

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

    45

			If Err.Number = 0 Then

    46

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

    47

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

    48

			End If

    49

		Case "g" ' git

    50

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

    51

			if Err.Number = 0 Then

    52

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

    53

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

    54

			End If

    55

	End Select

    56

    57

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

    58

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

    59

End Sub

    60

    61

Function ReadRegistryKey(shive, subkey, valuename, architecture)

    62

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

    63

    64

	' First, get the Registry Provider for the requested architecture

    65

	Set objCtx = CreateObject("WbemScripting.SWbemNamedValueSet")

    66

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

    67

	Set objLocator = CreateObject("Wbemscripting.SWbemLocator")

    68

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

    69

	Set objReg = objServices.Get("StdRegProv")

    70

    71

	' Check the hive and give it the right value

    72

	Select Case shive

    73

		Case "HKCR", "HKEY_CLASSES_ROOT"

    74

			hiveKey = &h80000000

    75

		Case "HKCU", "HKEY_CURRENT_USER"

    76

			hiveKey = &H80000001

    77

		Case "HKLM", "HKEY_LOCAL_MACHINE"

    78

			hiveKey = &h80000002

    79

		Case "HKU", "HKEY_USERS"

    80

			hiveKey = &h80000003

    81

		Case "HKCC", "HKEY_CURRENT_CONFIG"

    82

			hiveKey = &h80000005

    83

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

    84

			hiveKey = &h80000006

    85

		Case Else

    86

			MsgBox "Hive not valid (ReadRegistryKey)"

    87

	End Select

    88

    89

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

    90

	Inparams.Hdefkey = hiveKey

    91

	Inparams.Ssubkeyname = subkey

    92

	Inparams.Svaluename = valuename

    93

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

    94

    95

	ReadRegistryKey = Outparams.SValue

    96

End Function

    97

    98

Function DetermineSVNVersion()

    99

	Dim WshShell, version, url, oExec, line

   100

	Set WshShell = CreateObject("WScript.Shell")

   101

	On Error Resume Next

   102

   103

	' Try TortoiseSVN

   104

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

   105

	Dim sTortoise

   106

	' First, try with 32-bit architecture

   107

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

   108

	If sTortoise = Nothing Then

   109

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

   110

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

   111

	End If

   112

   113

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

   114

	If sTortoise <> Nothing Then

   115

		Dim file

   116

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

   117

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

   118

		file.WriteLine "r$WCREV$$WCMODS?M:$"

   119

		file.WriteLine "$WCURL$"

   120

		file.Close

   121

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

   122

		' Wait till the application is finished ...

   123

		Do

   124

			OExec.StdOut.ReadLine()

   125

		Loop While Not OExec.StdOut.atEndOfStream

   126

   127

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

   128

		version = file.ReadLine

   129

		url = file.ReadLine

   130

		file.Close

   131

   132

		Set file = FSO.GetFile("tsvn_tmp")

   133

		file.Delete

   134

	End If

   135

   136

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

   137

	If InStr(version, "$") Then

   138

		' Reset error and version

   139

		Err.Clear

   140

		version = "norev000"

   141

   142

		' Set the environment to english

   143

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

   144

   145

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

   146

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

   147

		If Err.Number = 0 Then

   148

			' Wait till the application is finished ...

   149

			Do While oExec.Status = 0

   150

			Loop

   151

   152

			line = OExec.StdOut.ReadLine()

   153

			If line <> "exported" Then

   154

				Dim modified

   155

				If InStr(line, "M") Then

   156

					modified = "M"

   157

				Else

   158

					modified = ""

   159

				End If

   160

   161

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

   162

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

   163

				If Err.Number = 0 Then

   164

					Do

   165

						line = OExec.StdOut.ReadLine()

   166

						If InStr(line, "URL") Then

   167

							url = line

   168

						End If

   169

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

   170

							version = "r" & Mid(line, 19) & modified

   171

						End If

   172

					Loop While Not OExec.StdOut.atEndOfStream

   173

				End If ' Err.Number = 0

   174

			End If ' line <> "exported"

   175

		End If ' Err.Number = 0

   176

	End If ' InStr(version, "$")

   177

   178

	If version <> "norev000" Then

   179

		If InStr(url, "branches") Then

   180

			url = Mid(url, InStr(url, "branches") + 8)

   181

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

   182

			version = version & Replace(url, "/", "-")

   183

		End If

   184

	Else ' version <> "norev000"

   185

		' svn detection failed, reset error and try git

   186

		Err.Clear

   187

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

   188

		If Err.Number = 0 Then

   189

			' Wait till the application is finished ...

   190

			Do While oExec.Status = 0

   191

			Loop

   192

   193

			If oExec.ExitCode = 0 Then

   194

				version = "g" & oExec.StdOut.ReadLine()

   195

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

   196

				If Err.Number = 0 Then

   197

					' Wait till the application is finished ...

   198

					Do While oExec.Status = 0

   199

					Loop

   200

   201

					If oExec.ExitCode = 1 Then

   202

						version = version & "M"

   203

					End If ' oExec.ExitCode = 1

   204

   205

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

   206

					If Err.Number = 0 Then

   207

						line = oExec.StdOut.ReadLine()

   208

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

   209

						If line <> "master" Then

   210

							version = version & "-" & line

   211

						End If ' line <> "master"

   212

					End If ' Err.Number = 0

   213

				End If ' Err.Number = 0

   214

			End If ' oExec.ExitCode = 0

   215

		End If ' Err.Number = 0

   216

   217

		If version = "norev000" Then

   218

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

   219

			Err.Clear

   220

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

   221

			If Err.Number = 0 Then

   222

				' Wait till the application is finished ...

   223

				Do While oExec.Status = 0

   224

				Loop

   225

   226

				If oExec.ExitCode = 0 Then

   227

					line = OExec.StdOut.ReadLine()

   228

					version = "h" & Mid(line, InStrRev(line, ":") + 1, 8)

   229

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

   230

					If Err.Number = 0 Then

   231

						Do

   232

							line = OExec.StdOut.ReadLine()

   233

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

   234

								version = version & "M"

   235

								Exit Do

   236

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

   237

						Loop While Not OExec.StdOut.atEndOfStream

   238

   239

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

   240

						If Err.Number = 0 Then

   241

							line = OExec.StdOut.ReadLine()

   242

							If line <> "default" Then

   243

								version = version & "-" & line

   244

							End If ' line <> "default"

   245

						End If ' Err.Number = 0

   246

					End If ' Err.Number = 0

   247

				End If ' oExec.ExitCode = 0

   248

			End If ' Err.Number = 0

   249

		End If ' version = "norev000"

   250

	End If ' version <> "norev000"

   251

   252

	DetermineSVNVersion = version

   253

End Function

   254

   255

Function IsCachedVersion(version)

   256

	Dim cache_file, cached_version

   257

	cached_version = ""

   258

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

   259

	If Not cache_file.atEndOfStream Then

   260

		cached_version = cache_file.ReadLine()

   261

	End If

   262

	cache_file.Close

   263

   264

	If version <> cached_version Then

   265

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

   266

		cache_file.WriteLine(version)

   267

		cache_file.Close

   268

		IsCachedVersion = False

   269

	Else

   270

		IsCachedVersion = True

   271

	End If

   272

End Function

   273

   274

Dim version

   275

version = DetermineSVNVersion

   276

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

   277

	UpdateFiles version

   278

End If