git.useribm.hu - mediacube.git/blob

   1 package user.jobengine.server.steps;\r
   2 \r
   3 import java.io.OutputStream;\r
   4 import java.net.InetAddress;\r
   5 import java.net.URI;\r
   6 import java.util.Calendar;\r
   7 import java.util.Date;\r
   8 import java.util.List;\r
   9 \r
  10 import org.apache.commons.lang.StringUtils;\r
  11 import org.apache.commons.net.ftp.FTP;\r
  12 import org.apache.commons.net.ftp.FTPClient;\r
  13 import org.apache.commons.net.ftp.FTPReply;\r
  14 import org.apache.logging.log4j.LogManager;\r
  15 import org.apache.logging.log4j.Logger;\r
  16 \r
  17 import com.ibm.nosql.json.api.BasicDBObject;\r
  18 import com.ibm.nosql.json.api.DB;\r
  19 import com.ibm.nosql.json.api.DBCollection;\r
  20 import com.ibm.nosql.json.api.DBCursor;\r
  21 import com.ibm.nosql.json.api.DBObject;\r
  22 \r
  23 import user.commons.CalendarUtils;\r
  24 import user.commons.ListUtils;\r
  25 import user.commons.StoreUri;\r
  26 import user.commons.nosql.NoSQLUtils;\r
  27 import user.commons.octopus.IOctopusAPI;\r
  28 import user.commons.octopus.OctopusAPI;\r
  29 import user.commons.remotestore.FtpDirectoryLister;\r
  30 import user.commons.remotestore.RemoteStoreProtocol;\r
  31 import user.jobengine.db.IItemManager;\r
  32 import user.jobengine.server.IJobEngine;\r
  33 import user.jobengine.server.IJobRuntime;\r
  34 import user.jobengine.server.steps.MetadataTypeDetector.MetadataType;\r
  35 \r
  36 public class CopyForArchiveNEXIOMaterialsStep extends JobStep {\r
  37         private static final String SCHEDULED_FORMAT = "yyyy.MM.dd HH:mm";\r
  38         private static final Logger logger = LogManager.getLogger();\r
  39         private static final String UTF_8 = "utf-8";\r
  40         private static final String JSON_EXT = ".json";\r
  41         private static final String XML_EXT = ".xml";\r
  42         private static final String DURATION = "duration";\r
  43         private static final String MXFEXT = ".MXF";\r
  44         private static final String NEXIOCLIPS = "nexioclips";\r
  45         private static final String LONGNAMEID = "longnameid";\r
  46         private static final String ARCHIVEDRUNDOWNS = "archivedrundowns";\r
  47         private static final String ID = "id";\r
  48         private OctopusAPI octopusAPI;\r
  49         private IItemManager manager;\r
  50 \r
  51         private DB db;\r
  52         private FTPClient sourceFtp;\r
  53         private FTPClient targetFtp;\r
  54         private StoreUri sourceUri;\r
  55         private StoreUri targetUri;\r
  56         private int nexioKillDateDays;\r
  57         private String nexioAgency;\r
  58 \r
  59         private int check(int value, String name) {\r
  60                 if (value == 0) {\r
  61                         logger.error(getMarker(), "A folyamat '{}' bemeneti paramétere 0.", name);\r
  62                         throw new NullPointerException(String.format("System is not configured properly, missing '%s' input parameter.", name));\r
  63                 }\r
  64                 return value;\r
  65         }\r
  66 \r
  67         private String check(String value, String name) {\r
  68                 if (value == null) {\r
  69                         logger.error(getMarker(), "A folyamat '{}' bemeneti paramétere üres.", name);\r
  70                         throw new NullPointerException(String.format("System is not configured properly, missing '%s' input parameter.", name));\r
  71                 }\r
  72                 return value;\r
  73         }\r
  74 \r
  75         private void copy(RundownArchive rundownArchive) throws Exception {\r
  76                 for (StoryArchive storyArchive : rundownArchive.getStoryArchives()) {\r
  77                         for (FileArchive fileArchive : storyArchive.getFileArchives()) {\r
  78                                 try {\r
  79                                         copyFile(fileArchive, rundownArchive, storyArchive);\r
  80                                 } catch (Exception e) {\r
  81                                         logger.error(getMarker(), "A '{}' clip archiválása sikertelen. A rendszer üzenete: {}", fileArchive.getFileName(), e.getMessage());\r
  82                                 }\r
  83                         }\r
  84                 }\r
  85         }\r
  86 \r
  87         private void copyFile(FileArchive fileArchive, RundownArchive rundownArchive, StoryArchive storyArchive) throws Exception {\r
  88                 String origFileName = fileArchive.getFileName();\r
  89                 String fileName = String.format("%s-%s", origFileName, rundownArchive.getItemHouseId());\r
  90                 String videoFileName = fileName + MXFEXT;\r
  91 \r
  92                 transferFile(origFileName + MXFEXT, videoFileName);\r
  93                 BasicDBObject metadata = createMetadata(rundownArchive, storyArchive, fileArchive);\r
  94                 transferMetadata(videoFileName, metadata);\r
  95                 createSourceKillDateFile(rundownArchive, origFileName);\r
  96         }\r
  97 \r
  98         private BasicDBObject createMetadata(RundownArchive rundownArchive, StoryArchive storyArchive, FileArchive fileArchive) {\r
  99                 //              {\r
 100                 //                        "itemHouseId": "43",\r
 101                 //                        "itemTitle": "Hazahúzó",\r
 102                 //                        "itemDescription": null,\r
 103                 //                        "mediaHouseId": "002570",\r
 104                 //                        "mediaTitle": "2017.12.13",\r
 105                 //                        "mediaDescription": null,\r
 106                 //                        "ok": false,\r
 107                 //                        "userName": "echotest"\r
 108                 //              }\r
 109                 BasicDBObject result = new BasicDBObject();\r
 110                 result.put("itemHouseId", rundownArchive.getItemHouseId());\r
 111                 result.put("itemTitle", rundownArchive.getItemTitle());\r
 112                 result.put("itemDescription", rundownArchive.getItemDesc());\r
 113                 result.put("userName", "mediacube");\r
 114 \r
 115                 result.put("mediaHouseId", storyArchive.getMediaHouseId());\r
 116                 result.put("mediaTitle", storyArchive.getMediaTitle());\r
 117                 result.put("mediaDescription", storyArchive.getMediaDesc());\r
 118 \r
 119                 result.put("duration", fileArchive.getDuration());\r
 120                 return result;\r
 121         }\r
 122 \r
 123         private void createSourceKillDateFile(RundownArchive rundownArchive, String fileName) throws Exception {\r
 124                 logger.info("Create killdate/agency for {}", fileName);\r
 125                 OutputStream outStream = null;\r
 126                 try {\r
 127                         sourceFtp = ((FtpDirectoryLister) sourceUri.getLister()).connect();\r
 128                         Calendar killDate = CalendarUtils.createCalendar(rundownArchive.getScheduleDate());\r
 129                         killDate.add(Calendar.DAY_OF_YEAR, nexioKillDateDays);\r
 130                         byte[] killDateFile = EscortFiles.createNEXIOKillDateFile(fileName, killDate.getTime(), null, nexioAgency);\r
 131                         outStream = sourceFtp.storeFileStream(fileName + XML_EXT);\r
 132                         if (outStream == null) {\r
 133                                 throw new NullPointerException("Can not open: " + fileName + XML_EXT + " Reply:" + sourceFtp.getReplyString());\r
 134                         }\r
 135                         outStream.write(killDateFile);\r
 136                         outStream.flush();\r
 137                 } catch (Exception e) {\r
 138                         logger.catching(e);\r
 139                         throw e;\r
 140                 } finally {\r
 141                         if (outStream != null)\r
 142                                 outStream.close();\r
 143                         sourceUri.cleanUp();\r
 144                 }\r
 145         }\r
 146 \r
 147         @StepEntry\r
 148         public Object[] execute(int nexioPort, String nexioUserName, String nexioPassword, String archiveFtp, String archiveUserName, String archivePassword,\r
 149                         int daysBeforeNow, int nexioKillDateDays, String nexioAgency, IJobEngine jobEngine, IJobRuntime jobRuntime) throws Exception {\r
 150                 setAndCheck(nexioPort, nexioUserName, nexioPassword, archiveFtp, archiveUserName, archivePassword, nexioKillDateDays, nexioAgency, jobEngine);\r
 151                 octopusAPI = new OctopusAPI();\r
 152                 Calendar scheduledDate = Calendar.getInstance();\r
 153                 scheduledDate.add(Calendar.DAY_OF_YEAR, -1 * daysBeforeNow);\r
 154                 List<DBObject> rundowns = octopusAPI.getRundowns(scheduledDate.getTime());\r
 155                 if (rundowns == null) {\r
 156                         logger.warn(getMarker(), "Nem található adástükör a {} napra.", CalendarUtils.toDateString(scheduledDate));\r
 157                         return null;\r
 158                 }\r
 159 \r
 160                 processRundowns(rundowns);\r
 161                 return null;\r
 162         }\r
 163 \r
 164         //      private String getVersionedFileName(String fileName, String extension) throws Exception {\r
 165         //              String result = fileName;\r
 166         //              try {\r
 167         //                      targetFtp = ((FtpDirectoryLister) targetUri.getLister()).connect();\r
 168         //                      FTPFile[] listFiles = targetFtp.listFiles();\r
 169         //                      List<String> fileNames = new ArrayList<>();\r
 170         //                      for (FTPFile ftpFile : listFiles) {\r
 171         //                              fileNames.add(ftpFile.getName());\r
 172         //                      }\r
 173         //                      while (fileNames.contains(result + extension)) {\r
 174         //\r
 175         //                      }\r
 176         //\r
 177         //              } catch (Exception e) {\r
 178         //                      logger.catching(e);\r
 179         //                      throw e;\r
 180         //              } finally {\r
 181         //                      targetUri.cleanUp();\r
 182         //              }\r
 183         //              return result;\r
 184         //      }\r
 185 \r
 186         private FileArchive processMosObject(BasicDBObject rundown, BasicDBObject story, BasicDBObject mosObject) throws Exception {\r
 187                 String mosID = mosObject.getString(IOctopusAPI.OBJ_ID);\r
 188                 if (MetadataTypeDetector.GuessMetadataType(mosID) != MetadataType.OCTOPUSPLACEHOLDER) {\r
 189                         logger.trace("Skipping MOS object {}", mosID);\r
 190                         return null;\r
 191                 }\r
 192                 DBCollection clips = db.getCollection(NEXIOCLIPS);\r
 193                 BasicDBObject clip = (BasicDBObject) clips.findOne(new BasicDBObject(LONGNAMEID, mosID));\r
 194                 if (clip == null) {\r
 195                         logger.debug("File NOT exists {}", mosID);\r
 196                         return null;\r
 197                         //throw new Exception(String.format("File NOT exists %s", mosID));\r
 198                 } else {\r
 199                         logger.debug("File exists {}", mosID);\r
 200                 }\r
 201                 long duration = NoSQLUtils.asLong(clip, DURATION);\r
 202                 return new FileArchive(mosID, duration);\r
 203         }\r
 204 \r
 205         private RundownArchive processRundow(DBObject r) throws Exception {\r
 206                 BasicDBObject rundown = (BasicDBObject) r;\r
 207                 long rundownID = rundown.getLong(ID);\r
 208                 logger.info("Processing rundown {} {}", rundownID, rundown.getString(IOctopusAPI.NAME));\r
 209                 List<DBObject> stories = octopusAPI.getRundownFullStories(rundownID);\r
 210                 if (stories == null)\r
 211                         return null;\r
 212                 RundownArchive result = new RundownArchive();\r
 213 \r
 214                 long id = NoSQLUtils.asLong(rundown, IOctopusAPI.ID);\r
 215                 if (id == 0)\r
 216                         return null;\r
 217                 String name = NoSQLUtils.asString(NoSQLUtils.asDBObject(rundown, IOctopusAPI.RUNDOWN_TYPE), IOctopusAPI.NAME);\r
 218                 if (StringUtils.isBlank(name))\r
 219                         return null;\r
 220                 String channel = NoSQLUtils.asString(NoSQLUtils.asDBObject(rundown, IOctopusAPI.CHANNEL), IOctopusAPI.NAME);\r
 221                 Date scheduledStart = rundown.getDate(IOctopusAPI.SCHEDULED_START);\r
 222                 if (scheduledStart == null)\r
 223                         return null;\r
 224                 String start = CalendarUtils.toString(CalendarUtils.createCalendar(scheduledStart), SCHEDULED_FORMAT);\r
 225                 result.setScheduleDate(scheduledStart);\r
 226                 result.setItemHouseId(String.valueOf(id));\r
 227                 result.setItemTitle(String.format("%s %s %s", start, name, channel));\r
 228 \r
 229                 for (DBObject s : stories) {\r
 230                         StoryArchive storyArchive = processStory(rundown, s);\r
 231                         if (storyArchive == null)\r
 232                                 continue;\r
 233                         result.addStoryArchive(storyArchive);\r
 234                 }\r
 235                 return result;\r
 236         }\r
 237 \r
 238         private void processRundowns(List<DBObject> rundowns) {\r
 239 \r
 240                 //TODO kiveni publikálás előtt\r
 241                 //db.getCollection(ARCHIVEDRUNDOWNS).drop();\r
 242                 List<BasicDBObject> archivedRundowns = queryArchivedRundowns();\r
 243 \r
 244                 int index = 1;\r
 245 \r
 246                 for (DBObject r : rundowns) {\r
 247                         BasicDBObject rundown = (BasicDBObject) r;\r
 248                         setProgress(index * 100 / rundowns.size());\r
 249                         long rundownID = NoSQLUtils.asLong(rundown, IOctopusAPI.ID);\r
 250                         String rundownName = NoSQLUtils.asString(rundown, IOctopusAPI.NAME);\r
 251                         try {\r
 252                                 BasicDBObject currentRundownID = new BasicDBObject(IOctopusAPI.ID, rundownID);\r
 253                                 if (archivedRundowns != null && archivedRundowns.contains(currentRundownID)) {\r
 254                                         logger.info("Skipping archived rundown {} {}", rundownID, rundownName);\r
 255                                         continue;\r
 256                                 }\r
 257 \r
 258                                 RundownArchive rundownArchive = processRundow(r);\r
 259                                 if (rundownArchive == null || rundownArchive.isEmpty()) {\r
 260                                         logger.info("Skipping rundown {} {}", rundownID, rundownName);\r
 261                                         continue;\r
 262                                 }\r
 263 \r
 264                                 logger.info("Saving rundown {} {}", rundownID, rundownName);\r
 265                                 copy(rundownArchive);\r
 266 \r
 267                                 db.getCollection(ARCHIVEDRUNDOWNS).save(currentRundownID);\r
 268                         } catch (Exception e) {\r
 269                                 logger.catching(e);\r
 270                                 logger.error(getMarker(),\r
 271                                                 String.format("A %s %s tükör archiválása nem lehetséges, mert a annak ellenőrzése hibát jelzett. A rendszer üzenete: %s", rundownID,\r
 272                                                                 rundownName, e.getMessage()));\r
 273                         }\r
 274                         index++;\r
 275                 }\r
 276 \r
 277         }\r
 278 \r
 279         private StoryArchive processStory(BasicDBObject rundown, DBObject s) throws Exception {\r
 280                 BasicDBObject story = (BasicDBObject) s;\r
 281                 String parentStoryID = story.getString(IOctopusAPI.PARENT_STORY_ID);\r
 282                 if (StringUtils.isBlank(parentStoryID)) {\r
 283                         logger.warn("Story parentStoryID is null: {}", story.toPrettyString(null));\r
 284                         return null;\r
 285                 } else\r
 286                         logger.debug("Processing story {}", parentStoryID);\r
 287                 List<BasicDBObject> mosObjects = NoSQLUtils.asList(story, IOctopusAPI.MOS_OBJECTS);\r
 288                 if (mosObjects == null)\r
 289                         return null;\r
 290                 StoryArchive storyArchive = null;\r
 291                 for (BasicDBObject mosObject : mosObjects) {\r
 292                         FileArchive fileArchive = processMosObject(rundown, story, mosObject);\r
 293                         if (fileArchive == null)\r
 294                                 continue;\r
 295                         if (storyArchive == null) {\r
 296                                 storyArchive = new StoryArchive();\r
 297                                 storyArchive.setMediaHouseId(parentStoryID);\r
 298                                 storyArchive.setMediaTitle(story.getString(IOctopusAPI.NAME));\r
 299                                 storyArchive.setMediaDesc(story.getString(IOctopusAPI.SCRIPT_CONTENT));\r
 300 \r
 301                         }\r
 302                         storyArchive.addFileArchive(fileArchive);\r
 303                 }\r
 304                 return storyArchive;\r
 305 \r
 306         }\r
 307 \r
 308         private List<BasicDBObject> queryArchivedRundowns() {\r
 309                 List<BasicDBObject> result = null;\r
 310                 DBCollection collection = db.getCollection(ARCHIVEDRUNDOWNS);\r
 311                 DBCursor find = collection.find(new BasicDBObject(), new BasicDBObject(IOctopusAPI._ID, 0).append(IOctopusAPI.ID, 1));\r
 312                 if (find.hasNext())\r
 313                         result = ListUtils.cast(find.toArray());\r
 314                 return result;\r
 315         }\r
 316 \r
 317         private void setAndCheck(int nexioPort, String nexioUserName, String nexioPassword, String archiveFtp, String archiveUserName, String archivePassword,\r
 318                         int nexioKillDateDays, String nexioAgency, IJobEngine jobEngine) throws Exception {\r
 319                 db = NoSQLUtils.getNoSQLDB();\r
 320                 if (db == null) {\r
 321                         logger.error(getMarker(), "Az NoSQL adatkezelő réteg nem elérhető.");\r
 322                         throw new NullPointerException("Internal error, missing NoSQL DB reference.");\r
 323                 }\r
 324 \r
 325                 if (jobEngine == null) {\r
 326                         logger.error(getMarker(), "Az folyamatkezelő réteg nem elérhető.");\r
 327                         throw new NullPointerException("Internal error, missing JobEngine reference.");\r
 328                 }\r
 329                 manager = jobEngine.getItemManager();\r
 330                 if (manager == null) {\r
 331                         logger.error(getMarker(), "Az adatbáziskezelő réteg nem elérhető.");\r
 332                         throw new NullPointerException("Internal error, missing ItemManager reference.");\r
 333                 }\r
 334                 String nexioHost = System.getProperty("nexio.host");\r
 335                 if (StringUtils.isBlank(nexioHost)) {\r
 336                         logger.error(getMarker(), "A 'nexio.host' rendszer paraméter nem található.");\r
 337                         throw new NullPointerException("System is not configured properly, 'jobengine.selenio.address' startup parameter missing.");\r
 338                 }\r
 339                 check(nexioPort, "nexioPort");\r
 340                 check(nexioUserName, "nexioUserName");\r
 341                 check(nexioPassword, "nexioPassword");\r
 342 \r
 343                 check(nexioKillDateDays, "nexioKillDateDays");\r
 344                 this.nexioKillDateDays = nexioKillDateDays;\r
 345                 check(nexioAgency, "nexioAgency");\r
 346                 this.nexioAgency = nexioAgency;\r
 347 \r
 348                 sourceUri = manager.createStoreUri(RemoteStoreProtocol.FTP, nexioHost);\r
 349                 sourceUri.setPortNumber(nexioPort);\r
 350                 sourceUri.setUserName(nexioUserName);\r
 351                 sourceUri.setPassword(nexioPassword);\r
 352                 if (sourceUri == null) {\r
 353                         logger.error(getMarker(), "A forrás nem elérhető.");\r
 354                         throw new NullPointerException("Internal error, missing 'sourceUri'.");\r
 355                 }\r
 356 \r
 357                 check(archiveFtp, "archiveFtp");\r
 358                 check(archiveUserName, "archiveUserName");\r
 359                 check(archivePassword, "archivePassword");\r
 360 \r
 361                 targetUri = manager.createStoreUri(new URI(archiveFtp));\r
 362                 targetUri.setUserName(archiveUserName);\r
 363                 targetUri.setPassword(archivePassword);\r
 364                 if (targetUri == null) {\r
 365                         logger.error(getMarker(), "A cél nem elérhető.");\r
 366                         throw new NullPointerException("Internal error, missing 'targetUri'.");\r
 367                 }\r
 368 \r
 369         }\r
 370 \r
 371         private void transferFile(String sourceFileName, String targetFileName) throws Exception {\r
 372                 int reply = 0;\r
 373                 logger.info("Transfer clip {}", sourceFileName);\r
 374                 try {\r
 375                         sourceFtp = ((FtpDirectoryLister) sourceUri.getLister()).connect();\r
 376                         targetFtp = ((FtpDirectoryLister) targetUri.getLister()).connect();\r
 377                         if (!targetFtp.enterRemotePassiveMode())\r
 378                                 throw new Exception("!PASV");\r
 379 \r
 380                         reply = sourceFtp.port(InetAddress.getByName(targetFtp.getPassiveHost()), targetFtp.getPassivePort());\r
 381                         if (!FTPReply.isPositiveCompletion(reply))\r
 382                                 throw new Exception("PORT parancs válasza: " + sourceFtp.getReplyString());\r
 383 \r
 384                         if (!sourceFtp.setFileType(FTP.BINARY_FILE_TYPE))\r
 385                                 throw new Exception("!SOURCE TYPE");\r
 386 \r
 387                         reply = sourceFtp.retr(sourceFileName);\r
 388                         if (!FTPReply.isPositivePreliminary(reply))\r
 389                                 throw new Exception("RETR parancs válasza: " + sourceFtp.getReplyString());\r
 390 \r
 391                         if (!targetFtp.setFileType(FTP.BINARY_FILE_TYPE))\r
 392                                 throw new Exception("!TARGET TYPE");\r
 393 \r
 394                         reply = targetFtp.stor(targetFileName);\r
 395                         if (!FTPReply.isPositivePreliminary(reply))\r
 396                                 throw new Exception("STOR parancs válasza: " + sourceFtp.getReplyString());\r
 397 \r
 398                         while (true) {\r
 399                                 reply = sourceFtp.stat();\r
 400                                 if (!FTPReply.isPositiveCompletion(reply))\r
 401                                         throw new Exception("STAT parancs válasza: " + sourceFtp.getReplyString());\r
 402 \r
 403                                 logger.info("Status: {}", sourceFtp.getReplyString());\r
 404                                 if (reply == 226) {\r
 405                                         break;\r
 406                                 }\r
 407                                 Thread.sleep(1000);\r
 408                         }\r
 409                 } catch (Exception e) {\r
 410                         logger.catching(e);\r
 411                         throw e;\r
 412                 } finally {\r
 413                         sourceUri.cleanUp();\r
 414                         targetUri.cleanUp();\r
 415                 }\r
 416 \r
 417         }\r
 418 \r
 419         private void transferMetadata(String fileName, BasicDBObject metadata) throws Exception {\r
 420                 logger.info("Transfer metadata {}", fileName);\r
 421                 OutputStream outStream = null;\r
 422                 try {\r
 423                         targetFtp = ((FtpDirectoryLister) targetUri.getLister()).connect();\r
 424                         if (!targetFtp.changeWorkingDirectory(EscortFiles.STATUSFOLDER)) {\r
 425                                 targetFtp.makeDirectory(EscortFiles.STATUSFOLDER);\r
 426                                 if (!targetFtp.changeWorkingDirectory(EscortFiles.STATUSFOLDER))\r
 427                                         throw new Exception("!STATUSFOLDER");\r
 428                         }\r
 429 \r
 430                         outStream = targetFtp.storeFileStream(fileName + JSON_EXT);\r
 431                         if (outStream == null) {\r
 432                                 throw new NullPointerException("Can not open: " + fileName + JSON_EXT + " Reply:" + targetFtp.getReplyString());\r
 433                         }\r
 434                         outStream.write(metadata.toString().getBytes(UTF_8));\r
 435                         outStream.flush();\r
 436                         //targetFtp.changeToParentDirectory();\r
 437                 } catch (Exception e) {\r
 438                         logger.catching(e);\r
 439                         throw e;\r
 440                 } finally {\r
 441                         if (outStream != null)\r
 442                                 outStream.close();\r
 443                         targetUri.cleanUp();\r
 444                 }\r
 445         }\r
 446 \r
 447 }\r