module Copernicium::Workspace

# File lib/workspace.rb, line 132
def Workspace.checkout(comm = UIComm.new)
  comm.rev = Repos.current_head if comm.rev.nil?
  if ! Repos.has_snapshots? # dont checkout
    puts 'No snapshots yet! Commit something before checkout.'.red
  elsif comm.files.nil? # checkout everything
    Repos.get_snapshot(comm.rev).files.each do |file|
      Workspace.checkout_file file
    end
  else # just checkout given files
    Repos.get_snapshot(comm.rev).files.select do |f|
      comm.files.include? f.path
    end .each do |file|
      Workspace.checkout_file file
    end
  end
end

# File lib/workspace.rb, line 124
def Workspace.checkout_file(file)
  idx = indexOf(file.path)
  idx.nil?? @@files << file : @@files[idx] = file
  Dir.mkdir file.base unless Dir.exist? file.base
  File.write(file.path, RevLog.get_file(file.last))
end

# File lib/workspace.rb, line 150
def Workspace.clean(comm = UIComm.new)
  if comm.files.nil? # reset everything
    Workspace.clear
  else # files are not nil
    comm.files.each do |x|
      idx = indexOf(x)
      if idx.nil?
        puts "Cannot clean #{x}:".yel + " does not exist in snapshot"
      else
        @@files.delete_at(idx)
        File.delete(x)
      end
    end
  end
  Workspace.checkout comm # cleanse state
end

# File lib/workspace.rb, line 118
def Workspace.clear
  @@files.each{ |x| File.delete(x.path) }
  @@files = []
end

# File lib/workspace.rb, line 184
def Workspace.commit(comm = UIComm.new)
  if comm.files.nil? # commit everything
    Workspace.status.each do |st|
      st.each { |x| Workspace.commit_file x }
    end
  else comm.files.each do |x|
      if File.exist? x
        Workspace.commit_file(x)
      else
        puts "Cannot commit #{x}:".yel + " file does not exist"
      end
    end
  end
  # todo - handle case of no changes, dont make a snapshot
  Repos.make_snapshot(@@files, comm.cmt_msg) # return snapshot id
end

# File lib/workspace.rb, line 167
def Workspace.commit_file(x)
  puts 'Committing: '.grn + x
  added, edits, remov = Workspace.status
  if added.include? x
    hash = RevLog.add_file(x, File.read(x))
    @@files.push(FileObj.new x, [hash])
  elsif edits.include? x
    hash = RevLog.add_file(x, File.read(x))
    @@files[indexOf x].history << hash
  elsif remov.include? x
    @@files.delete_at(indexOf x)
  else
    puts 'No changes: '.yel + x
  end
end

# File lib/workspace.rb, line 44
def Workspace.create_project(location = Dir.pwd, branch = 'master')
  Dir.mkdir location unless Dir.exist? location
  Dir.chdir location

  # create our copernicium folders
  @@copn = File.join('.', '.cn')
  @@snap = File.join(@@copn, 'snap')
  @@revs = File.join(@@copn, 'revs')
  @@head = File.join(@@copn, 'branch')
  @@hist = File.join(@@copn, 'history')
  Dir.mkdir(@@copn) unless Dir.exist?(@@copn)
  Dir.mkdir(@@snap) unless Dir.exist?(@@snap)
  Dir.mkdir(@@revs) unless Dir.exist?(@@revs)

  # make default branch, history
  hist = YAML.dump({branch => []})
  File.write @@head, branch
  File.write @@hist, hist

  if Dir.exist?(@@copn)
    location # return where we made the repo
  else # something has gone horribly wrong
    raise 'Could not create or find a Copernicium folder (.cn).'.red
  end
end

# File lib/workspace.rb, line 100
def Workspace.indexOf(x)
  @@files.each_with_index do |f, i|
    return i if f.path == x
  end
  nil # x is not included in @@files
end

# File lib/workspace.rb, line 205
def Workspace.merge(id)
  Repos.merge_snapshot(id)
end

# File lib/workspace.rb, line 30
def Workspace.setup
  @@cwd = Dir.pwd
  @@root = (noroot?? @@cwd : getroot)
  @@root.sub!(@@cwd, '.')
  @@copn = File.join(@@root, '.cn')
  @@snap = File.join(@@copn, 'snap')
  @@revs = File.join(@@copn, 'revs')
  RevLog.setup @@root
  Repos.setup @@root
  @@files = Repos.current_files
  @@branch = Repos.current
end

# File lib/workspace.rb, line 209
def Workspace.status
  added = []
  edits = []
  remov = []
  working_files.each do |f|
    idx = indexOf(f)
    if idx.nil? # new file
      added << f
    else # changed file
      edits << f if File.read(f) != RevLog.get_file(@@files[idx].last)
    end
  end

  # any deleted files from the last commit?
  remov = @@files.map(&:path) - working_files

  [added, edits, remov]
end

# File lib/workspace.rb, line 109
def Workspace.working_files
  (Dir[ File.join(@@root, '**', '*') ].reject do |p|
    File.directory? p || p.include?(@@copn)
  end).map do |p|
    p.sub!(/^\.\//, '') # delete leading './'
  end
end

# File lib/workspace.rb, line 74
def getroot
  cwd = Dir.pwd
  max = 0
  def more_folders() Dir.pwd != '/' end
  def root_found() Dir.exist? File.join(Dir.pwd, '.cn') end
  while max < 10 && more_folders && !root_found
    Dir.chdir(File.join(Dir.pwd, '..'))
    max += 1
  end

  if root_found # return where cn was found
    cnroot = Dir.pwd
    Dir.chdir(cwd)
    cnroot
  else # directory not found
    Dir.chdir(cwd)
    nil
  end
end

# File lib/workspace.rb, line 77
def more_folders() Dir.pwd != '/' end

# File lib/workspace.rb, line 95
def noroot?() getroot.nil? end

# File lib/workspace.rb, line 78
def root_found() Dir.exist? File.join(Dir.pwd, '.cn') end