#!/usr/bin/env python3
# Copyright 2017 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Small utility script to enable/disable `depot_tools` automatic updating."""

import argparse
import datetime
import os
import sys

DEPOT_TOOLS_ROOT = os.path.abspath(os.path.dirname(__file__))
SENTINEL_PATH = os.path.join(DEPOT_TOOLS_ROOT, ".disable_auto_update")


def main():
    parser = argparse.ArgumentParser()
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument(
        "--enable", action="store_true", help="Enable auto-updating."
    )
    group.add_argument(
        "--disable", action="store_true", help="Disable auto-updating."
    )
    args = parser.parse_args()

    if args.enable:
        if os.path.exists(SENTINEL_PATH):
            os.unlink(SENTINEL_PATH)
    if args.disable:
        if not os.path.exists(SENTINEL_PATH):
            with open(SENTINEL_PATH, "w") as fd:
                fd.write(
                    "Disabled by %s at %s\n"
                    % (__file__, datetime.datetime.now())
                )
    return 0


if __name__ == "__main__":
    sys.exit(main())