コード例 #1
0
  @SuppressWarnings("unchecked")
  @Override
  public List<Record> authorizeByConnector(
      List<Record> records,
      Collection<UserCredentials> userCredentialsList,
      ConnectorManager connectorManager) {
    List<Record> authorizedRecords = new ArrayList<Record>();

    Document document = DocumentHelper.createDocument();
    Element root = document.addElement(ServletUtil.XMLTAG_AUTHZ_QUERY);
    Element connectorQueryElement = root.addElement(ServletUtil.XMLTAG_CONNECTOR_QUERY);

    Map<ConnectorInstance, UserCredentials> credentialsMap =
        new HashMap<ConnectorInstance, UserCredentials>();
    Set<ConnectorInstance> connectorsWithoutCredentials = new HashSet<ConnectorInstance>();
    Map<String, Record> recordsByURLMap = new HashMap<String, Record>();
    boolean recordToValidate = false;
    for (Record record : records) {
      // Use to accelerate the matching between response urls and actual entities
      recordsByURLMap.put(record.getUrl(), record);
      ConnectorInstance connectorInstance = record.getConnectorInstance();
      UserCredentials connectorCredentials = credentialsMap.get(connectorInstance);
      if (connectorCredentials == null
          && !connectorsWithoutCredentials.contains(connectorInstance)) {
        RecordCollection collection = connectorInstance.getRecordCollection();
        for (CredentialGroup credentialGroup : collection.getCredentialGroups()) {
          if (credentialGroup.getConnectorInstances().contains(connectorInstance)) {
            for (UserCredentials userCredentials : userCredentialsList) {
              if (userCredentials.getCredentialGroup().equals(credentialGroup)) {
                connectorCredentials = userCredentials;
                credentialsMap.put(connectorInstance, userCredentials);
                break;
              }
            }
            break;
          }
        }
      }
      if (connectorCredentials == null) {
        connectorsWithoutCredentials.add(connectorInstance);
        LOGGER.warning("Missing credentials for connector " + connectorInstance.getName());
      } else {
        String username = connectorCredentials.getUsername();
        if (StringUtils.isNotBlank(username)) {
          String password = EncryptionUtils.decrypt(connectorCredentials.getEncryptedPassword());
          String domain = connectorCredentials.getDomain();
          Element identityElement = connectorQueryElement.addElement(ServletUtil.XMLTAG_IDENTITY);
          identityElement.setText(username);
          if (StringUtils.isNotBlank(domain)) {
            identityElement.addAttribute(ServletUtil.XMLTAG_DOMAIN_ATTRIBUTE, domain);
          }
          identityElement.addAttribute(ServletUtil.XMLTAG_PASSWORD_ATTRIBUTE, password);

          Element resourceElement = identityElement.addElement(ServletUtil.XMLTAG_RESOURCE);
          resourceElement.setText(record.getUrl());
          resourceElement.addAttribute(
              ServletUtil.XMLTAG_CONNECTOR_NAME_ATTRIBUTE, connectorInstance.getName());
          recordToValidate = true;
        }
      }
    }

    if (recordToValidate) {
      Element response =
          ConnectorManagerRequestUtils.sendPost(connectorManager, "/authorization", document);
      Element authzResponseElement = response.element(ServletUtil.XMLTAG_AUTHZ_RESPONSE);
      List<Element> answerElements = authzResponseElement.elements(ServletUtil.XMLTAG_ANSWER);
      for (Element answerElement : answerElements) {
        Element decisionElement = answerElement.element(ServletUtil.XMLTAG_DECISION);
        boolean permit = decisionElement.getTextTrim().equals("Permit");
        if (permit) {
          Element resourceElement = answerElement.element(ServletUtil.XMLTAG_RESOURCE);
          String recordUrl = resourceElement.getTextTrim();
          Record record = recordsByURLMap.get(recordUrl);
          authorizedRecords.add(record);
        }
      }
    }
    return authorizedRecords;
  }
コード例 #2
0
  @Override
  public void synchronizeWithDatabase(ConnectorManager connectorManager) {
    ConnectorTypeServices connectorTypeServices = ConstellioSpringUtils.getConnectorTypeServices();

    // Add missing connector types
    List<String> connectorTypeNames = getConnectorTypes(connectorManager);
    for (String connectorTypeName : connectorTypeNames) {
      ConnectorType connectorType = connectorTypeServices.get(connectorTypeName);
      if (connectorType == null) {
        connectorType = new ConnectorType();
        connectorType.setName(connectorTypeName);

        File connectorsDir = ConstellioSpringUtils.getGoogleConnectorsDir();
        File connectorTypeDir = new File(connectorsDir, connectorType.getName());

        File iconFile = new File(connectorTypeDir, "icon.gif");
        if (iconFile.exists()) {
          try {
            byte[] iconBytes = FileUtils.readFileToByteArray(iconFile);
            connectorType.setIconFileContent(iconBytes);
          } catch (IOException e) {
            throw new RuntimeException(e);
          }
        }

        File connectorTypeMetaXmlFile = new File(connectorTypeDir, "connectorTypeMeta.xml");
        if (connectorTypeMetaXmlFile.exists()) {
          String path = connectorTypeMetaXmlFile.toURI().toString();
          BeanFactory connectorTypeMeta = new FileSystemXmlApplicationContext(path);
          if (connectorTypeMeta.containsBean("searchResultPanelClassName")) {
            String searchResultPanelClassName =
                (String) connectorTypeMeta.getBean("searchResultPanelClassName");
            connectorType.setSearchResultPanelClassName(searchResultPanelClassName);
          }
          if (connectorTypeMeta.containsBean("initInstanceHandlerClassName")) {
            String initInstancePluginClassName =
                (String) connectorTypeMeta.getBean("initInstanceHandlerClassName");
            connectorType.setInitInstanceHandlerClassName(initInstancePluginClassName);
          }
        }

        connectorType.setConnectorManager(connectorManager);
        connectorTypeServices.makePersistent(connectorType);
      }
    }

    // Remove deleted connector types
    List<ConnectorType> dbConnectorType = connectorTypeServices.list();
    for (ConnectorType connectorType : dbConnectorType) {
      if (!connectorTypeNames.contains(connectorType.getName())) {
        connectorTypeServices.makeTransient(connectorType);
      }
    }

    ConnectorInstanceServices connectorInstanceServices =
        ConstellioSpringUtils.getConnectorInstanceServices();
    BackupServices backupServices = ConstellioSpringUtils.getBackupServices();
    List<ConnectorInstance> connectorInstances = connectorInstanceServices.list();
    for (ConnectorInstance connectorInstance : connectorInstances) {
      String connectorName = connectorInstance.getName();
      String connectorTypeName = connectorInstance.getConnectorType().getName();
      boolean existingConnector = isExistingConnector(connectorManager, connectorName);
      boolean hasConfigBackup = backupServices.hasConfigBackup(connectorName, connectorTypeName);
      if (!existingConnector && hasConfigBackup) {
        backupServices.restoreConfigBackup(connectorName, connectorTypeName);
      } else if (existingConnector && !hasConfigBackup) {
        backupServices.backupConfig(connectorName, connectorTypeName);
      }
    }
  }