clone-membership.php
ActivePublic
Actions

Authored by cburroughs on Oct 1 2014, 9:06 PM.

Tags

None

Referenced Files

	F211531: clone-membership.php
	Oct 1 2014, 9:13 PM

	F211524: clone-membership.php
	Oct 1 2014, 9:06 PM

Subscribers

None

	#!/usr/bin/env php
	<?php

	$root = dirname(dirname(__FILE__));
	require_once $root.'/scripts/__init_script__.php';

	$args = new PhutilArgumentParser($argv);
	$args->setTagline('set members');
	$args->setSynopsis(<<<EOSYNOPSIS
	cat list \| clone-members.php '#project'
	Set members of '#preffix' to be the same as '#project'.

	EOSYNOPSIS
	);
	$args->parseStandardArguments();
	$args->parse(
	array(
	array(
	'name' => 'dry-run',
	'help' => pht('Only show what changes would be made.'),
	),
	));

	$viewer = PhabricatorUser::getOmnipotentUser();

	// manually maintained config
	$conf = array(
	'Zelda' => 'Zelda-Sprint'
	);

	echo "Run at ".date(DateTime::ISO8601)."...\n";

	foreach($conf as $master_project_name => $clone_prefix) {
	$slug = trim($master_project_name, '#');
	$project = id(new PhabricatorProjectQuery())
	->setViewer($viewer)
	->withSlugs(array($slug))
	->needMembers(true)
	->executeOne();

	if (!$project) {
	throw new Exception(pht('No project "%s" exists!', $slug));
	}

	echo "Processing master project: ".$project->getName()."\n";

	$current_master_phids = $project->getMemberPHIDs();
	$current_master_users = array();
	if (!$current_master_users) {
	$current_master_users = id(new PhabricatorPeopleQuery())
	->setViewer($viewer)
	->withPHIDs($current_master_phids)
	->execute();
	}

	// We need all projects since there is no way to search by prefix
	$all_projects = id(new PhabricatorProjectQuery())
	->setViewer($viewer)
	->needMembers(true)
	->withStatus(PhabricatorProjectQuery::STATUS_ACTIVE)
	->execute();

	$clone_projects = array();
	foreach ($all_projects as $canidate_project) {
	if (startsWith($canidate_project->getName(), $clone_prefix)) {
	echo "Clone Project: ".$canidate_project->getName()."\n";
	cloneProjectMembers($viewer,
	mpull($current_master_users, null, 'getPHID'),
	$canidate_project,
	$args->getArg('dry-run'));
	}
	}
	}

	echo "Done.\n";


	// http://stackoverflow.com/questions/834303/startswith-and-endswith-functions-in-php
	function startsWith($haystack, $needle) {
	return $needle === "" \|\| strpos($haystack, $needle) === 0;
	}


	function cloneProjectMembers($viewer, $current_master_users, $clone_project, $dry_run) {
	$current_clone_phids = $clone_project->getMemberPHIDs();
	$current_clone_users = array();
	if (!empty($current_clone_phids)) {
	$current_clone_users = id(new PhabricatorPeopleQuery())
	->setViewer($viewer)
	->withPHIDs($current_clone_phids)
	->execute();
	}

	$all_users = mpull($current_master_users, null, 'getPHID') +
	mpull($current_clone_users, null, 'getPHID');

	$add = array_diff_key($current_master_users, array_fuse($current_clone_phids));
	$rem = array_diff_key(array_fuse($current_clone_phids), $current_master_users);

	$add_users = array_select_keys($all_users, array_keys($add));
	$rem_users = array_select_keys($all_users, array_keys($rem));

	if (!$add) {
	echo "No users will be added.\n\n";
	} else {
	echo "These users will be added:\n";
	foreach ($add_users as $user) {
	echo " ".$user->getUsername()."\n";
	}
	echo "\n";
	}

	if (!$rem) {
	echo "No users will be removed.\n\n";
	} else {
	echo "These users will be removed:\n";
	foreach ($rem as $rem_phid => $ignored) {
	$name = $rem_phid;
	if (isset($rem_users[$rem_phid])) {
	$name = $rem_users[$rem_phid]->getUsername();
	}
	echo " ".$name."\n";
	}
	echo "\n";
	}

	if ($dry_run) {
	echo "End of dry run.\n";
	exit(0);
	}

	if (!$add && !$rem) {
	echo "No changes to apply.\n";
	exit(0);
	}

	echo "Applying changes...\n";

	$type_member = PhabricatorEdgeConfig::TYPE_PROJ_MEMBER;

	$editor = id(new PhabricatorEdgeEditor());

	foreach ($add as $phid => $ignored) {
	$editor->addEdge($clone_project->getPHID(), $type_member, $phid);
	}
	foreach ($rem as $phid => $ignored) {
	$editor->removeEdge($clone_project->getPHID(), $type_member, $phid);
	}
	$editor->save();


	}

Event Timeline

cburroughs edited the content of this paste. (Show Details)Oct 1 2014, 9:06 PM

cburroughs changed the title of this paste from untitled to clone-membership.php.

cburroughs updated the paste's language from autodetect to autodetect.

cburroughs edited the content of this paste. (Show Details)Oct 1 2014, 9:13 PM

cburroughs mentioned this in T5550: simple CRUD 'api' for project membership.

clone-membership.phpActivePublicActions

Event Timeline

clone-membership.php
ActivePublic
Actions