实现当前IE的当前页面截图

由于最近工作比较忙没什么时间写文章，正好今天抽出点时间来写写博客。说实话做测试真是一个无聊的事情，哪怕它很重要但是真的很无聊。昨天英语课上还有人问什么才是Creative Bug, I think nobody find it expect you. This is the creative bug。总之在微软报bug也事件不容易的事情，特别你是flashman. 不过总是来也半年终于在最近的一个月报了11个bug，体验一把真正测试的feeling。废话少说，进入正题：
最近正在忙一个tool的发布准备，这个tool是一个实习生开发的，走后留给我来开发，其实功能比较简单。大家都知道键盘上有一个PrScrn的键，right! 截屏，我们这个工具也是干这个活的。不过呢不是Capture Screen, 而是Capture IE web page。这个工具的产生也主要是因为我们的automation case中需要这个功能，不过现在my boss want to public it to toolbox(a webstie in microsoft to public tool). 大家知道IE页面有的带滚动条有的不带，如果你通过系统的截屏功能那么只能截取到一部分的页面。What a pity! 不过现在网路上也有这样的工具来截整屏了，但是如果你要用在你的Program里，impossible! 所以我们开发了这样一个接口用于完成这个工作。截屏的原理很简单，
1. 获取页面高度和宽度
2. 截取当前页面的显示部分，画到一个graphic上
3. 滚动页面，重复2，直到结束
4. 保存成图片
这些工作我们可以使用.Net中的Bitmap, Graphic, HDC, IO 还有一些API来完成。

body.setAttribute("scroll", "yes", 0);
int webHeight = (int)body.getAttribute("scrollHeight", 0);
int webWidth = (int)body.getAttribute("scrollWidth", 0);

int screenHeight = (int)body.getAttribute("clientHeight", 0);
int screenWidth = (int)body.getAttribute("clientWidth", 0);
.
.
.
g = Graphics.FromImage(screenFragment);
//handle to the device context
hdc = g.GetHdc();
body.setAttribute("scrollTop", (screenHeight - 5) * i, 0);
brwTop = (int)body.getAttribute("scrollTop", 0);
//copies a visual window into the specified device context(DC)
//hwnd - Handle to the window that will be copied
//hdc - Handle to the device context
//0 - specifies the drawing options.

IntPtr hPageWin = hCurDisWin;
IEAttachAssist.PrintWindow(hPageWin, hdc, 0);
g.ReleaseHdc(hdc);
g.Flush();

screenFrag = Image.FromHbitmap(screenFragment.GetHbitmap());
try
catch (System.ArgumentNullException ex)
.
.
.这里是一部分代码片段。
比较麻烦的是，如何获取当前显示的浏览器还要兼容IE6和IE7。因为IE6是没有Tab页面标签的而IE7有。So, how do it?
还有对于html和DTD 声明的html处理的方式也是不一样的，这也是个问题。而我碰到最最麻烦的问题是在IE7里对每一个Tab页面都是一个单独的WebBroswer对象，但是呢，他们的ShellWindow句柄是一样的。如何获取当前显示页面句柄和对应的HtmlDocument?
这两天搞的我是头昏眼花，不过努力没有白费，问题逐一解决。So, next I will explain how to resolve these problem.
first. 兼容IE6和IE7? 这个问题主要是因为IE6的Window handle tree结构和IE7的不同。我通过编写各自的函数来解决，同时你要在架构是解决对象调用的透明性，我使用了工厂模式。
以下是获取IE7当前显示窗口句柄函数:

static public IntPtr GetCurPageFromTabWindow(IntPtr handle, ref StringBuilder title)
static public bool IsDTDDocument(object document)
static public IntPtr GetActiveIEHandle()

static private IntPtr GetProcessMainWindowHandle(string title)
Process[] curProcesses = Process.GetProcesses();

foreach (Process p in curProcesses)
if (p.MainWindowTitle.Contains(title))
return p.MainWindowHandle;
}

return IntPtr.Zero;
}
通过这次的开发，确实对IE对网页的处理加深了不少认识。同时我也发现了firefox的缺点，就是很难在程序中去操作firefox。how terrible!
希望这篇文章对大家有帮助。
这里是一个Demo工程，编译环境是.net2.0.