Elgato_dark/appwrite
1
0
Fork 0
mirror of https://github.com/appwrite/appwrite synced 2026-05-24 09:28:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

chore: refactor code

Browse source
This commit is contained in:
德华 刘 2024-01-22 08:26:38 +00:00
parent 9fc1b0b71b
commit ff7523d6db
1 changed files with 45 additions and 30 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
src/Appwrite/Platform/Workers/Hamster.php
View file

@ -218,6 +218,16 @@ class Hamster extends Action
} }
} }
/** Add billing information to the project */
$organization = $dbForConsole->findOne('teams', [
Query::equal('$internalId', [$teamInternalId])
]);
/** Add billing information */
$billing = $this->getBillingDetails($organization);
$statsPerProject['billing_plan'] = $billing['billing_plan'] ?? null;
$statsPerProject['billing_start_date'] = $billing['billing_start_date'] ?? null;
/** Get Domains */ /** Get Domains */
$statsPerProject['custom_domains'] = $dbForConsole->count('rules', [ $statsPerProject['custom_domains'] = $dbForConsole->count('rules', [
Query::equal('projectInternalId', [$project->getInternalId()]), Query::equal('projectInternalId', [$project->getInternalId()]),
@ -362,32 +372,22 @@ class Hamster extends Action
$membership = $dbForConsole->findOne('memberships', [ $membership = $dbForConsole->findOne('memberships', [
Query::equal('teamInternalId', [$organization->getInternalId()]), Query::equal('teamInternalId', [$organization->getInternalId()]),
]); ]);
if (!$membership || $membership->isEmpty()) { if (!$membership || $membership->isEmpty()) {
throw new \Exception('Membership not found. Skipping organization : ' . $organization->getId()); throw new \Exception('Membership not found. Skipping organization : ' . $organization->getId());
} }
$billingPlan = $organization->getAttribute('billingPlan', null);
$billingPlanDowngrade = $organization->getAttribute('billingPlanDowngrade', null);
if (!empty($billingPlan) && is_null($billingPlanDowngrade)) {
$statsPerOrganization['billing_plan'] = $billingPlan;
}
if (in_array($billingPlan, ['tier-1', 'tier-2'])) {
$billingStartDate = $organization->getAttribute('billingStartDate', null);
$statsPerOrganization['billing_start_date'] = $billingStartDate;
}
$statsPerOrganization['marked_for_deletion'] = $organization->getAttribute('markedForDeletion', 0);
$statsPerOrganization['billing_plan_downgrade'] = $billingPlanDowngrade;
$userId = $membership->getAttribute('userId', null); $userId = $membership->getAttribute('userId', null);
if ($userId) { if ($userId) {
$user = $dbForConsole->getDocument('users', $userId); $user = $dbForConsole->getDocument('users', $userId);
$statsPerOrganization['email'] = $user->getAttribute('email', null); $statsPerOrganization['email'] = $user->getAttribute('email', null);
} }
/** Add billing information */
$billing = $this->getBillingDetails($organization);
$statsPerOrganization['billing_plan'] = $billing['billing_plan'] ?? null;
$statsPerOrganization['billing_start_date'] = $billing['billing_start_date'] ?? null;
$statsPerOrganization['marked_for_deletion'] = $billing['markedForDeletion'] ?? 0;
$statsPerOrganization['billing_plan_downgrade'] = $billing['billing_plan_downgrade'] ?? null;
/** Organization Creation Date */ /** Organization Creation Date */
$statsPerOrganization['created'] = $organization->getAttribute('$createdAt'); $statsPerOrganization['created'] = $organization->getAttribute('$createdAt');
@ -428,19 +428,10 @@ class Hamster extends Action
Query::equal('userInternalId', [$user->getInternalId()]) Query::equal('userInternalId', [$user->getInternalId()])
]); ]);
if (!empty($organization) && !$organization->isEmpty()) { /** Add billing information */
$billingPlan = $organization->getAttribute('billingPlan', null); $billing = $this->getBillingDetails($organization);
$billingPlanDowngrade = $organization->getAttribute('billingPlanDowngrade', null); $statsPerUser['billing_plan'] = $billing['billing_plan'] ?? null;
$statsPerUser['billing_start_date'] = $billing['billing_start_date'] ?? null;
if (!empty($billingPlan) && is_null($billingPlanDowngrade)) {
$statsPerUser['billing_plan'] = $billingPlan;
}
if (in_array($billingPlan, ['tier-1', 'tier-2'])) {
$billingStartDate = $organization->getAttribute('billingStartDate', null);
$statsPerUser['billing_start_date'] = $billingStartDate;
}
}
$statsPerUser['time'] = $user->getAttribute('$time'); $statsPerUser['time'] = $user->getAttribute('$time');
@ -478,4 +469,28 @@ class Hamster extends Action
Console::error($e->getMessage()); Console::error($e->getMessage());
} }
} }
private function getBillingDetails(Document $team): array
{
$billing = [];
if (!empty($team) && !$team->isEmpty()) {
$billingPlan = $team->getAttribute('billingPlan', null);
$billingPlanDowngrade = $team->getAttribute('billingPlanDowngrade', null);
if (!empty($billingPlan) && empty($billingPlanDowngrade)) {
$billing['billing_plan'] = $billingPlan;
}
if (in_array($billingPlan, ['tier-1', 'tier-2'])) {
$billingStartDate = $team->getAttribute('billingStartDate', null);
$billing['billing_start_date'] = $billingStartDate;
}
$billing['marked_for_deletion'] = $team->getAttribute('markedForDeletion', 0);
$billing['billing_plan_downgrade'] = $billingPlanDowngrade;
}
return $billing;
}
} }