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(revision, version, cur_date, filename)

    18

	FSO.CopyFile filename & ".in", filename

    19

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

    20

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

    21

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

    22

End Sub

    23

    24

Sub UpdateFiles(version)

    25

	Dim WshShell, cur_date, revision, oExec

    26

	Set WshShell = CreateObject("WScript.Shell")

    27

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

    28

	revision = 0

    29

	Select Case Mid(version, 1, 1)

    30

		Case "r" ' svn

    31

			revision = Mid(version, 2)

    32

			If InStr(revision, "M") Then

    33

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

    34

			End If

    35

			If InStr(revision, "-") Then

    36

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

    37

			End If

    38

		Case "h" ' mercurial (hg)

    39

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

    40

			If Err.Number = 0 Then

    41

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

    42

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

    43

			End If

    44

		Case "g" ' git

    45

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

    46

			if Err.Number = 0 Then

    47

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

    48

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

    49

			End If

    50

	End Select

    51

    52

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

    53

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

    54

End Sub

    55

    56

Function ReadRegistryKey(shive, subkey, valuename, architecture)

    57

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

    58

    59

	' First, get the Registry Provider for the requested architecture

    60

	Set objCtx = CreateObject("WbemScripting.SWbemNamedValueSet")

    61

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

    62

	Set objLocator = CreateObject("Wbemscripting.SWbemLocator")

    63

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

    64

	Set objReg = objServices.Get("StdRegProv")

    65

    66

	' Check the hive and give it the right value

    67

	Select Case shive

    68

		Case "HKCR", "HKEY_CLASSES_ROOT"

    69

			hiveKey = &h80000000

    70

		Case "HKCU", "HKEY_CURRENT_USER"

    71

			hiveKey = &H80000001

    72

		Case "HKLM", "HKEY_LOCAL_MACHINE"

    73

			hiveKey = &h80000002

    74

		Case "HKU", "HKEY_USERS"

    75

			hiveKey = &h80000003

    76

		Case "HKCC", "HKEY_CURRENT_CONFIG"

    77

			hiveKey = &h80000005

    78

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

    79

			hiveKey = &h80000006

    80

		Case Else

    81

			MsgBox "Hive not valid (ReadRegistryKey)"

    82

	End Select

    83

    84

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

    85

	Inparams.Hdefkey = hiveKey

    86

	Inparams.Ssubkeyname = subkey

    87

	Inparams.Svaluename = valuename

    88

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

    89

    90

	ReadRegistryKey = Outparams.SValue

    91

End Function

    92

    93

Function DetermineSVNVersion()

    94

	Dim WshShell, version, url, oExec, line

    95

	Set WshShell = CreateObject("WScript.Shell")

    96

	On Error Resume Next

    97

    98

	' Try TortoiseSVN

    99

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

   100

	Dim sTortoise

   101

	' First, try with 32-bit architecture

   102

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

   103

	If sTortoise = Nothing Then

   104

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

   105

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

   106

	End If

   107

   108

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

   109

	If sTortoise <> Nothing Then

   110

		Dim file

   111

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

   112

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

   113

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

   114

		file.WriteLine "$WCURL$"

   115

		file.Close

   116

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

   117

		' Wait till the application is finished ...

   118

		Do

   119

			OExec.StdOut.ReadLine()

   120

		Loop While Not OExec.StdOut.atEndOfStream

   121

   122

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

   123

		version = file.ReadLine

   124

		url = file.ReadLine

   125

		file.Close

   126

   127

		Set file = FSO.GetFile("tsvn_tmp")

   128

		file.Delete

   129

	End If

   130

   131

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

   132

	If InStr(version, "$") Then

   133

		' Reset error and version

   134

		Err.Clear

   135

		version = "norev000"

   136

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

   137

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

   138

		If Err.Number = 0 Then

   139

			Dim modified

   140

			If InStr(OExec.StdOut.ReadLine(), "M") Then

   141

				modified = "M"

   142

			Else

   143

				modified = ""

   144

			End If

   145

   146

			' Set the environment to english

   147

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

   148

   149

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

   150

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

   151

			If Err.Number = 0 Then

   152

				Do

   153

					line = OExec.StdOut.ReadLine()

   154

					If InStr(line, "URL") Then

   155

						url = line

   156

					End If

   157

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

   158

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

   159

					End If

   160

				Loop While Not OExec.StdOut.atEndOfStream

   161

			End If

   162

		End If

   163

	End If

   164

   165

	If version <> "norev000" Then

   166

		If InStr(url, "branches") Then

   167

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

   168

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

   169

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

   170

		End If

   171

	Else

   172

		' svn detection failed, reset error and try git

   173

		Err.Clear

   174

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

   175

		If Err.Number = 0 Then

   176

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

   177

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

   178

			Do While oExec.Status = 0 And Err.Number = 0

   179

			Loop

   180

			If Err.Number = 0 And oExec.ExitCode = 1 Then

   181

				version = version & "M"

   182

			End If

   183

   184

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

   185

			If Err.Number = 0 Then

   186

				line = oExec.StdOut.ReadLine()

   187

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

   188

				If line <> "master" Then

   189

					version = version & "-" & line

   190

				End If

   191

			End If

   192

		Else

   193

			' try mercurial (hg)

   194

			Err.Clear

   195

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

   196

			If Err.Number = 0 Then

   197

				version = "h" & Mid(OExec.StdOut.ReadLine(), 19, 8)

   198

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

   199

				If Err.Number = 0 Then

   200

					Do

   201

						line = OExec.StdOut.ReadLine()

   202

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

   203

							version = version & "M"

   204

							Exit Do

   205

						End If

   206

					Loop While Not OExec.StdOut.atEndOfStream

   207

				End If

   208

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

   209

				If Err.Number = 0 Then

   210

						line = OExec.StdOut.ReadLine()

   211

						If line <> "default" Then

   212

							version = version & "-" & line

   213

						End If

   214

				End If

   215

			End If

   216

		End If

   217

	End If

   218

   219

	DetermineSVNVersion = version

   220

End Function

   221

   222

Function IsCachedVersion(version)

   223

	Dim cache_file, cached_version

   224

	cached_version = ""

   225

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

   226

	If Not cache_file.atEndOfStream Then

   227

		cached_version = cache_file.ReadLine()

   228

	End If

   229

	cache_file.Close

   230

   231

	If version <> cached_version Then

   232

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

   233

		cache_file.WriteLine(version)

   234

		cache_file.Close

   235

		IsCachedVersion = False

   236

	Else

   237

		IsCachedVersion = True

   238

	End If

   239

End Function

   240

   241

Dim version

   242

version = DetermineSVNVersion

   243

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

   244

	UpdateFiles version

   245

End If