LÖVE

Code: Select all

-- check if drive is external and return its size
function GetDriveSize(driveLetter)
	local text = ""
	local ret = ""
	local script = [[ Get-WmiObject -Class win32_logicaldisk -Filter "DriveType = '2'" | Select-Object -Property DeviceID,Size ]]
		
	local sd = love.filesystem.getAppdataDirectory() .. "/LOVE/temp.tmp"

	local p = io.popen("powershell -command - >" .. sd, "w")
	p:write(script)
	p:close()
	
	if GetExtFileExists(sd) then
		file = io.open(sd, "rb")
		local text = file:read("*a")
		file:close()

		local l = utf8.len(text)		
		
		for i = 1, l do
			local c = string.sub(text, utf8.offset(text, i), utf8.offset(text, i))
			local a = bit.band(c:byte(), 127)

			-- drive letter found
			if a == string.byte(driveLetter) then
				if i < l then
					c = string.sub(text, utf8.offset(text, i + 1), utf8.offset(text, i + 1))
					a = bit.band(c:byte(), 127)
					
					if Chr(a) == ":" then
						for j = i + 2, l do
							c = string.sub(text, utf8.offset(text, j), utf8.offset(text, j))
							a = bit.band(c:byte(), 127)
							
							if (a >= 48 and a <= 57) then
								ret = ret .. string.char(a)
							elseif a ~= 32 then
								break
							end
						end
					end
					
					if ret ~= "" then break end
				end
			end
		end
		
		os.remove(sd)
	else
		return ""
	end

	return tonumber(ret)
end

function Asc(c)
	if c == nil or #c > 1 then return nil end
	
	return string.byte(c, 1, 1)
end

function GetExtFileExists(filename)
	local f = io.open(filename, "r")
	if f ~= nil then io.close(f) return true else return false end
end