create.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Copyright 2020 Google LLC.
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
 
 
"""Create the asset."""
 
 
import argparse
import subprocess
import os
import shutil
 
 
REPO_URL = "https://github.com/vektra/mockery.git"
REPO_DIR = "mockery_repo"
BINARY_NAME = "mockery"
 
 
def create_asset(target_dir):
  """Create the asset."""
  os.chdir(target_dir)
 
  # We build mockery 2.4.0 from source to fix an issue with Go 1.18. Read the
  # comments below for details.
  output = subprocess.check_output(["git", "clone", REPO_URL, REPO_DIR])
  print(output)
  os.chdir(os.path.join(target_dir, REPO_DIR))
  output = subprocess.check_output(["git", "checkout", "v2.4.0"])
 
  # Under Go 1.18, mockery v2.4.0 through v2.10.0 fails with errors such as:
  #
  #     internal error: package "fmt" without types was imported from ...
  #
  # This can be fixed by updating golang.org/x/tools to a more recent version.
  # For more details, please see https://github.com/vektra/mockery/issues/434
  # and https://github.com/golang/go/issues/49608.
  output = subprocess.check_output(["go", "get", "golang.org/x/tools@v0.1.10"])
  print(output)
  output = subprocess.check_output(["go", "mod", "tidy"])
  print(output)
 
  # Build mockery with the same flags as in release builds.
  #
  # If we don't specify a SemVer value, mockery will generate files with a
  # "Code generated by mockery v0.0.0-dev. DO NOT EDIT." comment at the top,
  # which causes diffs due to the changed version. Thus, it is important to set
  # the SemVer value to the correct version.
  #
  # See
  # https://github.com/vektra/mockery/blob/271c74610ef710a4c30e19a42733796c50e7ea3f/.goreleaser.yml#L9.
  ldflags = "-s -w -X github.com/vektra/mockery/v2/pkg/config.SemVer=2.4.0"
  build_command = ["go", "build", "-ldflags=\"%s\"" % ldflags]
  print("Building with command:", build_command)
  output = subprocess.check_output(["go", "build", "-ldflags=" + ldflags])
  print(output)
 
  # Copy binary outside of the cloned repository directory and clean up.
  output = subprocess.check_output(["cp", BINARY_NAME, ".."])
  shutil.copy(os.path.join(target_dir, REPO_DIR, BINARY_NAME), target_dir)
  shutil.rmtree(os.path.join(target_dir, REPO_DIR))
  os.chdir(target_dir)
 
 
def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('--target_dir', '-t', required=True)
  args = parser.parse_args()
  create_asset(args.target_dir)
 
 
if __name__ == '__main__':
  main()