Modernize Visit entity

This commit is contained in:
Alejandro Celaya 2024-03-18 19:57:30 +01:00
parent b2dee43bb0
commit 78526fb405
16 changed files with 68 additions and 108 deletions

View file

@ -44,7 +44,7 @@ class GetDomainVisitsCommand extends AbstractVisitsListCommand
*/ */
protected function mapExtraFields(Visit $visit): array protected function mapExtraFields(Visit $visit): array
{ {
$shortUrl = $visit->getShortUrl(); $shortUrl = $visit->shortUrl;
return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)]; return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)];
} }
} }

View file

@ -44,7 +44,7 @@ class GetTagVisitsCommand extends AbstractVisitsListCommand
*/ */
protected function mapExtraFields(Visit $visit): array protected function mapExtraFields(Visit $visit): array
{ {
$shortUrl = $visit->getShortUrl(); $shortUrl = $visit->shortUrl;
return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)]; return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)];
} }
} }

View file

@ -40,7 +40,7 @@ class GetNonOrphanVisitsCommand extends AbstractVisitsListCommand
*/ */
protected function mapExtraFields(Visit $visit): array protected function mapExtraFields(Visit $visit): array
{ {
$shortUrl = $visit->getShortUrl(); $shortUrl = $visit->shortUrl;
return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)]; return $shortUrl === null ? [] : ['shortUrl' => $this->shortUrlStringifier->stringify($shortUrl)];
} }
} }

View file

@ -42,6 +42,6 @@ class GetOrphanVisitsCommand extends AbstractVisitsListCommand
*/ */
protected function mapExtraFields(Visit $visit): array protected function mapExtraFields(Visit $visit): array
{ {
return ['type' => $visit->type()->value]; return ['type' => $visit->type->value];
} }
} }

View file

@ -132,7 +132,7 @@ class LocateVisitsCommand extends AbstractLockedCommand implements VisitGeolocat
*/ */
public function geolocateVisit(Visit $visit): Location public function geolocateVisit(Visit $visit): Location
{ {
$ipAddr = $visit->getRemoteAddr() ?? '?'; $ipAddr = $visit->remoteAddr ?? '?';
$this->io->write(sprintf('Processing IP <fg=blue>%s</>', $ipAddr)); $this->io->write(sprintf('Processing IP <fg=blue>%s</>', $ipAddr));
try { try {
@ -156,7 +156,7 @@ class LocateVisitsCommand extends AbstractLockedCommand implements VisitGeolocat
{ {
if (! $visitLocation->isEmpty) { if (! $visitLocation->isEmpty) {
$this->io->writeln(sprintf(' [<info>Address located in "%s"</info>]', $visitLocation->countryName)); $this->io->writeln(sprintf(' [<info>Address located in "%s"</info>]', $visitLocation->countryName));
} elseif ($visit->hasRemoteAddr() && $visit->getRemoteAddr() !== IpAddress::LOCALHOST) { } elseif ($visit->hasRemoteAddr() && $visit->remoteAddr !== IpAddress::LOCALHOST) {
$this->io->writeln(' <comment>[Could not locate address]</comment>'); $this->io->writeln(' <comment>[Could not locate address]</comment>');
} }
} }

View file

@ -55,7 +55,7 @@ class LocateVisit
} }
$isLocatable = $originalIpAddress !== null || $visit->isLocatable(); $isLocatable = $originalIpAddress !== null || $visit->isLocatable();
$addr = $originalIpAddress ?? $visit->getRemoteAddr() ?? ''; $addr = $originalIpAddress ?? $visit->remoteAddr ?? '';
try { try {
$location = $isLocatable ? $this->ipLocationResolver->resolveIpLocation($addr) : Location::emptyInstance(); $location = $isLocatable ? $this->ipLocationResolver->resolveIpLocation($addr) : Location::emptyInstance();

View file

@ -46,9 +46,9 @@ class SendVisitToMatomo
$tracker $tracker
->setUrl($this->resolveUrlToTrack($visit)) ->setUrl($this->resolveUrlToTrack($visit))
->setCustomTrackingParameter('type', $visit->type()->value) ->setCustomTrackingParameter('type', $visit->type->value)
->setUserAgent($visit->userAgent()) ->setUserAgent($visit->userAgent)
->setUrlReferrer($visit->referer()); ->setUrlReferrer($visit->referer);
$location = $visit->getVisitLocation(); $location = $visit->getVisitLocation();
if ($location !== null) { if ($location !== null) {
@ -60,7 +60,7 @@ class SendVisitToMatomo
} }
// Set not obfuscated IP if possible, as matomo handles obfuscation itself // Set not obfuscated IP if possible, as matomo handles obfuscation itself
$ip = $visitLocated->originalIpAddress ?? $visit->getRemoteAddr(); $ip = $visitLocated->originalIpAddress ?? $visit->remoteAddr;
if ($ip !== null) { if ($ip !== null) {
$tracker->setIp($ip); $tracker->setIp($ip);
} }
@ -79,9 +79,9 @@ class SendVisitToMatomo
public function resolveUrlToTrack(Visit $visit): string public function resolveUrlToTrack(Visit $visit): string
{ {
$shortUrl = $visit->getShortUrl(); $shortUrl = $visit->shortUrl;
if ($shortUrl === null) { if ($shortUrl === null) {
return $visit->visitedUrl() ?? ''; return $visit->visitedUrl ?? '';
} }
return $this->shortUrlStringifier->stringify($shortUrl); return $this->shortUrlStringifier->stringify($shortUrl);

View file

@ -20,7 +20,7 @@ final class PublishingUpdatesGenerator implements PublishingUpdatesGeneratorInte
public function newVisitUpdate(Visit $visit): Update public function newVisitUpdate(Visit $visit): Update
{ {
return Update::forTopicAndPayload(Topic::NEW_VISIT->value, [ return Update::forTopicAndPayload(Topic::NEW_VISIT->value, [
'shortUrl' => $this->shortUrlTransformer->transform($visit->getShortUrl()), 'shortUrl' => $this->shortUrlTransformer->transform($visit->shortUrl),
'visit' => $visit->jsonSerialize(), 'visit' => $visit->jsonSerialize(),
]); ]);
} }
@ -34,7 +34,7 @@ final class PublishingUpdatesGenerator implements PublishingUpdatesGeneratorInte
public function newShortUrlVisitUpdate(Visit $visit): Update public function newShortUrlVisitUpdate(Visit $visit): Update
{ {
$shortUrl = $visit->getShortUrl(); $shortUrl = $visit->shortUrl;
$topic = Topic::newShortUrlVisit($shortUrl?->getShortCode()); $topic = Topic::newShortUrlVisit($shortUrl?->getShortCode());
return Update::forTopicAndPayload($topic, [ return Update::forTopicAndPayload($topic, [

View file

@ -20,29 +20,22 @@ use function Shlinkio\Shlink\Core\normalizeDate;
class Visit extends AbstractEntity implements JsonSerializable class Visit extends AbstractEntity implements JsonSerializable
{ {
private string $referer; private function __construct(
private Chronos $date; public readonly ?ShortUrl $shortUrl,
private ?string $remoteAddr = null; public readonly VisitType $type,
private ?string $visitedUrl = null; public readonly string $userAgent,
private string $userAgent; public readonly string $referer,
private VisitType $type; private readonly bool $potentialBot,
private ?ShortUrl $shortUrl; public readonly ?string $remoteAddr = null,
private ?VisitLocation $visitLocation = null; public readonly ?string $visitedUrl = null,
private bool $potentialBot; private ?VisitLocation $visitLocation = null,
private Chronos $date = new Chronos(),
private function __construct(?ShortUrl $shortUrl, VisitType $type) ) {
{
$this->shortUrl = $shortUrl;
$this->date = Chronos::now();
$this->type = $type;
} }
public static function forValidShortUrl(ShortUrl $shortUrl, Visitor $visitor, bool $anonymize = true): self public static function forValidShortUrl(ShortUrl $shortUrl, Visitor $visitor, bool $anonymize = true): self
{ {
$instance = new self($shortUrl, VisitType::VALID_SHORT_URL); return self::hydrateFromVisitor($shortUrl, VisitType::VALID_SHORT_URL, $visitor, $anonymize);
$instance->hydrateFromVisitor($visitor, $anonymize);
return $instance;
} }
public static function fromImport(ShortUrl $shortUrl, ImportedShlinkVisit $importedVisit): self public static function fromImport(ShortUrl $shortUrl, ImportedShlinkVisit $importedVisit): self
@ -52,13 +45,10 @@ class Visit extends AbstractEntity implements JsonSerializable
public static function fromOrphanImport(ImportedShlinkOrphanVisit $importedVisit): self public static function fromOrphanImport(ImportedShlinkOrphanVisit $importedVisit): self
{ {
$instance = self::fromImportOrOrphanImport( return self::fromImportOrOrphanImport(
$importedVisit, $importedVisit,
VisitType::tryFrom($importedVisit->type) ?? VisitType::IMPORTED, VisitType::tryFrom($importedVisit->type) ?? VisitType::IMPORTED,
); );
$instance->visitedUrl = $importedVisit->visitedUrl;
return $instance;
} }
private static function fromImportOrOrphanImport( private static function fromImportOrOrphanImport(
@ -66,52 +56,52 @@ class Visit extends AbstractEntity implements JsonSerializable
VisitType $type, VisitType $type,
?ShortUrl $shortUrl = null, ?ShortUrl $shortUrl = null,
): self { ): self {
$instance = new self($shortUrl, $type);
$instance->userAgent = $importedVisit->userAgent;
$instance->potentialBot = isCrawler($instance->userAgent);
$instance->referer = $importedVisit->referer;
$instance->date = normalizeDate($importedVisit->date);
$importedLocation = $importedVisit->location; $importedLocation = $importedVisit->location;
$instance->visitLocation = $importedLocation !== null ? VisitLocation::fromImport($importedLocation) : null; return new self(
shortUrl: $shortUrl,
return $instance; type: $type,
userAgent: $importedVisit->userAgent,
referer: $importedVisit->referer,
potentialBot: isCrawler($importedVisit->userAgent),
visitedUrl: $importedVisit instanceof ImportedShlinkOrphanVisit ? $importedVisit->visitedUrl : null,
visitLocation: $importedLocation !== null ? VisitLocation::fromImport($importedLocation) : null,
date: normalizeDate($importedVisit->date),
);
} }
public static function forBasePath(Visitor $visitor, bool $anonymize = true): self public static function forBasePath(Visitor $visitor, bool $anonymize = true): self
{ {
$instance = new self(null, VisitType::BASE_URL); return self::hydrateFromVisitor(null, VisitType::BASE_URL, $visitor, $anonymize);
$instance->hydrateFromVisitor($visitor, $anonymize);
return $instance;
} }
public static function forInvalidShortUrl(Visitor $visitor, bool $anonymize = true): self public static function forInvalidShortUrl(Visitor $visitor, bool $anonymize = true): self
{ {
$instance = new self(null, VisitType::INVALID_SHORT_URL); return self::hydrateFromVisitor(null, VisitType::INVALID_SHORT_URL, $visitor, $anonymize);
$instance->hydrateFromVisitor($visitor, $anonymize);
return $instance;
} }
public static function forRegularNotFound(Visitor $visitor, bool $anonymize = true): self public static function forRegularNotFound(Visitor $visitor, bool $anonymize = true): self
{ {
$instance = new self(null, VisitType::REGULAR_404); return self::hydrateFromVisitor(null, VisitType::REGULAR_404, $visitor, $anonymize);
$instance->hydrateFromVisitor($visitor, $anonymize);
return $instance;
} }
private function hydrateFromVisitor(Visitor $visitor, bool $anonymize = true): void private static function hydrateFromVisitor(
{ ?ShortUrl $shortUrl,
$this->userAgent = $visitor->userAgent; VisitType $type,
$this->referer = $visitor->referer; Visitor $visitor,
$this->remoteAddr = $this->processAddress($anonymize, $visitor->remoteAddress); bool $anonymize,
$this->visitedUrl = $visitor->visitedUrl; ): self {
$this->potentialBot = $visitor->isPotentialBot(); return new self(
shortUrl: $shortUrl,
type: $type,
userAgent: $visitor->userAgent,
referer: $visitor->referer,
potentialBot: $visitor->isPotentialBot(),
remoteAddr: self::processAddress($anonymize, $visitor->remoteAddress),
visitedUrl: $visitor->visitedUrl,
);
} }
private function processAddress(bool $anonymize, ?string $address): ?string private static function processAddress(bool $anonymize, ?string $address): ?string
{ {
// Localhost addresses do not need to be anonymized // Localhost addresses do not need to be anonymized
if (! $anonymize || $address === null || $address === IpAddress::LOCALHOST) { if (! $anonymize || $address === null || $address === IpAddress::LOCALHOST) {
@ -125,21 +115,11 @@ class Visit extends AbstractEntity implements JsonSerializable
} }
} }
public function getRemoteAddr(): ?string
{
return $this->remoteAddr;
}
public function hasRemoteAddr(): bool public function hasRemoteAddr(): bool
{ {
return ! empty($this->remoteAddr); return ! empty($this->remoteAddr);
} }
public function getShortUrl(): ?ShortUrl
{
return $this->shortUrl;
}
public function getVisitLocation(): ?VisitLocation public function getVisitLocation(): ?VisitLocation
{ {
return $this->visitLocation; return $this->visitLocation;
@ -161,23 +141,13 @@ class Visit extends AbstractEntity implements JsonSerializable
return $this->shortUrl === null; return $this->shortUrl === null;
} }
public function visitedUrl(): ?string
{
return $this->visitedUrl;
}
public function type(): VisitType
{
return $this->type;
}
/** /**
* Needed only for ArrayCollections to be able to apply criteria filtering * Needed only for ArrayCollections to be able to apply criteria filtering
* @internal * @internal
*/ */
public function getType(): VisitType public function getType(): VisitType
{ {
return $this->type(); return $this->type;
} }
/** /**
@ -188,16 +158,6 @@ class Visit extends AbstractEntity implements JsonSerializable
return $this->date; return $this->date;
} }
public function userAgent(): string
{
return $this->userAgent;
}
public function referer(): string
{
return $this->referer;
}
public function jsonSerialize(): array public function jsonSerialize(): array
{ {
return [ return [

View file

@ -26,7 +26,7 @@ class VisitToLocationHelper implements VisitToLocationHelperInterface
throw IpCannotBeLocatedException::forEmptyAddress(); throw IpCannotBeLocatedException::forEmptyAddress();
} }
$ipAddr = $visit->getRemoteAddr() ?? ''; $ipAddr = $visit->remoteAddr ?? '';
if ($ipAddr === IpAddress::LOCALHOST) { if ($ipAddr === IpAddress::LOCALHOST) {
throw IpCannotBeLocatedException::forLocalhost(); throw IpCannotBeLocatedException::forLocalhost();
} }

View file

@ -15,8 +15,8 @@ class OrphanVisitDataTransformer implements DataTransformerInterface
public function transform($visit): array // phpcs:ignore public function transform($visit): array // phpcs:ignore
{ {
$serializedVisit = $visit->jsonSerialize(); $serializedVisit = $visit->jsonSerialize();
$serializedVisit['visitedUrl'] = $visit->visitedUrl(); $serializedVisit['visitedUrl'] = $visit->visitedUrl;
$serializedVisit['type'] = $visit->type()->value; $serializedVisit['type'] = $visit->type->value;
return $serializedVisit; return $serializedVisit;
} }

View file

@ -157,7 +157,7 @@ class LocateVisitTest extends TestCase
#[Test, DataProvider('provideIpAddresses')] #[Test, DataProvider('provideIpAddresses')]
public function locatableVisitsResolveToLocation(Visit $visit, ?string $originalIpAddress): void public function locatableVisitsResolveToLocation(Visit $visit, ?string $originalIpAddress): void
{ {
$ipAddr = $originalIpAddress ?? $visit->getRemoteAddr(); $ipAddr = $originalIpAddress ?? $visit->remoteAddr;
$location = new Location('', '', '', '', 0.0, 0.0, ''); $location = new Location('', '', '', '', 0.0, 0.0, '');
$event = new UrlVisited('123', $originalIpAddress); $event = new UrlVisited('123', $originalIpAddress);

View file

@ -76,13 +76,13 @@ class SendVisitToMatomoTest extends TestCase
if ($visit->isOrphan()) { if ($visit->isOrphan()) {
$tracker->expects($this->exactly(2))->method('setCustomTrackingParameter')->willReturnMap([ $tracker->expects($this->exactly(2))->method('setCustomTrackingParameter')->willReturnMap([
['type', $visit->type()->value, $tracker], ['type', $visit->type->value, $tracker],
['orphan', 'true', $tracker], ['orphan', 'true', $tracker],
]); ]);
} else { } else {
$tracker->expects($this->once())->method('setCustomTrackingParameter')->with( $tracker->expects($this->once())->method('setCustomTrackingParameter')->with(
'type', 'type',
$visit->type()->value, $visit->type->value,
)->willReturn($tracker); )->willReturn($tracker);
} }

View file

@ -93,8 +93,8 @@ class PublishingUpdatesGeneratorTest extends TestCase
'visitLocation' => null, 'visitLocation' => null,
'date' => $orphanVisit->getDate()->toAtomString(), 'date' => $orphanVisit->getDate()->toAtomString(),
'potentialBot' => false, 'potentialBot' => false,
'visitedUrl' => $orphanVisit->visitedUrl(), 'visitedUrl' => $orphanVisit->visitedUrl,
'type' => $orphanVisit->type()->value, 'type' => $orphanVisit->type->value,
], ],
], $update->payload); ], $update->payload);
} }

View file

@ -244,7 +244,7 @@ class ImportedLinksProcessorTest extends TestCase
$this->em->expects($this->once())->method('persist')->willReturnCallback( $this->em->expects($this->once())->method('persist')->willReturnCallback(
static fn (Visit $visit) => Assert::assertSame( static fn (Visit $visit) => Assert::assertSame(
$foundShortUrl ?? $originalShortUrl, $foundShortUrl ?? $originalShortUrl,
$visit->getShortUrl(), $visit->shortUrl,
), ),
); );

View file

@ -49,7 +49,7 @@ class VisitTest extends TestCase
$anonymize, $anonymize,
); );
self::assertEquals($expectedAddress, $visit->getRemoteAddr()); self::assertEquals($expectedAddress, $visit->remoteAddr);
} }
public static function provideAddresses(): iterable public static function provideAddresses(): iterable