ObligationCsvExport.php

Go to the documentation of this file.

<?php
/*
 SPDX-FileCopyrightText: © 2017 Siemens AG
 
 SPDX-License-Identifier: GPL-2.0-only
*/
 
namespace Fossology\Lib\Application;
 
use Fossology\Lib\Db\DbManager;
 
class ObligationCsvExport
{
  protected $dbManager;
  protected $delimiter = ',';
  protected $enclosure = '"';
 
  public function __construct(DbManager $dbManager)
  {
    $this->dbManager = $dbManager;
    $this->obligationMap = $GLOBALS['container']->get('businessrules.obligationmap');
  }
 
  public function setDelimiter($delimiter=',')
  {
    $this->delimiter = substr($delimiter,0,1);
  }
 
  public function setEnclosure($enclosure='"')
  {
    $this->enclosure = substr($enclosure,0,1);
  }
 
  public function createCsv($ob=0, $generateJson=false)
  {
    $csvarray = array();
    $sql = "SELECT ob_pk,ob_type AS Type,ob_topic AS \"Obligation or Risk topic\",ob_text AS \"Full Text\",
            ob_classification AS Classification,ob_modifications AS \"Apply on modified source code\",ob_comment AS Comment
            FROM obligation_ref";
    if ($ob>0) {
      $stmt = __METHOD__.'.ob';
      $sql .= ' WHERE ob_pk=$1;';
      $row = $this->dbManager->getSingleRow($sql, [$ob], $stmt);
      $liclist = $this->obligationMap->getLicenseList($ob);
      $candidatelist = $this->obligationMap->getLicenseList($ob, True);
      array_shift($row);
      $row["Associated Licenses"] = $liclist;
      $row["Associated candidate Licenses"] = $candidatelist;
      $csvarray[] = $row;
    } else {
      $stmt = __METHOD__;
      $this->dbManager->prepare($stmt,$sql);
      $res = $this->dbManager->execute($stmt);
      $vars = $this->dbManager->fetchAll($res);
      $this->dbManager->freeResult($res);
 
      foreach ($vars as $row) {
        $liclist = $this->obligationMap->getLicenseList($row['ob_pk']);
        $candidatelist = $this->obligationMap->getLicenseList($row['ob_pk'], True);
        array_shift($row);
        $row["Associated Licenses"] = $liclist;
        $row["Associated candidate Licenses/"] = $candidatelist;
        $csvarray[] = $row;
      }
    }
    if ($generateJson) {
      return json_encode($csvarray, JSON_PRETTY_PRINT);
    } else {
      $out = fopen('php://output', 'w');
      ob_start();
      $head = array('Type','Obligation or Risk topic','Full Text','Classification','Apply on modified source code','Comment','Associated Licenses','Associated candidate Licenses');
      fputcsv($out, $head, $this->delimiter, $this->enclosure);
      foreach ($csvarray as $row) {
        fputcsv($out, $row, $this->delimiter, $this->enclosure);
      }
      $content = ob_get_contents();
      ob_end_clean();
      return $content;
    }
  }
}