scripts/internal/journal/journal_context.lua at 713b0902022972c48e3c09b77a1ab04e3966733e · DFHack/scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
--@ module = true

local widgets = require 'gui.widgets'
local utils = require 'utils'
local DummyJournalContext = reqscript('internal/journal/contexts/dummy').DummyJournalContext
local FortressJournalContext = reqscript('internal/journal/contexts/fortress').FortressJournalContext
local WorldmapJournalContext = reqscript('internal/journal/contexts/worldmap').WorldmapJournalContext
local AdventurerJournalContext = reqscript('internal/journal/contexts/adventure').AdventurerJournalContext

JOURNAL_CONTEXT_MODE = {
  FORTRESS='fortress',
  ADVENTURE='adventure',
  WORLDMAP='worldmap',
  LEGENDS='legends',
  DUMMY='dummy'
}

function detect_journal_context_mode()
  if not dfhack.isMapLoaded() and dfhack.world.isFortressMode() then
    return JOURNAL_CONTEXT_MODE.WORLDMAP
  elseif dfhack.isMapLoaded() and dfhack.world.isFortressMode() then
    return JOURNAL_CONTEXT_MODE.FORTRESS
  elseif dfhack.isMapLoaded() and dfhack.world.isAdventureMode() then
    return JOURNAL_CONTEXT_MODE.ADVENTURE
  elseif dfhack.world.isLegends() then
    return JOURNAL_CONTEXT_MODE.LEGENDS
  else
    return nil
  end
end

function journal_context_factory(journal_context_mode, save_prefix)
  local world_id =  df.global.world.cur_savegame.world_header.id1
  local worldmap_journal_context = WorldmapJournalContext{save_prefix=save_prefix, world_id=world_id}

  if journal_context_mode == JOURNAL_CONTEXT_MODE.FORTRESS then
    return FortressJournalContext{
      save_prefix=save_prefix,
      worldmap_journal_context=worldmap_journal_context
    }
  elseif journal_context_mode == JOURNAL_CONTEXT_MODE.ADVENTURE then
    local interactions = df.global.adventure.interactions
    if #interactions.party_core_members == 0 or interactions.party_core_members[0] == nil then
      qerror('Can not identify party core member')
    end

    local adventurer_id = interactions.party_core_members[0]

    return AdventurerJournalContext{
      save_prefix=save_prefix,
      adventurer_id=adventurer_id,
      worldmap_journal_context=worldmap_journal_context
    }
  elseif journal_context_mode == JOURNAL_CONTEXT_MODE.WORLDMAP then
    return worldmap_journal_context
  elseif journal_context_mode == JOURNAL_CONTEXT_MODE.LEGENDS then
    return worldmap_journal_context
  elseif journal_context_mode == JOURNAL_CONTEXT_MODE.DUMMY then
    return DummyJournalContext{}
  else
    qerror('unsupported game mode')
  end
end