#!/usr/bin/env python

# Copyright 2017 The Kubernetes Authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# This python script helps sync godeps from the k8s repos into our git submodules
# It generates bash commands where changes are needed
# We can probably also use it for deps when the time comes!

import json
import sys
import subprocess
from pprint import pprint
from os.path import expanduser, join

kops_dir = expanduser('~/k8s/src/k8s.io/kops')
k8s_dir = expanduser('~/k8s/src/k8s.io/kubernetes')

with open(join(k8s_dir, 'Godeps/Godeps.json')) as data_file:
    godeps = json.load(data_file)

#pprint(godeps)

godep_map = {}

for godep in godeps['Deps']:
  #print("%s %s" % (godep['ImportPath'], godep['Rev']))
  godep_map[godep['ImportPath']] = godep['Rev']


process = subprocess.Popen(['git', 'submodule', 'status'], stdout=subprocess.PIPE, cwd=kops_dir)
submodule_status, err = process.communicate()
for submodule_line in submodule_status.splitlines():
  tokens = submodule_line.split()
  dep = tokens[1]
  dep = dep.replace('_vendor/', '')
  sha = tokens[0]
  sha = sha.replace('+', '')
  godep_sha = godep_map.get(dep)
  if not godep_sha:
    for k in godep_map:
      if k.startswith(dep):
        godep_sha = godep_map[k]
        break
  if godep_sha:
    if godep_sha != sha:
      print("# update needed: %s vs %s" % (godep_sha, sha))
      print("pushd _vendor/{dep}; git fetch; git checkout {sha}; popd".format(dep=dep, sha=godep_sha))
  else:
    print("# UNKNOWN dep %s" % dep)