#!/usr/bin/env python3

import os
import re
import subprocess
import sys

def call(cmd):
    print(" ".join(cmd))
    retcode = subprocess.call(cmd)
    if retcode != 0:
      sys.exit(retcode)

def print_stage(text):
    print("")
    print(text)
    print("")

# Setup for precompiled libraries and tests from svn
lib_dirpath = os.path.join('..', 'lib')
svn_url = "https://svn.blender.org/svnroot/bf-blender/trunk/lib/"

# Checkout precompiled libraries
if sys.platform == 'darwin':
    lib_platform = "darwin"
elif sys.platform == 'win32':
    # Windows checkout is usually handled by bat scripts since python3 to run
    # this script is bundled as part of the precompiled libraries. However it
    # is used by the buildbot.
    lib_platform = "win64_vc14"
else:
    # No precompiled libraries for Linux.
    lib_platform = None

if lib_platform:
    lib_platform_dirpath = os.path.join(lib_dirpath, lib_platform)

    if not os.path.exists(lib_platform_dirpath):
        print_stage("Checking out Precompiled Libraries")

        svn_url_platform = svn_url + lib_platform
        call(["svn", "checkout", svn_url_platform, lib_platform_dirpath])

# Update precompiled libraries and tests
print_stage("Updating Precompiled Libraries and Tests")

if os.path.isdir(lib_dirpath):
  for dirname in os.listdir(lib_dirpath):
    if dirname == ".svn":
        continue

    dirpath = os.path.join(lib_dirpath, dirname)
    svn_dirpath = os.path.join(dirpath, ".svn")
    svn_root_dirpath = os.path.join(lib_dirpath, ".svn")

    if os.path.isdir(dirpath) and \
       (os.path.exists(svn_dirpath) or os.path.exists(svn_root_dirpath)):
        call(["svn", "cleanup", dirpath])
        call(["svn", "update", dirpath])

# Update blender repository and submodules
print_stage("Updating Blender Git Repository and Submodules")

call(["git", "pull", "--rebase"])
call(["git", "submodule", "update", "--init", "--recursive"])
call(["git", "submodule", "foreach", "git", "checkout", "master"])
call(["git", "submodule", "foreach", "git", "pull", "--rebase", "origin", "master"])