Skip to content

Commit

Permalink
Merge pull request #137 from Inchoo/feature/#136-block-by-ip-acl-and-…
Browse files Browse the repository at this point in the history 
…country

Feature/#136 block by ip acl and country
  • Loading branch information
@vvuksan
vvuksan authored Feb 14, 2018
2 parents 43cd521 + b59f6a7 commit 6a73be5
Show file tree
Hide file tree
Showing 13 changed files with 883 additions and 2 deletions.
66 changes: 66 additions & 0 deletions Block/System/Config/Form/Field/BlockingBtn.php
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
<?php

namespace Fastly\Cdn\Block\System\Config\Form\Field;

use Magento\Config\Block\System\Config\Form\Field;
use Magento\Framework\Data\Form\Element\AbstractElement;

class BlockingBtn extends Field
{
protected function _construct() // @codingStandardsIgnoreLine - required by parent class
{
$this->_template = 'Fastly_Cdn::system/config/form/field/blockingBtn.phtml';

parent::_construct();
}

/**
* Remove scope label
*
* @param AbstractElement $element
* @return string
*/
public function render(AbstractElement $element)
{
$element->unsScope()->unsCanUseWebsiteValue()->unsCanUseDefaultValue();
return parent::render($element);
}

/**
* Return element html
*
* @param AbstractElement $element
* @return string
*/
protected function _getElementHtml(AbstractElement $element) // @codingStandardsIgnoreLine - required by parent class
{
return $this->_toHtml();
}

/**
* Return ajax url for collect button
*
* @return string
*/
public function getAjaxUrl()
{
return $this->getUrl('adminhtml/fastlyCdn/vcl/serviceinfo');
}

/**
* Render Blocking button HTML
* @return mixed
* @throws \Magento\Framework\Exception\LocalizedException
*/
public function getButtonHtml()
{
$button = $this->getLayout()->createBlock(
'Magento\Backend\Block\Widget\Button'
)->setData([
'id' => 'fastly_blocking_button',
'label' => __('Toggle Blocking')
]);

return $button->toHtml();
}
}
71 changes: 71 additions & 0 deletions Controller/Adminhtml/FastlyCdn/Edge/Acl/GetAcl.php
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
<?php

namespace Fastly\Cdn\Controller\Adminhtml\FastlyCdn\Edge\Acl;

use Magento\Framework\Controller\Result\JsonFactory;
use Magento\Backend\App\Action;
use Magento\Backend\App\Action\Context;
use Fastly\Cdn\Model\Api;
use Fastly\Cdn\Helper\Acl;

class GetAcl extends Action
{
/**
* @var JsonFactory
*/
private $resultJson;

/**
* @var Api
*/
private $api;

/**
* @var Acl
*/
private $acl;

/**
* GetAcl constructor.
* @param Context $context
* @param JsonFactory $resultJsonFactory
* @param Api $api
* @param Acl $acl
*/
public function __construct(
Context $context,
JsonFactory $resultJsonFactory,
Api $api,
Acl $acl
) {
$this->resultJson = $resultJsonFactory;
$this->api = $api;
$this->acl = $acl;
parent::__construct($context);
}

/**
* @return $this|\Magento\Framework\App\ResponseInterface|\Magento\Framework\Controller\ResultInterface
*/
public function execute()
{
$result = $this->resultJson->create();

try {
$service = $this->api->checkServiceDetails();
if (!$service) {
return $result->setData([
'status' => false,
'msg' => 'Failed to check Service details.'
]);
}

return $result->$this->acl->determineVersions($service->versions);
} catch (\Exception $e) {
return $result->setData([
'status' => false,
'msg' => $e->getMessage()
]);
}
}
}
238 changes: 238 additions & 0 deletions Controller/Adminhtml/FastlyCdn/Vcl/Blocking.php
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,238 @@
<?php

namespace Fastly\Cdn\Controller\Adminhtml\FastlyCdn\Vcl;

use Magento\Backend\App\Action;
use Magento\Backend\App\Action\Context;
use Magento\Framework\App\Request\Http;
use Magento\Framework\Controller\Result\JsonFactory;
use Fastly\Cdn\Model\Config;
use Fastly\Cdn\Model\Api;
use Fastly\Cdn\Helper\Vcl;

class Blocking extends Action
{
/**
* @var Http
*/
private $request;

/**
* @var JsonFactory
*/
private $resultJson;

/**
* @var Config
*/
private $config;

/**
* @var Api
*/
private $api;

/**
* @var Vcl
*/
private $vcl;

/**
* Blocking constructor.
*
* @param Context $context
* @param Http $request
* @param JsonFactory $resultJsonFactory
* @param Config $config
* @param Api $api
* @param Vcl $vcl
*/
public function __construct(
Context $context,
Http $request,
JsonFactory $resultJsonFactory,
Config $config,
Api $api,
Vcl $vcl
) {
$this->request = $request;
$this->resultJson = $resultJsonFactory;
$this->config = $config;
$this->api = $api;
$this->vcl = $vcl;
parent::__construct($context);
}

/**
* Upload Blocking snippets
*
* @return $this|\Magento\Framework\App\ResponseInterface|\Magento\Framework\Controller\ResultInterface
*/
public function execute()
{
try {
$result = $this->resultJson->create();
$activeVersion = $this->getRequest()->getParam('active_version');
$activateVcl = $this->getRequest()->getParam('activate_flag');
$service = $this->api->checkServiceDetails();

if (!$service) {
return $result->setData([
'status' => false,
'msg' => 'Failed to check Service details.'
]);
}

$currActiveVersion = $this->vcl->determineVersions($service->versions);
if ($currActiveVersion['active_version'] != $activeVersion) {
return $result->setData([
'status' => false,
'msg' => 'Active versions mismatch.'
]);
}

$clone = $this->api->cloneVersion($currActiveVersion['active_version']);
if (!$clone) {
return $result->setData([
'status' => false,
'msg' => 'Failed to clone active version.'
]);
}

$reqName = Config::FASTLY_MAGENTO_MODULE . '_blocking';
$checkIfReqExist = $this->api->getRequest($activeVersion, $reqName);
$snippet = $this->config->getVclSnippets('/vcl_snippets_blocking', 'recv.vcl');

$blockedCountries = $this->config->getBlockByCountry();
$blockedAcls = $this->config->getBlockByAcl();

$country_codes = '';
$acls = '';

if ($blockedCountries != null) {
$blockedCountriesPieces = explode(",", $blockedCountries);
foreach ($blockedCountriesPieces as $code) {
$country_codes .= ' client.geo.country_code == "' . $code . '" ||';
}
}

if ($blockedAcls != null) {
$blockedAclsPieces = explode(",", $blockedAcls);
foreach ($blockedAclsPieces as $acl) {
$acls .= ' client.ip ~ ' . $acl . ' ||';
}
}

$blockedItems = $country_codes . $acls;
$strippedBlockedItems = substr($blockedItems, 0, strrpos($blockedItems, '||', -1));

$condition = [
'name' => Config::FASTLY_MAGENTO_MODULE . '_block',
'statement' => 'req.http.x-pass',
'type' => 'REQUEST',
'priority' => 5
];

$createCondition = $this->api->createCondition($clone->number, $condition);
if (!$createCondition) {
return $result->setData([
'status' => false,
'msg' => 'Failed to create a REQUEST condition.'
]);
}

if (!$checkIfReqExist) {
$request = [
'name' => $reqName,
'service_id' => $service->id,
'version' => $currActiveVersion['active_version'],
'force_ssl' => true,
'request_condition' => $createCondition->name
];

$createReq = $this->api->createRequest($clone->number, $request);
if (!$createReq) {
return $result->setData([
'status' => false,
'msg' => 'Failed to create the REQUEST object.'
]);
}

// Add blocking snippet
foreach ($snippet as $key => $value) {
if ($strippedBlockedItems === '') {
$value = '';
} else {
$value = str_replace('####BLOCKED_ITEMS####', $strippedBlockedItems, $value);
}

$snippetData = [
'name' => Config::FASTLY_MAGENTO_MODULE . '_blocking_' . $key,
'type' => $key,
'dynamic' => "0",
'priority' => 5,
'content' => $value
];

$status = $this->api->uploadSnippet($clone->number, $snippetData);

if (!$status) {
return $result->setData([
'status' => false,
'msg' => 'Failed to upload the Snippet file.'
]);
}
}
} else {
$deleteRequest = $this->api->deleteRequest($clone->number, $reqName);
if (!$deleteRequest) {
return $result->setData([
'status' => false,
'msg' => 'Failed to delete the REQUEST object.'
]);
}

// Remove blocking snippet
foreach ($snippet as $key => $value) {
$name = Config::FASTLY_MAGENTO_MODULE . '_blocking_' . $key;
$status = $this->api->removeSnippet($clone->number, $name);
if (!$status) {
return $result->setData([
'status' => false,
'msg' => 'Failed to remove the Snippet file.'
]);
}
}
}

$validate = $this->api->validateServiceVersion($clone->number);
if ($validate->status == 'error') {
return $result->setData([
'status' => false,
'msg' => 'Failed to validate service version: ' . $validate->msg
]);
}

if ($activateVcl === 'true') {
$this->api->activateVersion($clone->number);
}

if ($this->config->areWebHooksEnabled() && $this->config->canPublishConfigChanges()) {
if ($checkIfReqExist) {
$this->api->sendWebHook('*Blocking has been turned OFF in Fastly version ' . $clone->number . '*');
} else {
$this->api->sendWebHook('*Blocking has been turned ON in Fastly version ' . $clone->number . '*');
}
}

return $result->setData([
'status' => true
]);
} catch (\Exception $e) {
return $result->setData([
'status' => false,
'msg' => $e->getMessage()
]);
}
}
}
Loading

0 comments on commit 6a73be5

Please sign in to comment.