Allow dev to apply pivot data on attaching a team.

Signed-off-by: Micheal Mand <[email protected]>

src/Mpociot/Teamwork/Contracts/TeamworkUserInterface.php

@@ -49,8 +49,9 @@ public function isOwnerOfTeam( $team );
  * Alias to eloquent many-to-many relation's attach() method.
  *
  * @param mixed $team
+ * @param array $pivotData
  */
- public function attachTeam( $team );
+ public function attachTeam( $team, $pivotData = [] );
 
  /**
  * Alias to eloquent many-to-many relation's detach() method.

src/Mpociot/Teamwork/Traits/UserHasTeams.php

@@ -130,9 +130,10 @@ public function isOwnerOfTeam( $team )
  * Alias to eloquent many-to-many relation's attach() method.
  *
  * @param mixed $team
+ * @param array $pivotData
  * @return $this
  */
- public function attachTeam( $team )
+ public function attachTeam( $team, $pivotData = [] )
  {
  $team = $this->retrieveTeamId( $team );
  /**
@@ -155,7 +156,7 @@ public function attachTeam( $team )
 
  if( !$this->teams->contains( $team ) )
  {
- $this->teams()->attach( $team );
+ $this->teams()->attach( $team, $pivotData );
 
  if( $this->relationLoaded('teams') ) {
  $this->load('teams');

tests/UserHasTeamsTraitTest.php

@@ -108,6 +108,24 @@ public function testCanAttachTeamAsIDToUser()
  $this->assertEquals(TeamworkTeam::find(1)->toArray(), $this->user->currentTeam->toArray());
  }
 
+ public function testCanSetPivotDataOnAttachTeamMethod()
+ {
+ \Schema::table(config( 'teamwork.team_user_table' ), function ($table) {
+ $table->boolean('pivot_set')->default(false);
+ });
+
+ $team = TeamworkTeam::create(['name' => 'Test-Team']);
+ $pivotData = ['pivot_set' => true];
+
+ $this->user->attachTeam($team, $pivotData);
+
+ $this->seeInDatabase(config('teamwork.team_user_table'), [
+ 'user_id' => $this->user->getKey(),
+ 'team_id' => $team->getKey(),
+ 'pivot_set' => true
+ ]);
+ }
+
  public function testIsTeamOwner()
  {
  $team = TeamworkTeam::create(['name' => 'Test-Team']);