/**
 * jQuery.selection - jQuery Plugin
 *
 * Under The MIT License
 * Copyright (c) 2010 Iwasaki. (http://d.hatena.ne.jp/ja9/)
 * 
 * Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 * 
 * The above copyright notice and this permission notice shall be
 * included in all copies or substantial portions of the Software.
 * 
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 * Version: 1.0 beta
 * Revision: $Rev$
 * Date: $Date$
 */
(function($) {
  var selectionFunctions = {
        getCaretData: function(element) {
          var res = {
                text: '',
                startNode: undefined,
                endNode: undefined,
                start: 0,
                end: 0
              };

          if (!element.value && !$(element).text()) {
            // 値がない、もしくは空文字列
            return res;
          }

          try {
            if (window.getSelection && 'undefined' == typeof document.all) {
              // IE 以外
              if ($(element).attr('contenteditable')) {
                // {{{ contenteditable
                var r = window.getSelection().getRangeAt(0).cloneRange();
                res.startNode = r.startContainer;
                res.endNode = r.endContainer;
                res.start = r.startOffset;
                res.end = r.endOffset;
                // }}}
              } else {
                res.start = element.selectionStart;
                res.end = element.selectionEnd;
              }
              res.text = (element.value || $(element).text()).slice(res.start, res.end);
            } else if (document.selection) {
              // IE
              element.focus();

              var range = document.selection.createRange(),
                  range2 = document.body.createTextRange(),
                  text,
                  text2;

              res.text = range.text;

              range2.moveToElementText(element);
              // EndToStart だと後ろの改行がトリムされる
              range2.setEndPoint('StartToStart', range);

              if ($(element).attr('contenteditable')) {
                // {{{ contenteditable
                text = element.innerText.replace(/\r\n|\r|\n/g, '_');
                text2 = range2.text.replace(/\r\n|\r|\n/g, '_');
                // }}}
              } else {
                text = element.value;
                text2 = range2.text;
              }

              res.start = text.length - text2.length;
              res.end = res.start + range.text.replace(/\r\n|\r|\n/g, '_').length;
            }
          } catch (e) {
          }

          return res;
        },

        getCaret: function(element) {
          var tmp = this.getCaretData(element);
          return {start: tmp.start, end: tmp.end, startNode: tmp.startNode, endNode: tmp.endNode};
        },

        getText: function(element) {
          return this.getCaretData(element).text;
        },

        caretMode: function(caret) {
          caret = caret || 'keep';
          if (caret == false) {
            caret = 'end';
          }

          switch (caret) {
            case 'keep':
            case 'start':
            case 'end':
              break;

            default:
              caret = 'keep';
          }

          return caret;
        },

        replace: function(element, text, caret) {
          var tmp = this.getCaretData(element),
              orig = element.value,
              pos = $(element).scrollTop(),
              range = {start: tmp.start, end: tmp.start + text.length};

          element.value = orig.substr(0, tmp.start) + text + orig.substr(tmp.end);

          $(element).scrollTop(pos);
          this.setCaret(element, range, caret);
          },

        insertBefore: function(element, text, caret) {
          var tmp = this.getCaretData(element),
              orig = element.value,
              pos = $(element).scrollTop(),
              range = {start: tmp.start + text.length, end: tmp.end + text.length};

          element.value = orig.substr(0, tmp.start) + text + orig.substr(tmp.start);

          $(element).scrollTop(pos);
          this.setCaret(element, range, caret);
        },

        insertAfter: function(element, text, caret) {
          var tmp = this.getCaretData(element),
              orig = element.value,
              pos = $(element).scrollTop(),
              range = {start: tmp.start, end: tmp.end};

          element.value = orig.substr(0, tmp.end) + text + orig.substr(tmp.end);

          $(element).scrollTop(pos);
          this.setCaret(element, range, caret);
        },

        setCaret: function(element, toRange, caret) {
          caret = this.caretMode(caret);

          if (caret == 'start') {
            toRange.end = toRange.start;
          } else if (caret == 'end') {
            toRange.start = toRange.end;
          }

          element.focus();
          try {
            if (element.createTextRange || document.body.createTextRange) {
              var range = null;
              if ($(element).attr('contenteditable')) {
                // {{{ contenteditable
                range = document.body.createTextRange();
                range.moveToElementText($(element).get(0));
                // }}}
              } else {
                range = element.createTextRange();
              }

              if (window.navigator.userAgent.toLowerCase().indexOf("msie") >= 0) {
                var v = ((element.value || element.innerText) || '').replace(/\r\n|\r|\n/g, '_');
                toRange.start = v.substr(0, toRange.start).length;
                toRange.end = v.substr(0, toRange.end).length;
                // {{{ contenteditable
                if ($(element).attr('contenteditable') && window.navigator.userAgent.toLowerCase().indexOf("msie 9.0") >= 0) {
                  // count image or line break
                  var f_special_count = function(element, index) {
                    var base_text = element.innerText.replace(/\r\n|\r|\n/g, '_').substr(0, index);
                    var html_text = $(element).html().replace(/<br[^>]*>/ig, '\x0b').replace(/\r\n|\r|\n/g, '\x0b').replace(/<img[^>]*>/ig, '\x0b').replace(/<[^>]*>/g, '');
                    var count = 0;
                    for (var i = 0; i < base_text.length; i++) {
                      if (html_text.substr(i, 1) == '\x0b') {
                        count++;
                      }
                    }
                    return count;
                  };
                  toRange.start = toRange.start + f_special_count(element, toRange.start);
                  toRange.end = toRange.end + f_special_count(element, toRange.end - 1);
                }
                // }}}
              }

              range.collapse(true);
              range.moveStart('character', toRange.start);
              range.moveEnd('character', toRange.end - toRange.start);

              range.select();
            } else if (element.setSelectionRange) {
              if ($(element).attr('contenteditable')) {
                var s = window.getSelection();
                var r = document.createRange();
                r.setStart(toRange.startNode, toRange.start);
                r.setEnd(toRange.endNode, toRange.end);
                s.removeAllRanges();
                s.addRange(r);
              } else {
                element.setSelectionRange(toRange.start, toRange.end);
              }
            }
          } catch (e) {
          }
        }
      },

      selectionMethods = {
        getSelection: function(mode) {
          var getText = ((mode || 'text').toLowerCase() == 'text');

          try {
            if (window.getSelection) {
              if (getText) {
                // get text
                return window.getSelection().toString();
              } else {
                // get html
                var sel = window.getSelection(), range;

                if (sel.getRangeAt) {
                  range = sel.getRangeAt(0);
                } else {
                  range = document.createRange();
                  range.setStart(sel.anchorNode, sel.anchorOffset);
                  range.setEnd(sel.focusNode, sel.focusOffset);
                }

                return $('<div></div>').append(range.cloneContents()).html();
              }
            } else if (document.selection) {
              if (getText) {
                // get text
                return document.selection.createRange().text;
              } else {
                // get html
                return document.selection.createRange().htmlText;
              }
            }
          } catch (e) {
          }

          return '';
        }
      },

      selectionPrototypeMethods = {
        getSelection: function() {
          var res = [];

          this.each(function() {
            res.push(selectionFunctions.getText(this));
          });

          return (res.length < 2) ? res[0] : res;
        },

        replaceSelection: function(text, caret) {
          return this.each(function() {
            selectionFunctions.replace(this, text, caret);
          });
        },

        insertBeforeSelection: function(text, caret) {
          return this.each(function() {
            selectionFunctions.insertBefore(this, text, caret);
          });
        },

        insertAfterSelection: function(text, caret) {
          return this.each(function() {
            selectionFunctions.insertAfter(this, text, caret);
          });
        },

        getCaretPos: function() {
          var res = [];

          this.each(function() {
            res.push(selectionFunctions.getCaret(this));
          });

          return (res.length < 2) ? res[0] : res;
        },

        setCaretPos: function(range) {
          return this.each(function() {
            selectionFunctions.setCaret(this, range);
          });
        }
      };

  $.extend(selectionMethods);
  $.fn.extend(selectionPrototypeMethods);
})(jQuery);