100 lines
3.7 KiB
PHP
100 lines
3.7 KiB
PHP
<?php
|
|
|
|
class UsageGuard {
|
|
|
|
private PDO $pdo;
|
|
private int $company_id;
|
|
private array $pkg;
|
|
|
|
public function __construct(PDO $pdo1, int $company_id, array $packages) {
|
|
$this->pdo = $pdo1;
|
|
$this->company_id = $company_id;
|
|
$this->pkg = $this->resolvePackage($packages);
|
|
}
|
|
|
|
private function resolvePackage(array $packages): array {
|
|
$sth = $this->pdo->prepare(
|
|
"SELECT package FROM company_list WHERE company_id = :cid LIMIT 1"
|
|
);
|
|
$sth->execute([':cid' => $this->company_id]);
|
|
$name = $sth->fetchColumn() ?: 'starter';
|
|
return $packages[$name] ?? $packages['starter'] ?? [
|
|
'daily_limit' => 30,
|
|
'weekly_limit' => 100,
|
|
'lock_on_limit' => ['dashboard'],
|
|
];
|
|
}
|
|
|
|
public function increment(): void {
|
|
$today = date('Y-m-d');
|
|
$this->pdo->prepare(
|
|
"INSERT INTO company_usage (company_id, day_date, daily_count)
|
|
VALUES (:cid, :today, 1)
|
|
ON DUPLICATE KEY UPDATE daily_count = daily_count + 1"
|
|
)->execute([':cid' => $this->company_id, ':today' => $today]);
|
|
}
|
|
|
|
public function assertFeatureAccessible(string $feature): void {
|
|
$lock_on = $this->pkg['lock_on_limit'] ?? [];
|
|
if (!in_array($feature, $lock_on, true)) {
|
|
return;
|
|
}
|
|
|
|
$status = $this->getStatus();
|
|
$daily_limit = (int)($this->pkg['daily_limit'] ?? 0);
|
|
$weekly_limit = (int)($this->pkg['weekly_limit'] ?? 0);
|
|
|
|
$over_daily = $daily_limit > 0 && $status['daily_count'] >= $daily_limit;
|
|
$over_weekly = $weekly_limit > 0 && $status['weekly_count'] >= $weekly_limit;
|
|
|
|
if ($over_daily || $over_weekly) {
|
|
http_response_code(402);
|
|
exit(json_encode([
|
|
'success' => 0,
|
|
'message' => 'Usage limit reached. Upgrade your package to access reports.',
|
|
'limit_reached' => true,
|
|
'daily_count' => $status['daily_count'],
|
|
'daily_limit' => $status['daily_limit'],
|
|
'weekly_count' => $status['weekly_count'],
|
|
'weekly_limit' => $status['weekly_limit'],
|
|
]));
|
|
}
|
|
}
|
|
|
|
public function getStatus(): array {
|
|
$today = date('Y-m-d');
|
|
$day_of_week = (int)date('N');
|
|
$week_start = date('Y-m-d', strtotime('-' . ($day_of_week - 1) . ' days'));
|
|
|
|
$sth = $this->pdo->prepare(
|
|
"SELECT daily_count FROM company_usage
|
|
WHERE company_id = :cid AND day_date = :today LIMIT 1"
|
|
);
|
|
$sth->execute([':cid' => $this->company_id, ':today' => $today]);
|
|
$daily_count = (int)($sth->fetchColumn() ?: 0);
|
|
|
|
$sth = $this->pdo->prepare(
|
|
"SELECT COALESCE(SUM(daily_count), 0) FROM company_usage
|
|
WHERE company_id = :cid AND day_date BETWEEN :wstart AND :today"
|
|
);
|
|
$sth->execute([
|
|
':cid' => $this->company_id,
|
|
':wstart'=> $week_start,
|
|
':today' => $today,
|
|
]);
|
|
$weekly_count = (int)$sth->fetchColumn();
|
|
|
|
$daily_limit = (int)($this->pkg['daily_limit'] ?? 0);
|
|
$weekly_limit = (int)($this->pkg['weekly_limit'] ?? 0);
|
|
|
|
return [
|
|
'daily_count' => $daily_count,
|
|
'weekly_count' => $weekly_count,
|
|
'daily_limit' => $daily_limit,
|
|
'weekly_limit' => $weekly_limit,
|
|
'daily_pct' => $daily_limit > 0 ? min(100, (int)round($daily_count / $daily_limit * 100)) : 0,
|
|
'weekly_pct' => $weekly_limit > 0 ? min(100, (int)round($weekly_count / $weekly_limit * 100)) : 0,
|
|
];
|
|
}
|
|
}
|