#!/usr/bin/env python
# Copyright (c) 2012 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Helper script to copy policy files into the correct location the machine."""

import os
import shutil
import sys


def main():
  assert os.geteuid() == 0, 'Need superuser privileges'
  if sys.argv[1] == 'copy':
    assert os.path.isdir(sys.argv[3])
    shutil.copy(sys.argv[2], sys.argv[3])
    dirList = os.listdir(sys.argv[3])
    for fname in dirList:
      filename = os.path.join(sys.argv[3], fname)
      os.chmod(filename, 0755)
  elif sys.argv[1] == 'setup_dir':
    os.makedirs(sys.argv[2])
  elif sys.argv[1] == 'perm_dir':
    os.system('chmod -R 755 "%s/../.."' % sys.argv[2])
  elif sys.argv[1] == 'remove_dir':
    os.system('rm -rf "%s"' % sys.argv[2])
  else:
    print >>sys.stderr, (
        'Invalid syntax. Possible values are [copy], [setup_dir], '
        '[perm_dir], [remove_dir]')
    return 1
  return 0


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