Registry = require 'winreg'
Path = require 'path'

exeName = Path.basename(process.execPath)
appPath = "\"#{process.execPath}\""
appName = exeName.replace('atom', 'Atom').replace('beta', 'Beta').replace('.exe', '')

class ShellOption
  constructor: (key, parts) ->
    @key = key
    @parts = parts
  isRegistered: (callback) =>
    new Registry({hive: 'HKCU', key: "#{@key}\\#{@parts[0].key}"})
      .get @parts[0].name, (err, val) ->
        callback(not err? and val.value is @parts[0].value)
  register: (callback) ->
    doneCount = @parts.length
    @parts.forEach((part) ->
      reg = new Registry({hive: 'HKCU', key: if @part.key? then "#{@key}\\#{@part.key}" else @key})
      reg.create( -> reg.set part.name, Registry.REG_SZ, part.value, -> callback() if doneCount-- is 0))
  deregister: (callback) ->
    @isRegistered (isRegistered) ->
      if isRegistered
        new Registry({hive: 'HKCU', key: @key}).destroy callback null, true
      else
        callback null, false
  update: (callback) ->
    new Registry({hive: 'HKCU', key: "#{@key}\\#{@parts[0].key}"})
      .get @parts[0].name, (err, val) ->
        if err? or not val.value.endsWith '\\' + exeName
          callback err, val
        else
          register callback

exports.fileHandler = new ShellOption("\\Software\\Classes\\Applications\\#{exeName}",
  parts: [{key: 'shell\\open\\command', name: '', value: "#{appPath} \"%1\""}]
)

contextRegistrationParts = [
    {key: 'command', name: '', value: "#{appPath} \"%1\""},
    {name: '', value: "Open with #{appName}"},
    {name: 'Icon', value: "#{appPath}"}
]

exports.fileContextMenu = new ShellOption("\\Software\\Classes\\*\\shell\\#{appName}", parts: contextRegistryParts)

exports.folderContextMenu = new ShellOption("\\Software\\Classes\\Directory\\shell\\#{appName}", parts: contextRegistryParts)

exports.folderBackgroundContextMenu = new ShellOption("\\Software\\Classes\\Directory\\background\\shell\\#{appName}",
  parts: JSON.parse(JSON.stringify(contextRegistryParts).replace('%1', '%V'))
)