mirror of
https://github.com/apernet/hysteria.git
synced 2025-04-03 20:47:38 +03:00
feat: hyperbole mockgen subcommand
This commit is contained in:
Toby
parent
723612c297
commit
d10398a11f
1 changed files with 111 additions and 72 deletions
183
hyperbole.py
183
hyperbole.py
|
|
@ -21,74 +21,79 @@ LOGO = """
|
||||||
░▀░▀░░▀░░▀░░░▀▀▀░▀░▀░▀▀░░▀▀▀░▀▀▀░▀▀▀
|
░▀░▀░░▀░░▀░░░▀▀▀░▀░▀░▀▀░░▀▀▀░▀▀▀░▀▀▀
|
||||||
"""
|
"""
|
||||||
|
|
||||||
DESC = 'Hyperbole is the official build script for Hysteria.'
|
DESC = "Hyperbole is the official build script for Hysteria."
|
||||||
|
|
||||||
BUILD_DIR = 'build'
|
BUILD_DIR = "build"
|
||||||
|
|
||||||
APP_SRC_DIR = './app'
|
APP_SRC_DIR = "./app"
|
||||||
APP_SRC_CMD_PKG = 'github.com/apernet/hysteria/app/cmd'
|
APP_SRC_CMD_PKG = "github.com/apernet/hysteria/app/cmd"
|
||||||
|
|
||||||
|
|
||||||
def check_command(args):
|
def check_command(args):
|
||||||
try:
|
try:
|
||||||
subprocess.check_call(args,
|
subprocess.check_call(
|
||||||
stdout=subprocess.DEVNULL,
|
args, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL
|
||||||
stderr=subprocess.DEVNULL)
|
)
|
||||||
return True
|
return True
|
||||||
except Exception:
|
except Exception:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
def check_build_env():
|
def check_build_env():
|
||||||
if not check_command(['git', '--version']):
|
if not check_command(["git", "--version"]):
|
||||||
print('Git is not installed. Please install Git and try again.')
|
print("Git is not installed. Please install Git and try again.")
|
||||||
return False
|
return False
|
||||||
if not check_command(['git', 'rev-parse', '--is-inside-work-tree']):
|
if not check_command(["git", "rev-parse", "--is-inside-work-tree"]):
|
||||||
print('Not in a Git repository. Please go to the project root and try again.')
|
print("Not in a Git repository. Please go to the project root and try again.")
|
||||||
return False
|
return False
|
||||||
if not check_command(['go', 'version']):
|
if not check_command(["go", "version"]):
|
||||||
print('Go is not installed. Please install Go and try again.')
|
print("Go is not installed. Please install Go and try again.")
|
||||||
return False
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
def get_app_version():
|
def get_app_version():
|
||||||
app_version = os.environ.get('HY_APP_VERSION')
|
app_version = os.environ.get("HY_APP_VERSION")
|
||||||
if not app_version:
|
if not app_version:
|
||||||
try:
|
try:
|
||||||
output = subprocess.check_output(
|
output = (
|
||||||
['git', 'describe', '--tags', '--always', '--match', 'app/v*']).decode().strip()
|
subprocess.check_output(
|
||||||
app_version = output.split('/')[-1]
|
["git", "describe", "--tags", "--always", "--match", "app/v*"]
|
||||||
|
)
|
||||||
|
.decode()
|
||||||
|
.strip()
|
||||||
|
)
|
||||||
|
app_version = output.split("/")[-1]
|
||||||
except Exception:
|
except Exception:
|
||||||
app_version = 'Unknown'
|
app_version = "Unknown"
|
||||||
return app_version
|
return app_version
|
||||||
|
|
||||||
|
|
||||||
def get_app_commit():
|
def get_app_commit():
|
||||||
app_commit = os.environ.get('HY_APP_COMMIT')
|
app_commit = os.environ.get("HY_APP_COMMIT")
|
||||||
if not app_commit:
|
if not app_commit:
|
||||||
try:
|
try:
|
||||||
app_commit = subprocess.check_output(
|
app_commit = (
|
||||||
['git', 'rev-parse', 'HEAD']).decode().strip()
|
subprocess.check_output(["git", "rev-parse", "HEAD"]).decode().strip()
|
||||||
|
)
|
||||||
except Exception:
|
except Exception:
|
||||||
app_commit = 'Unknown'
|
app_commit = "Unknown"
|
||||||
return app_commit
|
return app_commit
|
||||||
|
|
||||||
|
|
||||||
def get_app_platforms():
|
def get_app_platforms():
|
||||||
platforms = os.environ.get('HY_APP_PLATFORMS')
|
platforms = os.environ.get("HY_APP_PLATFORMS")
|
||||||
if not platforms:
|
if not platforms:
|
||||||
d_os = subprocess.check_output(['go', 'env', 'GOOS']).decode().strip()
|
d_os = subprocess.check_output(["go", "env", "GOOS"]).decode().strip()
|
||||||
d_arch = subprocess.check_output(
|
d_arch = subprocess.check_output(["go", "env", "GOARCH"]).decode().strip()
|
||||||
['go', 'env', 'GOARCH']).decode().strip()
|
|
||||||
return [(d_os, d_arch)]
|
return [(d_os, d_arch)]
|
||||||
|
|
||||||
result = []
|
result = []
|
||||||
for platform in platforms.split(','):
|
for platform in platforms.split(","):
|
||||||
platform = platform.strip()
|
platform = platform.strip()
|
||||||
if not platform:
|
if not platform:
|
||||||
continue
|
continue
|
||||||
parts = platform.split('/')
|
parts = platform.split("/")
|
||||||
if len(parts) != 2:
|
if len(parts) != 2:
|
||||||
continue
|
continue
|
||||||
result.append((parts[0], parts[1]))
|
result.append((parts[0], parts[1]))
|
||||||
|
|
@ -102,44 +107,53 @@ def cmd_build(release=False):
|
||||||
os.makedirs(BUILD_DIR, exist_ok=True)
|
os.makedirs(BUILD_DIR, exist_ok=True)
|
||||||
|
|
||||||
app_version = get_app_version()
|
app_version = get_app_version()
|
||||||
app_date = datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%SZ')
|
app_date = datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
|
||||||
app_commit = get_app_commit()
|
app_commit = get_app_commit()
|
||||||
|
|
||||||
ldflags = [
|
ldflags = [
|
||||||
'-X', APP_SRC_CMD_PKG + '.appVersion=' + app_version,
|
"-X",
|
||||||
'-X', APP_SRC_CMD_PKG + '.appDate=' + app_date,
|
APP_SRC_CMD_PKG + ".appVersion=" + app_version,
|
||||||
'-X', APP_SRC_CMD_PKG + '.appType=' +
|
"-X",
|
||||||
('release' if release else 'dev'),
|
APP_SRC_CMD_PKG + ".appDate=" + app_date,
|
||||||
'-X', APP_SRC_CMD_PKG + '.appCommit=' + app_commit,
|
"-X",
|
||||||
|
APP_SRC_CMD_PKG + ".appType=" + ("release" if release else "dev"),
|
||||||
|
"-X",
|
||||||
|
APP_SRC_CMD_PKG + ".appCommit=" + app_commit,
|
||||||
]
|
]
|
||||||
if release:
|
if release:
|
||||||
ldflags.append('-s')
|
ldflags.append("-s")
|
||||||
ldflags.append('-w')
|
ldflags.append("-w")
|
||||||
|
|
||||||
for os_name, arch in get_app_platforms():
|
for os_name, arch in get_app_platforms():
|
||||||
print('Building for %s/%s...' % (os_name, arch))
|
print("Building for %s/%s..." % (os_name, arch))
|
||||||
|
|
||||||
out_name = 'hysteria-%s-%s' % (os_name, arch)
|
out_name = "hysteria-%s-%s" % (os_name, arch)
|
||||||
if os_name == 'windows':
|
if os_name == "windows":
|
||||||
out_name += '.exe'
|
out_name += ".exe"
|
||||||
|
|
||||||
env = os.environ.copy()
|
env = os.environ.copy()
|
||||||
env['GOOS'] = os_name
|
env["GOOS"] = os_name
|
||||||
env['GOARCH'] = arch
|
env["GOARCH"] = arch
|
||||||
|
|
||||||
cmd = ['go', 'build', '-o',
|
cmd = [
|
||||||
os.path.join(BUILD_DIR, out_name), '-ldflags', ' '.join(ldflags)]
|
"go",
|
||||||
|
"build",
|
||||||
|
"-o",
|
||||||
|
os.path.join(BUILD_DIR, out_name),
|
||||||
|
"-ldflags",
|
||||||
|
" ".join(ldflags),
|
||||||
|
]
|
||||||
if release:
|
if release:
|
||||||
cmd.append('-trimpath')
|
cmd.append("-trimpath")
|
||||||
cmd.append(APP_SRC_DIR)
|
cmd.append(APP_SRC_DIR)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
subprocess.check_call(cmd, env=env)
|
subprocess.check_call(cmd, env=env)
|
||||||
except Exception:
|
except Exception:
|
||||||
print('Failed to build for %s/%s' % (os_name, arch))
|
print("Failed to build for %s/%s" % (os_name, arch))
|
||||||
return
|
return
|
||||||
|
|
||||||
print('Built %s' % out_name)
|
print("Built %s" % out_name)
|
||||||
|
|
||||||
|
|
||||||
def cmd_run(args):
|
def cmd_run(args):
|
||||||
|
|
@ -147,17 +161,21 @@ def cmd_run(args):
|
||||||
return
|
return
|
||||||
|
|
||||||
app_version = get_app_version()
|
app_version = get_app_version()
|
||||||
app_date = datetime.datetime.utcnow().strftime('%Y-%m-%dT%H:%M:%SZ')
|
app_date = datetime.datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%SZ")
|
||||||
app_commit = get_app_commit()
|
app_commit = get_app_commit()
|
||||||
|
|
||||||
ldflags = [
|
ldflags = [
|
||||||
'-X', APP_SRC_CMD_PKG + '.appVersion=' + app_version,
|
"-X",
|
||||||
'-X', APP_SRC_CMD_PKG + '.appDate=' + app_date,
|
APP_SRC_CMD_PKG + ".appVersion=" + app_version,
|
||||||
'-X', APP_SRC_CMD_PKG + '.appType=dev-run',
|
"-X",
|
||||||
'-X', APP_SRC_CMD_PKG + '.appCommit=' + app_commit,
|
APP_SRC_CMD_PKG + ".appDate=" + app_date,
|
||||||
|
"-X",
|
||||||
|
APP_SRC_CMD_PKG + ".appType=dev-run",
|
||||||
|
"-X",
|
||||||
|
APP_SRC_CMD_PKG + ".appCommit=" + app_commit,
|
||||||
]
|
]
|
||||||
|
|
||||||
cmd = ['go', 'run', '-ldflags', ' '.join(ldflags)]
|
cmd = ["go", "run", "-ldflags", " ".join(ldflags)]
|
||||||
cmd.append(APP_SRC_DIR)
|
cmd.append(APP_SRC_DIR)
|
||||||
cmd.extend(args)
|
cmd.extend(args)
|
||||||
|
|
||||||
|
|
@ -171,14 +189,29 @@ def cmd_run(args):
|
||||||
|
|
||||||
|
|
||||||
def cmd_format():
|
def cmd_format():
|
||||||
if not check_command(['gofumpt', '-version']):
|
if not check_command(["gofumpt", "-version"]):
|
||||||
print('gofumpt is not installed. Please install gofumpt and try again.')
|
print("gofumpt is not installed. Please install gofumpt and try again.")
|
||||||
return
|
return
|
||||||
|
|
||||||
try:
|
try:
|
||||||
subprocess.check_call(['gofumpt', '-w', '-l', '-extra', '.'])
|
subprocess.check_call(["gofumpt", "-w", "-l", "-extra", "."])
|
||||||
except Exception:
|
except Exception:
|
||||||
print('Failed to format code')
|
print("Failed to format code")
|
||||||
|
|
||||||
|
|
||||||
|
def cmd_mockgen():
|
||||||
|
if not check_command(["mockery", "--version"]):
|
||||||
|
print("mockery is not installed. Please install mockery and try again.")
|
||||||
|
return
|
||||||
|
|
||||||
|
for dirpath, dirnames, filenames in os.walk("."):
|
||||||
|
dirnames[:] = [d for d in dirnames if not d.startswith(".")]
|
||||||
|
if ".mockery.yaml" in filenames:
|
||||||
|
print("Generating mocks for %s..." % dirpath)
|
||||||
|
try:
|
||||||
|
subprocess.check_call(["mockery"], cwd=dirpath)
|
||||||
|
except Exception:
|
||||||
|
print("Failed to generate mocks for %s" % dirpath)
|
||||||
|
|
||||||
|
|
||||||
def cmd_clean():
|
def cmd_clean():
|
||||||
|
|
@ -193,40 +226,46 @@ def cmd_about():
|
||||||
def main():
|
def main():
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
p_cmd = parser.add_subparsers(dest='command')
|
p_cmd = parser.add_subparsers(dest="command")
|
||||||
p_cmd.required = True
|
p_cmd.required = True
|
||||||
|
|
||||||
# Run
|
# Run
|
||||||
p_run = p_cmd.add_parser('run', help='Run the app')
|
p_run = p_cmd.add_parser("run", help="Run the app")
|
||||||
p_run.add_argument('args', nargs=argparse.REMAINDER)
|
p_run.add_argument("args", nargs=argparse.REMAINDER)
|
||||||
|
|
||||||
# Build
|
# Build
|
||||||
p_build = p_cmd.add_parser('build', help='Build the app')
|
p_build = p_cmd.add_parser("build", help="Build the app")
|
||||||
p_build.add_argument('-r', '--release', action='store_true',
|
p_build.add_argument(
|
||||||
help='Build a release version')
|
"-r", "--release", action="store_true", help="Build a release version"
|
||||||
|
)
|
||||||
|
|
||||||
# Format
|
# Format
|
||||||
p_cmd.add_parser('format', help='Format the code')
|
p_cmd.add_parser("format", help="Format the code")
|
||||||
|
|
||||||
|
# Mockgen
|
||||||
|
p_cmd.add_parser("mockgen", help="Generate mock interfaces")
|
||||||
|
|
||||||
# Clean
|
# Clean
|
||||||
p_cmd.add_parser('clean', help='Clean the build directory')
|
p_cmd.add_parser("clean", help="Clean the build directory")
|
||||||
|
|
||||||
# About
|
# About
|
||||||
p_cmd.add_parser('about', help='Print about information')
|
p_cmd.add_parser("about", help="Print about information")
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
if args.command == 'run':
|
if args.command == "run":
|
||||||
cmd_run(args.args)
|
cmd_run(args.args)
|
||||||
elif args.command == 'build':
|
elif args.command == "build":
|
||||||
cmd_build(args.release)
|
cmd_build(args.release)
|
||||||
elif args.command == 'format':
|
elif args.command == "format":
|
||||||
cmd_format()
|
cmd_format()
|
||||||
elif args.command == 'clean':
|
elif args.command == "mockgen":
|
||||||
|
cmd_mockgen()
|
||||||
|
elif args.command == "clean":
|
||||||
cmd_clean()
|
cmd_clean()
|
||||||
elif args.command == 'about':
|
elif args.command == "about":
|
||||||
cmd_about()
|
cmd_about()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue